// connect to remote servant peer
func (this *funServantPeerPool) connect(peerAddr string) (*rpc.FunServantClient,
error) {
transportFactory := thrift.NewTBufferedTransportFactory(this.cf.BufferSize)
transport, err := thrift.NewTSocketTimeout(peerAddr, this.cf.IoTimeout)
if err != nil {
return nil, err
}
if err = transport.Open(); err != nil {
log.Error("connect peer[%s]: %s", peerAddr, err)
return nil, err
}
if tcpConn, ok := transport.Conn().(*net.TCPConn); ok {
// nagle's only applies to client rather than server
tcpConn.SetNoDelay(this.cf.TcpNoDelay)
}
client := rpc.NewFunServantClientFactory(
transportFactory.GetTransport(transport),
thrift.NewTBinaryProtocolFactoryDefault())
return client, nil
}
// thrift internal layer
//
// Server
// Processor (compiler genereated)
// Protocol (JSON/compact/...), what is transmitted
// Transport (TCP/HTTP/...), how is transmitted
func (this *Engine) launchRpcServe() (done chan null.NullStruct) {
var (
protocolFactory thrift.TProtocolFactory
serverTransport thrift.TServerTransport
transportFactory thrift.TTransportFactory
err error
serverNetwork string
)
switch config.Engine.Rpc.Protocol {
case "binary":
protocolFactory = thrift.NewTBinaryProtocolFactoryDefault()
case "json":
protocolFactory = thrift.NewTJSONProtocolFactory()
case "simplejson":
protocolFactory = thrift.NewTSimpleJSONProtocolFactory()
case "compact":
protocolFactory = thrift.NewTCompactProtocolFactory()
default:
panic(fmt.Sprintf("Unknown protocol: %s", config.Engine.Rpc.Protocol))
}
// client-side Thrift protocol/transport stack must match
// the server-side, otherwise you are very likely to get in trouble
switch {
case config.Engine.Rpc.Framed:
// each payload is sent over the wire with a frame header containing its size
transportFactory = thrift.NewTFramedTransportFactory(transportFactory)
default:
// there is no BufferedTransport in Java: only FramedTransport
transportFactory = thrift.NewTBufferedTransportFactory(
config.Engine.Rpc.BufferSize)
}
switch {
case strings.Contains(config.Engine.Rpc.ListenAddr, "/"):
serverNetwork = "unix"
if config.Engine.Rpc.SessionTimeout > 0 {
serverTransport, err = NewTUnixSocketTimeout(
config.Engine.Rpc.ListenAddr, config.Engine.Rpc.SessionTimeout)
} else {
serverTransport, err = NewTUnixSocket(
config.Engine.Rpc.ListenAddr)
}
default:
serverNetwork = "tcp"
if config.Engine.Rpc.SessionTimeout > 0 {
serverTransport, err = thrift.NewTServerSocketTimeout(
config.Engine.Rpc.ListenAddr, config.Engine.Rpc.SessionTimeout)
} else {
serverTransport, err = thrift.NewTServerSocket(
config.Engine.Rpc.ListenAddr)
}
}
if err != nil {
panic(err)
}
// dial zk before startup servants
// because proxy servant is dependent upon zk
if config.Engine.EtcdSelfAddr != "" {
if err := etclib.Dial(config.Engine.EtcdServers); err != nil {
panic(err)
} else {
log.Debug("etcd connected: %+v", config.Engine.EtcdServers)
}
}
// when config loaded, create the servants
this.svt = servant.NewFunServantWrapper(config.Engine.Servants)
this.rpcProcessor = rpc.NewFunServantProcessor(this.svt)
this.svt.Start()
this.rpcServer = NewTFunServer(this,
config.Engine.Rpc.PreforkMode,
this.rpcProcessor,
serverTransport, transportFactory, protocolFactory)
log.Info("RPC server ready at %s:%s", serverNetwork, config.Engine.Rpc.ListenAddr)
this.launchDashboard()
done = make(chan null.NullStruct)
go func() {
if err = this.rpcServer.Serve(); err != nil {
log.Error("RPC server: %+v", err)
}
done <- null.Null
}()
return done
}