Golang RPC 之 Thrift
2017-03-16 本文已影响3412人
谢烟客
Thrift 简介:
Thrift 是一款高性能、开源的 RPC 框架,产自 Facebook 后贡献给了 Apache,Thrift 囊括了整个 RPC 的上下游体系,自带序列化编译工具,因为 Thrift 采用的是二进制序列化,并且与 gRPC 一样使用的都是长连接建立 client 与 server 之间的通讯,相比于比传统的使用XML,JSON,SOAP等短连接的解决方案性能要快得多。
本篇只介绍 Golang 关于 Thrift 的基础使用。
安装
- 安装 Thrift 的 Golang 库有两种方案:
- 直接通过 go get 命令安装,缺点是因为不可抗拒的网络因素大部分人可能会失败:$ go get git.apache.org/thrift.git/lib/go/thrift
- 通过源码安装:
- 在 $GOPATH 的 src 目录下创建多层级目录:git.apache.org/thrift.git/lib/go
- 从 github 上下载 thrift 0.10.0 的源码 ,解压后进入 thrift-0.10.0/lib/go 目录下,将 thrift 目录 copy 到刚创建的 $GOPATH/src/git.apache.org/thrift.git/lib/go 目录下
- 在任意目录下执行 $ go install git.apache.org/thrift.git/lib/go/thrift 就完成了 golang 的 thrift 库的安装
- 安装 Thrift 的 IDL 编译工具
- windows 平台下安装:
直接下载:thrift complier 下载地址,下载完成后改名为:thrift.exe 并将其放入到系统环境变量下即可使用
- Linux 平台下安装:
从 github 上下载 thrift 0.10.0 的源码,解压后进入:thrift-0.10.0/compiler/cpp 目录执行如下命令完成编译后,将其放入到系统环境变量下即可使用:
$ mkdir cmake-build
$ cd cmake-build
$ cmake ..
$ make
- 验证是否安装成功:
$ thrift -version,如果打印出来:Thrift version 0.10.0 表明 complier 安装成功
实践:
下面我们使用 Thrift 定义一个接口,该接口实现对传入的数据进行大写的格式化处理。
- 创建 golang 项目 ThriftDemo 工程:
- client目录下的 client.go 实现了客户端用于发送数据并打印接收到 server 端处理后的数据
- server 目录下的 server.go 实现了服务端用于接收客户端发送的数据,并对数据进行大写处理后返回给客户端
- thrift_file 用于存放 thrift 的 IDL 文件: *.thrift
- 定义 Thrift RPC 接口
example.thrift:
namespace py example
struct Data {
1: string text
}
service format_data {
Data do_format(1:Data data),
}
- 编译 thrift 文件
Paste_Image.png进入 thrift_file 目录执行:$ thrift -out .. --gen go example.thrift,就会在 thrift_file 的同级目录下生成 golang 的包:example,其中 format_data-remote 是生成的测试代码可以不用特别关注
- 实现 server 端:
package main
import (
"ThriftDemo/example"
"strings"
"git.apache.org/thrift.git/lib/go/thrift"
"fmt"
"log"
)
type FormatDataImpl struct {}
func (fdi *FormatDataImpl) DoFormat(data *example.Data) (r *example.Data, err error){
var rData example.Data
rData.Text = strings.ToUpper(data.Text)
return &rData, nil
}
const (
HOST = "localhost"
PORT = "8080"
)
func main() {
handler := &FormatDataImpl{}
processor := example.NewFormatDataProcessor(handler)
serverTransport, err := thrift.NewTServerSocket(HOST + ":" + PORT)
if err != nil {
log.Fatalln("Error:", err)
}
transportFactory := thrift.NewTFramedTransportFactory(thrift.NewTTransportFactory())
protocolFactory := thrift.NewTBinaryProtocolFactoryDefault()
server := thrift.NewTSimpleServer4(processor, serverTransport, transportFactory, protocolFactory)
fmt.Println("Running at:", HOST + ":" + PORT)
server.Serve()
}
- 实现 client 端
package main
import (
"git.apache.org/thrift.git/lib/go/thrift"
"net"
"fmt"
"ThriftDemo/example"
"log"
)
const (
HOST = "localhost"
PORT = "8080"
)
func main() {
tSocket, err := thrift.NewTSocket(net.JoinHostPort(HOST, PORT))
if err != nil {
log.Fatalln("tSocket error:", err)
}
transportFactory := thrift.NewTFramedTransportFactory(thrift.NewTTransportFactory())
transport := transportFactory.GetTransport(tSocket)
protocolFactory := thrift.NewTBinaryProtocolFactoryDefault()
client := example.NewFormatDataClientFactory(transport, protocolFactory)
if err := transport.Open(); err != nil {
log.Fatalln("Error opening:", HOST + ":" + PORT)
}
defer transport.Close()
data := example.Data{Text:"hello,world!"}
d, err := client.DoFormat(&data)
fmt.Println(d.Text)
}
- 执行验证结果:
- 先启动 server,之后再执行 client
- client 侧控制台如果打印的结果为: HELLO,WORLD! ,证明 Thrift 的 RPC 接口定义成功