GOLANG使用THRIFT的鉴权和多路复用

作者：袖梨 2022-06-29

最近梳理公司的服务时，决定把基础组件抽离出来服务化。因为公司里有多种语言开发的服务，决定使用thrift来解决跨语言通讯。

启动一个server

thrift对golang的支持已经很好了，官方包里提供了一个server基类和一个简单服务simple_server.go.自己可以实现server来做鉴权和其它统计操作.
server定义了一个TServer接口和一些方法,源码如下:

type TServer interface {
    ProcessorFactory() TProcessorFactory          //服务
    ServerTransport() TServerTransport            //连接
    InputTransportFactory() TTransportFactory     //输入
    OutputTransportFactory() TTransportFactory    //输出
    InputProtocolFactory() TProtocolFactory       //协议
    OutputProtocolFactory() TProtocolFactory      //协议

    // Starts the server
    Serve() error
    // Stops the server. This is optional on a per-implementation basis. Not
    // all servers are required to be cleanly stoppable.
    Stop() error
}
Server方法进行监听,接受请求,调度处理.下面是simple_server.go的Server方法.

func (p *TSimpleServer) Serve() error {
    err := p.Listen()
    if err != nil {
        return err
    }
    p.AcceptLoop()
    return nil
}
func (p *TSimpleServer) AcceptLoop() error {
    for {
        client, err := p.serverTransport.Accept()
        if err != nil {
            select {
            case <-p.quit:
                return nil
            default:
            }
            return err
        }
        if client != nil {
            go func() {
                if err := p.processRequests(client); err != nil {
                    log.Println("error processing request:", err)
                }
            }()
        }
    }
}
只要自己实现相关接口方法,在方法中就可以进行相关鉴权/统计等处理了。