Add type annotation

yuokada · Dec 8, 2019 · c1c7685 · c1c7685
1 parent 3a73fa2
commit c1c7685
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main/scala/org/embulk/output/orc/OrcOutputPluginHelper.scala b/src/main/scala/org/embulk/output/orc/OrcOutputPluginHelper.scala
@@ -35,8 +35,8 @@ object OrcOutputPluginHelper {
     case _ => false
   }
 
-  def getSchema(fpath: String) = {
-    val schema = fpath.split("://").toList.apply(0)
+  def getSchema(fpath: String): String = {
+    val schema = fpath.split("://").toList.head
     schema match {
       case "s3" | "s3a" | "s3n" => schema
       case _ => {
@@ -46,12 +46,13 @@ object OrcOutputPluginHelper {
     }
   }
 
-  def parseS3Url(s3url: String) = {
+  def parseS3Url(s3url: String): AmazonS3URILikeObject = {
     val parts = s3url.split("(://|/)").toList
     val bucket = parts.apply(1)
     val key = parts.slice(2, parts.size).mkString("/")
     new OrcOutputPluginHelper.AmazonS3URILikeObject(bucket, key)
   }
 
   case class AmazonS3URILikeObject(@BeanProperty bucket: String, @BeanProperty key: String)
+
 }