Commit e1c201cc authored by 赵威's avatar 赵威

add printer

parent 033915c9
...@@ -53,9 +53,10 @@ object Main { ...@@ -53,9 +53,10 @@ object Main {
val stream: DataStream[UserInfo] = env.addSource(kafkaConsumer) val stream: DataStream[UserInfo] = env.addSource(kafkaConsumer)
stream.map { user => stream.map { user =>
println(user.deviceId)
val projects = user.projects.toList val projects = user.projects.toList
val secondDemands = user.secondDemands.toList val secondDemands = user.secondDemands.toList
println(user.deviceId)
println(projects.mkString(" "))
if ((projects.size > 0) || (secondDemands.size > 0)) { if ((projects.size > 0) || (secondDemands.size > 0)) {
val diaryQuery = ES.generateDiaryQuery(projects, secondDemands, -1) val diaryQuery = ES.generateDiaryQuery(projects, secondDemands, -1)
val tractateQuery = ES.generateTractateQuery(projects, secondDemands) val tractateQuery = ES.generateTractateQuery(projects, secondDemands)
......
...@@ -42,12 +42,12 @@ object ESClient { ...@@ -42,12 +42,12 @@ object ESClient {
} }
object ES { object ES {
case class Character(id: String, index: String) case class Character(id: String, index: String, projects: Array[String])
implicit object CharacterHitReader extends HitReader[Character] { implicit object CharacterHitReader extends HitReader[Character] {
override def read(hit: Hit): Try[Character] = { override def read(hit: Hit): Try[Character] = {
val source = hit.sourceAsMap val source = hit.sourceAsMap
Try(Character(source("id").toString, hit.index)) Try(Character(source("id").toString, hit.index, source("tags_v3").asInstanceOf[Array[String]]))
} }
} }
...@@ -71,7 +71,9 @@ object ES { ...@@ -71,7 +71,9 @@ object ES {
println("%%%%%%%%%%%") println("%%%%%%%%%%%")
println(a.size) println(a.size)
a.foreach(x => println(x.id, x.index)) a.foreach { x =>
println(x.id, x.index, x.projects.mkString(" "))
}
println("%%%%%%%%%%%") println("%%%%%%%%%%%")
resp resp
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment