scala spark 数据对比,小于Spark Scala RDD中日期的比较

I want to print data of employees who joined before 1991. Below is my sample data:

69062,FRANK,ANALYST,5646,1991-12-03,3100.00,,2001

63679,SANDRINE,CLERK,69062,1990-12-18,900.00,,2001

Initial RDD for loading data:

val rdd=sc.textFile("file:home/hduser/Desktop/Employees/employees.txt").filter(p=>{p!=null && p.trim.length>0})

UDF for converting string column to date column:

def convertStringToDate(s: String): Date = {

val dateFormat = new SimpleDateFormat("yyyy-MM-dd")

dateFormat.parse(s)

}

Mapping each and every column to its datatype:

val dateRdd=rdd.map(_.split(",")).map(p=>(if(p(0).length >0 )p(0).toLong else 0L,p(1),p(2),if(p(3).length > 0)p(3).toLong else 0L,convertStringToDate(p(4)),if(p(5).length >0)p(5).toDouble else 0D,if(p(6).length > 0)p(6).toDouble else 0D,if(p(7).length> 0)p(7).toInt else 0))

Now I get data in tuples as below:

(69062,FRANK,ANALYST,5646,Tue Dec 03 00:00:00 IST 1991,3100.0,0.0,2001)

(63679,SANDRINE,CLERK,69062,Tue Dec 18 00:00:00 IST 1990,900.0,0.0,2001)

Now when I execute command I am getting below error:

scala> dateRdd.map(p=>(!(p._5.before("1991")))).foreach(println)

:36: error: type mismatch;

found : String("1991")

required: java.util.Date

dateRdd.map(p=>(!(p._5.before("1991")))).foreach(println)

^

So where am I going wrong ???

解决方案

Since you are working with rdd's and no df's and you have date strings with simple date checking, the following non-complicated way for an RDD:

val rdd = sc.parallelize(Seq((69062,"FRANK","ANALYST",5646, "1991-12-03",3100.00,2001),(63679,"SANDRINE","CLERK",69062,"1990-12-18",900.00,2001)))

rdd.filter(p=>(p._5 < "1991-01-01")).foreach(println)