在项目中,经常需要对字符串进行切割转换,以日期字符串“2018121810”为例,惯性的思维是利用DateFormat函数,如下:
val srcDf:Sdf = new Sdf("yyyyMMddHH")
val time: Date = srcDf.parse("2018080810")
val toDf: Sdf = new Sdf("yyyy/MM/dd/HH")
val output = toDf.format(time)
上述方法进行了多次类型转换,效率不高,所以还不如直接切割,如下:
// 依次获取年月日时
val year = srcDate.substring(0, 4)
val month = srcDate.substring(4, 6)
val day = srcDate.substring(6, 8)
val hour = srcDate.substring(8)
return s"$year/$month/$day/$hour";
但这种方法重复太多,略显啰嗦,太不scala了,再次改进如下:
implicit class ToDatePath(srcDate: String) {
def toPath(): String = {
val ds = for(i <- 0 to 8 if i % 2 == 0 if i != 2) yield srcDate.substring(i, i + (if(i == 0) 4 else 2))
ds.mkString("/")
}
}
重复scala的特性,生成了新的字符集合,然后再进行合并,代码一气呵成,很赞!
更值得注意的是,还为字符串(String)扩展了新的方法,使得字符串的调用更加美观直接,如下:
// 输出2018/09/03/10
"2018090310".toPath()
扩展字符串函数的两种方法
1. 定义隐式对象
// 注意这里只能使用def定义
implicit def toDatePath(srcDate: String) = new {
def toPath(): String = {
val ds = for(i <- 0 to 8 if i % 2 == 0 if i != 2) yield srcDate.substring(i, i + (if(i == 0) 4 else 2))
ds.mkString("/")
}
}
2. 定义隐式类
implicit class ToDatePath(srcDate: String) {
def toPath(): String = {
val ds = for(i <- 0 to 8 if i % 2 == 0 if i != 2) yield srcDate.substring(i, i + (if(i == 0) 4 else 2))
ds.mkString("/")
}
}