English | 中文
一种无需生成代码、高性能的动态 Thrift 编解码器。
它实现了纯 Go 版本(或reflect版本)和 即时编译(JIT)版本。 由于在大多数情况下,reflect版本的性能优于 JIT 版本,并且它可以在不同的 CPU 架构上运行, 因此我们计划从 Go 1.24 开始废弃 JIT 版本。
在 Go 1.24 之前:
- JIT 版本是
amd64
的默认编解码器。 - reflect 版本仅在运行在
!amd64
或windows
上时启用 - 您可以通过指定操作系统环境变量
FRUGAL_NO_JIT=1
来启用reflect版本
自 Go 1.24 起:
- reflect 版本成为默认编解码器。
- JIT 代码将在 go build 时跳过不执行
传统的 Thrift 编解码方式,要求用户必须要先生成编解码代码,Frugal 通过反射 struct field tag 动态生成编解码器避免了这一过程。
基于 frugal/tests
的测试用例,Frugal 的性能 比 Apache Thrift (TBinaryProtocol) 好 1 到 4 倍。
不同的测试用例,结果可能会有些差异。欢迎给我们分享你的测试数据。
go version go1.22.5 linux/amd64
goos: linux
goarch: amd64
pkg: github.com/cloudwego/frugal/tests
cpu: Intel(R) Xeon(R) Gold 5118 CPU @ 2.30GHz
Marshal_ApacheThrift/small-4 2070745 584.0 ns/op 998.32 MB/s 112 B/op 1 allocs/op
Marshal_ApacheThrift/medium-4 78729 13680 ns/op 1280.57 MB/s 112 B/op 1 allocs/op
Marshal_ApacheThrift/large-4 3097 376184 ns/op 1179.75 MB/s 620 B/op 1 allocs/op
Marshal_Frugal_JIT/small-4 4939591 242.1 ns/op 2407.83 MB/s 13 B/op 0 allocs/op
Marshal_Frugal_JIT/medium-4 160820 7485 ns/op 2340.29 MB/s 54 B/op 0 allocs/op
Marshal_Frugal_JIT/large-4 5370 214258 ns/op 2071.35 MB/s 338 B/op 0 allocs/op
Marshal_Frugal_Reflect/small-4 10171197 117.3 ns/op 4970.90 MB/s 0 B/op 0 allocs/op
Marshal_Frugal_Reflect/medium-4 180207 6644 ns/op 2636.73 MB/s 0 B/op 0 allocs/op
Marshal_Frugal_Reflect/large-4 6312 185534 ns/op 2392.04 MB/s 0 B/op 0 allocs/op
Unmarshal_ApacheThrift/small-4 768525 1443 ns/op 403.94 MB/s 1232 B/op 5 allocs/op
Unmarshal_ApacheThrift/medium-4 24463 47067 ns/op 372.19 MB/s 44816 B/op 176 allocs/op
Unmarshal_ApacheThrift/large-4 1053 1155725 ns/op 384.00 MB/s 1135540 B/op 4433 allocs/op
Unmarshal_Frugal_JIT/small-4 2575767 466.3 ns/op 1250.36 MB/s 547 B/op 2 allocs/op
Unmarshal_Frugal_JIT/medium-4 62128 19333 ns/op 906.12 MB/s 19404 B/op 89 allocs/op
Unmarshal_Frugal_JIT/large-4 2328 496431 ns/op 893.99 MB/s 495906 B/op 2283 allocs/op
Unmarshal_Frugal_Reflect/small-4 2770252 437.2 ns/op 1333.60 MB/s 544 B/op 1 allocs/op
Unmarshal_Frugal_Reflect/medium-4 64232 18183 ns/op 963.45 MB/s 19945 B/op 57 allocs/op
Unmarshal_Frugal_Reflect/large-4 2325 496415 ns/op 894.02 MB/s 511876 B/op 1467 allocs/op
使用 Frugal 作为 Kitex 的编解码
可以不用再生成大量的编解码代码,使仓库变得干净整洁,review 时也不用再带上一堆无意义的 diff。然后相比于生成的编解码代码,Frugal 的性能更高。
在 Thriftgo 生成的 struct 上进行编解码
如果你只需要使用 Thrift 的编解码能力,同时也定义好了 IDL,那么只需要用 Thriftgo 生成 IDL 对应的 Go 语言 struct,就可以使用 Frugal 的编解码能力了。
如果你们连 IDL 都不想有,没问题,直接定义好 Go 语言 struct 后,给每个 Field 带上 Frugal 所需的 tag,就可以直接使用 Frugal 进行编解码了。
go get github.com/cloudwego/kitex@latest
示例:
kitex -thrift frugal_tag -service a.b.c my.thrift
如果不需要编解码代码,可以带上 -thrift template=slim
参数
kitex -thrift frugal_tag,template=slim -service a.b.c my.thrift
client 示例:
package client
import (
"context"
"example.com/kitex_test/client/kitex_gen/a/b/c/echo"
"github.com/cloudwego/kitex/client"
"github.com/cloudwego/kitex/pkg/remote/codec/thrift"
)
func Echo() {
code := thrift.NewThriftCodecWithConfig(thrift.FastRead | thrift.FastWrite | thrift.FrugalRead | thrift.FrugalWrite)
cli := echo.MustNewClient("a.b.c", client.WithPayloadCodec(codec))
...
}
server 示例:
package main
import (
"log"
"github.com/cloudwego/kitex/server"
c "example.com/kitex_test/kitex_gen/a/b/c/echo"
"github.com/cloudwego/kitex/pkg/remote/codec/thrift"
)
func main() {
code := thrift.NewThriftCodecWithConfig(thrift.FastRead | thrift.FastWrite | thrift.FrugalRead | thrift.FrugalWrite)
svr := c.NewServer(new(EchoImpl), server.WithPayloadCodec(code))
err := svr.Run()
if err != nil {
log.Println(err.Error())
}
}
现在假设我们有如下 Thrift 文件:
my.thrift:
struct MyStruct {
1: string msg
2: i64 code
}
定义好需要的 Thrift 文件后,在使用 Thriftgo 生成 Go 语言代码时使用 frugal_tag
参数。
示例:
thriftgo -r -o thrift -g go:frugal_tag,package_prefix=example.com/kitex_test/thrift my.thrift
如果不需要编解码代码,可以带上 template=slim
参数
thriftgo -r -o thrift -g go:frugal_tag,template=slim,package_prefix=example.com/kitex_test/thrift my.thrift
生成所需要的结构体后,直接使用 Frugal 进行编解码即可。
示例:
package main
import (
"github.com/cloudwego/frugal"
"example.com/kitex_test/thrift"
)
func main() {
ms := &thrift.MyStruct{
Msg: "my message",
Code: 1024,
}
...
buf := make([]byte, frugal.EncodedSize(ms))
frugal.EncodeObject(buf, nil, ms)
...
got := &thrift.MyStruct{}
frugal.DecodeObject(buf, got)
...
}
现在假设我们需要如下 struct:
type MyStruct struct {
Msg string
Code int64
Numbers []int64
}
Frugal 所需的 tag 形如 frugal:"1,default,string"
,其中 1
为字段 ID, default
为字段的 requiredness, string
表示字段的类型。字段 ID 和 字段 requiredness 是必须的,但是字段类型只有当字段为 list
、set
和 enum
时是必须的。
上述的 MyStruct
可以添加如下 tag:
type MyStruct struct {
Msg string `frugal:"1,default"`
Code int64 `frugal:"2,default"`
Numbers []int64 `frugal:"3,default,list<i64>"`
}
下面是完整的类型示例:
type MyEnum int64
type Example struct {
MyOptBool *bool `frugal:"1,optional"`
MyReqBool bool `frugal:"2,required"`
MyOptByte *int8 `frugal:"3,optional"`
MyReqByte int8 `frugal:"4,required"`
MyOptI16 *int16 `frugal:"5,optional"`
MyReqI16 int16 `frugal:"6,required"`
MyOptI32 *int32 `frugal:"7,optional"`
MyReqI32 int32 `frugal:"8,required"`
MyOptI64 *int64 `frugal:"9,optional"`
MyReqI64 int64 `frugal:"10,required"`
MyOptString *string `frugal:"11,optional"`
MyReqString string `frugal:"12,required"`
MyOptBinary []byte `frugal:"13,optional"`
MyReqBinary []byte `frugal:"14,required"`
MyOptI64Set []int64 `frugal:"15,optional,set<i64>"`
MyReqI64Set []int64 `frugal:"16,required,set<i64>"`
MyOptI64List []int64 `frugal:"17,optional,list<i64>"`
MyReqI64List []int64 `frugal:"18,required,list<i64>"`
MyOptI64StringMap map[int64]string `frugal:"19,optional"`
MyReqI64StringMap map[int64]string `frugal:"20,required"`
MyOptEnum *MyEnum `frugal:"21,optional,i64"`
MyReqEnum *MyEnum `frugal:"22,optional,i64"`
}
直接使用 Frugal 进行编解码即可。
示例:
package main
import (
"github.com/cloudwego/frugal"
)
func main() {
ms := &thrift.MyStruct{
Msg: "my message",
Code: 1024,
Numbers: []int64{0, 1, 2, 3, 4},
}
...
buf := make([]byte, frugal.EncodedSize(ms))
frugal.EncodeObject(buf, nil, ms)
...
got := &thrift.MyStruct{}
frugal.DecodeObject(buf, got)
...
}