5.scala模式匹配及样例类的简单练习--RPC通信

需求：NodeManager作为客户端，ResourceManager作为服务端，客户端向服务端发送注册信息，服务端处理并返回给客户端成功注册的提示，客户端之后再保持心跳。大致过程如图：



代码部分：

样例类：Message.scala

package lesson10

//这里为了体现trait的作用，其实样例类底层已经实现了序列化
trait Message extends Serializable
//客户端向服务进行注册的消息
case class RegisterMessage(username:String,password:String) extends Message
//结果消息
case class ResultMessage(code:Int,content:String) extends  Message
//客户端向服务端发送心跳消息
case class HeartBeatMessage(id:String,content:String) extends  Message

客户端：MyNodeManager

package lesson10

import java.io.{ObjectInputStream, ObjectOutputStream}
import java.net.Socket

/**
* 客户端
*/
object MyNodeManager {

def main(args: Array[String]): Unit = {
val socket = new Socket("127.0.0.1",8888)
//发送注册消息 （case class   Object对象）
val oos = new ObjectOutputStream(socket.getOutputStream())
oos.writeObject(RegisterMessage("xiaoma","123"))
oos.flush()

val ois = new ObjectInputStream(socket.getInputStream)
val registerResultMessage = ois.readObject()
//打印注册消息
println(registerResultMessage)

registerResultMessage match {
case ResultMessage(id,content) =>{
oos.writeObject(HeartBeatMessage("hadoop1","alive"))
oos.flush()
}
}
val heartBeatMessage=ois.readObject()
//打印心跳
println(heartBeatMessage)

oos.close()

}

}

服务端：MyResourceManager

package lesson10

import java.io.{ObjectInputStream, ObjectOutputStream}
import java.net.ServerSocket

/**
* 服务端
*/
object MyResourceManager {
//处理客户端注册请求
def handlerRegister(username:String,password:String):ResultMessage={
// 如果这儿是spark，那么spark就会把注册过来的信息存到 内存里面
//内存易丢失，他还会把信息存到zookeeper里面
println("成功调用处理注册的方法")
ResultMessage(1,"success")
}

def hendlerHeartBeatMessage(hostname:String,status:String): ResultMessage={
println("成功调用处理心跳的方法")
ResultMessage(2,"success")
}

def main(args: Array[String]): Unit = {
//服务端
val serverSocket = new ServerSocket(8888)
val accept = serverSocket.accept()
val ois = new ObjectInputStream(accept.getInputStream)
val oos = new ObjectOutputStream(accept.getOutputStream)

while (true){
val clientMsg = ois.readObject()
val result = clientMsg match {
case RegisterMessage(username, password) => {
//做相对应的操作就可以了,并返回处理结果
handlerRegister(username, password)
}
case HeartBeatMessage(hostname,status) =>{
hendlerHeartBeatMessage(hostname,status)
}
}

oos.writeObject(result)
oos.flush()
}

}

}