core/src/main/scala/spark/network/ReceiverTest.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

package spark.network

import java.nio.ByteBuffer
import java.net.InetAddress

object ReceiverTest {

  def main(args: Array[String]) {
    val manager = new ConnectionManager(9999)
    println("Started connection manager with id = " + manager.id)
    
    manager.onReceiveMessage((msg: Message, id: ConnectionManagerId) => { 
      /*println("Received [" + msg + "] from [" + id + "] at " + System.currentTimeMillis)*/
      val buffer = ByteBuffer.wrap("response".getBytes())
      Some(Message.createBufferMessage(buffer, msg.id))
    })
    Thread.currentThread.join()  
  }
}