func (this *Engine) launchRpcServe() (done chan interface{}) {
var protocolFactory thrift.TProtocolFactory
switch this.conf.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("Invalid protocol: %s", this.conf.rpc.protocol))
}
transportFactory := thrift.NewTTransportFactory()
if this.conf.rpc.framed {
transportFactory = thrift.NewTFramedTransportFactory(transportFactory)
}
serverTransport, err := thrift.NewTServerSocketTimeout(this.conf.rpc.listenAddr,
this.conf.rpc.clientTimeout)
if err != nil {
panic(err)
}
this.rpcServer = thrift.NewTSimpleServer4(this.rpcProcessor,
serverTransport, transportFactory, protocolFactory)
log.Info("RPC server ready at %s", this.conf.rpc.listenAddr)
done = make(chan interface{})
go func() {
for {
err = this.rpcServer.Serve()
if err != nil {
log.Error(err)
break
}
}
done <- 1
}()
return done
}
func startRpcServe(port string) {
socket, err := thrift.NewTServerSocketTimeout(fmt.Sprintf(":%s", port), TIMEOUT)
if err != nil {
log.Fatalln("Unable to create server socket", err)
}
protocol := thrift.NewTBinaryProtocolFactoryDefault()
transport := thrift.NewTFramedTransportFactory(thrift.NewTTransportFactory())
processor := thrift.NewTMultiplexedProcessor()
registerProcessors(processor)
server := thrift.NewTSimpleServer4(processor, socket, transport, protocol)
server.Serve()
}
func (this *Engine) launchRpcServe() (done chan interface{}) {
var (
protocolFactory thrift.TProtocolFactory
serverTransport thrift.TServerTransport
transportFactory thrift.TTransportFactory
err error
serverNetwork string
)
switch this.conf.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("Invalid protocol: %s", this.conf.rpc.protocol))
}
switch {
case this.conf.rpc.framed:
transportFactory = thrift.NewTFramedTransportFactory(transportFactory)
default:
transportFactory = thrift.NewTTransportFactory()
}
switch {
case strings.Contains(this.conf.rpc.listenAddr, "/"):
serverNetwork = "unix"
serverTransport, err = NewTUnixSocketTimeout(
this.conf.rpc.listenAddr, this.conf.rpc.clientTimeout)
default:
serverNetwork = "tcp"
serverTransport, err = thrift.NewTServerSocketTimeout(
this.conf.rpc.listenAddr, this.conf.rpc.clientTimeout)
}
if err != nil {
panic(err)
}
this.rpcServer = NewTFunServer(this, this.rpcProcessor,
serverTransport, transportFactory, protocolFactory)
log.Info("RPC server ready at %s:%s", serverNetwork, this.conf.rpc.listenAddr)
done = make(chan interface{})
go func() {
for {
err = this.rpcServer.Serve()
if err != nil {
log.Error(err)
break
}
}
done <- 1
}()
return done
}