thrift 最初是 facebook 開發使用的 rpc 通信框架，后來貢獻給了 apache 基金會，出來得比較早，幾乎支持所有的后端語言，使用非常廣泛，是不可不知的一個網絡框架
和 grpc 一樣，需要先定義通信協議，然后實現自己業務邏輯，下面還是通過一個簡單示例（之前的echo程序）說明 thrift 的用法，下面示例使用的完整代碼在下列地址：
實現文件：https://github.com/hatlonely/hellogolang/tree/master/cmd/thrift
協議文件：https://github.com/hatlonely/hellogolang/tree/master/api/echo_thrift
簡單 echo 服務
獲取 thrift
go get git.apache.org/thrift.git/lib/go
定義協議文件
namespace go echo
struct EchoReq {
1: string msg;
}
struct EchoRes {
1: string msg;
}
service Echo {
EchoRes echo(1: EchoReq req);
}
執行 thrift -r --gen go echo.thrift 命令會生成 gen-go 文件夾，這個過程其實是將上面的協議翻譯成 golang 代碼
這個命令依賴于 thrift 工具，可以通過下面命令獲取
Mac
brew install thrift
Linux
wget http://www-us.apache.org/dist/thrift/0.11.0/thrift-0.11.0.tar.gz
tar -xzvf thrift-0.11.0.tar.gz
cd thrift-0.11.0
./configure
make -j8
[sudo] make install
實現服務端
type EchoServerImp struct {
}
func (e *EchoServerImp) Echo(ctx context.Context, req *echo.EchoReq) (*echo.EchoRes, error) {
fmt.Printf("message from client: %v\n", req.GetMsg())
res := &echo.EchoRes{
Msg: req.GetMsg(),
}
return res, nil
}
func main() {
transport, err := thrift.NewTServerSocket(":3000")
if err != nil {
panic(err)
}
processor := echo.NewEchoProcessor(&EchoServerImp{})
server := thrift.NewTSimpleServer4(
processor,
transport,
thrift.NewTBufferedTransportFactory(8192),
thrift.NewTCompactProtocolFactory(),
)
if err := server.Serve(); err != nil {
panic(err)
}
}
這個過程和 grpc 類似，不同的地方在于，thrift 支持更多的服務器類型，支持不同的協議打包方式，方便用戶選擇，這里的 compact 協議是一種壓縮的協議，使用比較多
實現客戶端
func main() {
var transport thrift.TTransport
var err error
transport, err = thrift.NewTSocket("localhost:3000")
if err != nil {
fmt.Errorf("NewTSocket failed. err: [%v]\n", err)
return
}
transport, err = thrift.NewTBufferedTransportFactory(8192).GetTransport(transport)
if err != nil {
fmt.Errorf("NewTransport failed. err: [%v]\n", err)
return
}
defer transport.Close()
if err := transport.Open(); err != nil {
fmt.Errorf("Transport.Open failed. err: [%v]\n", err)
return
}
protocolFactory := thrift.NewTCompactProtocolFactory()
iprot := protocolFactory.GetProtocol(transport)
oprot := protocolFactory.GetProtocol(transport)
client := echo.NewEchoClient(thrift.NewTStandardClient(iprot, oprot))
var res *echo.EchoRes
res, err = client.Echo(context.Background(), &echo.EchoReq{
Msg: strings.Join(os.Args[1:], " "),
})
if err != nil {
fmt.Errorf("client echo failed. err: [%v]", err)
return
}
fmt.Printf("message from server: %v", res.GetMsg())
}
這個 client 相對復雜一些，需要和 server 端設置一致的打包方式，如果不一致會出現通信失敗，這一點需要特別注意一下
參考鏈接
- thrift go 官網: http://thrift.apache.org/tutorial/go
- thrift github: https://github.com/apache/thrift/
- thrift go tutorial: https://github.com/apache/thrift/tree/master/tutorial/go/src
轉載請注明出處
本文鏈接：http://hatlonely.github.io/2018/02/04/golang-%E7%BD%91%E7%BB%9C%E6%A1%86%E6%9E%B6%E4%B9%8B-thrift/
