Thrift 编码测试： 枚举类型类型是如何被传递的？ 编码和解码过程小探索

​

1 示例：

enum MyBzType {

    Unknown  = 0;   // 未知

    BOYIN   = 1;       // 波音公司

}



type EncDecEnumSample struct {

ID int64 `thrift:"id,1" json:"id"`



StuID int64 `thrift:"stu_id,2" json:"stu_id"`



BzID int64 `thrift:”bz_id,3" json:”bz_id"`



MyBzType enum_type.MyBzType`thrift:”my_bz_type,4" json:"my_bz_type"`



}
复制代码

2 首先解码部分(fast)： 可见传输使用的i32

func (p *EncDecEnumSample) FastReadField4(buf []byte) (int, error) {

offset := 0



if v, l, err := bthrift.Binary.ReadI32(buf[offset:]); err != nil {

return offset, err

} else {

offset += l



p.MyBzType = enum_type.MyBzType(v)



}

return offset, nil

}



func (p *EncDecEnumSample) ReadField4(iprot thrift.TProtocol) error {

if v, err := iprot.ReadI32(); err != nil {

return err

} else {

p.MyBzType = enum_type.MyBzType(v)

}

return nil

}
复制代码

3 编码部分： 可见传输使用的i32

func (p *EncDecEnumSample) fastWriteField4(buf []byte, binaryWriter bthrift.BinaryWriter) int {

offset := 0

offset += bthrift.Binary.WriteFieldBegin(buf[offset:], "my_bz_type", thrift.I32, 4)

offset += bthrift.Binary.WriteI32(buf[offset:], int32(p.MyBzType))



offset += bthrift.Binary.WriteFieldEnd(buf[offset:])

return offset

}





func (p *EncDecEnumSample) writeField4(oprot thrift.TProtocol) (err error) {

if err = oprot.WriteFieldBegin("my_bz_type", thrift.I32, 4); err != nil {

goto WriteFieldBeginError

}

if err := oprot.WriteI32(int32(p.MyBzType)); err != nil {

return err

}

if err = oprot.WriteFieldEnd(); err != nil {

goto WriteFieldEndError

}

return nil

WriteFieldBeginError:

return thrift.PrependError(fmt.Sprintf("%T write field 4 begin error: ", p), err)

WriteFieldEndError:

return thrift.PrependError(fmt.Sprintf("%T write field 4 end error: ", p), err)

}
复制代码

4 结论：

1： thrift enum类型的传值使用i32进行传值的，不会传输tag信息

2： 基于1的结论，如果参数直接传输i32类型的整数也是可以的，只要在枚举范围内

3： 如果api升级i32类型的整数可直接升级到枚举类型，反过来也是可行的。只要范围适当不会有兼容性问题。

​