2023-06-30 10:04:17 +00:00
|
|
|
package pblite
|
2023-06-30 09:54:08 +00:00
|
|
|
|
|
|
|
/*
|
|
|
|
in protobuf, a message looks like this:
|
|
|
|
|
|
|
|
message SomeMessage {
|
|
|
|
string stringField1 = 1;
|
|
|
|
int64 intField = 6;
|
|
|
|
bytes byteField = 9;
|
|
|
|
}
|
|
|
|
|
|
|
|
but when this function is done serializing this protobuf message into a slice, it should look something like this:
|
|
|
|
|
|
|
|
[
|
|
|
|
"someString",
|
|
|
|
nil,
|
|
|
|
nil,
|
|
|
|
nil,
|
|
|
|
nil,
|
|
|
|
6,
|
|
|
|
nil,
|
|
|
|
nil,
|
|
|
|
"\x9\x91\x942"
|
|
|
|
]
|
|
|
|
|
|
|
|
Any integer should be translated into int64, it doesn't matter if it's defined as int32 in the proto schema.
|
|
|
|
In the finished serialized slice it should be int64.
|
|
|
|
Let's also take in count where there is a message nested inside a message:
|
|
|
|
message SomeMessage {
|
|
|
|
string stringField1 = 1;
|
|
|
|
NestedMessage1 nestedMessage1 = 2;
|
|
|
|
int64 intField = 6;
|
|
|
|
bytes byteField = 9;
|
|
|
|
}
|
|
|
|
|
|
|
|
message NestedMessage1 {
|
|
|
|
string msg1 = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
Then the serialized output would be:
|
|
|
|
[
|
|
|
|
"someString",
|
|
|
|
["msg1FieldValue"],
|
|
|
|
nil,
|
|
|
|
nil,
|
|
|
|
nil,
|
|
|
|
6,
|
|
|
|
nil,
|
|
|
|
nil,
|
|
|
|
"\x9\x91\x942"
|
|
|
|
]
|
|
|
|
This means that any slice inside of the current slice, indicates another message nested inside of it.
|
|
|
|
*/
|
|
|
|
|
|
|
|
import (
|
2023-06-30 10:04:17 +00:00
|
|
|
"encoding/base64"
|
|
|
|
"fmt"
|
|
|
|
|
2023-06-30 09:54:08 +00:00
|
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
|
|
)
|
|
|
|
|
2023-06-30 10:04:17 +00:00
|
|
|
func Serialize(m protoreflect.Message) ([]any, error) {
|
2023-06-30 09:55:49 +00:00
|
|
|
maxFieldNumber := 0
|
|
|
|
for i := 0; i < m.Descriptor().Fields().Len(); i++ {
|
|
|
|
fieldNumber := int(m.Descriptor().Fields().Get(i).Number())
|
|
|
|
if fieldNumber > maxFieldNumber {
|
|
|
|
maxFieldNumber = fieldNumber
|
|
|
|
}
|
|
|
|
}
|
2023-06-30 09:54:08 +00:00
|
|
|
|
2023-06-30 10:04:17 +00:00
|
|
|
serialized := make([]any, maxFieldNumber)
|
2023-06-30 09:55:49 +00:00
|
|
|
for i := 0; i < m.Descriptor().Fields().Len(); i++ {
|
|
|
|
fieldDescriptor := m.Descriptor().Fields().Get(i)
|
|
|
|
fieldValue := m.Get(fieldDescriptor)
|
|
|
|
fieldNumber := int(fieldDescriptor.Number())
|
2023-06-30 10:04:17 +00:00
|
|
|
if !m.Has(fieldDescriptor) {
|
|
|
|
continue
|
|
|
|
}
|
2023-06-30 09:55:49 +00:00
|
|
|
switch fieldDescriptor.Kind() {
|
|
|
|
case protoreflect.MessageKind:
|
2023-06-30 10:04:17 +00:00
|
|
|
serializedMsg, err := Serialize(fieldValue.Message().Interface().ProtoReflect())
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
2023-06-30 09:55:49 +00:00
|
|
|
}
|
2023-06-30 10:04:17 +00:00
|
|
|
serialized[fieldNumber-1] = serializedMsg
|
2023-06-30 09:55:49 +00:00
|
|
|
case protoreflect.BytesKind:
|
2023-06-30 10:04:17 +00:00
|
|
|
serialized[fieldNumber-1] = base64.StdEncoding.EncodeToString(fieldValue.Bytes())
|
2023-06-30 09:55:49 +00:00
|
|
|
case protoreflect.Int32Kind, protoreflect.Int64Kind:
|
2023-06-30 10:04:17 +00:00
|
|
|
serialized[fieldNumber-1] = fieldValue.Int()
|
|
|
|
case protoreflect.Uint32Kind, protoreflect.Uint64Kind:
|
|
|
|
serialized[fieldNumber-1] = fieldValue.Uint()
|
|
|
|
case protoreflect.FloatKind, protoreflect.DoubleKind:
|
|
|
|
serialized[fieldNumber-1] = fieldValue.Float()
|
|
|
|
case protoreflect.EnumKind:
|
|
|
|
serialized[fieldNumber-1] = int(fieldValue.Enum())
|
|
|
|
case protoreflect.BoolKind:
|
|
|
|
serialized[fieldNumber-1] = fieldValue.Bool()
|
2023-06-30 09:55:49 +00:00
|
|
|
case protoreflect.StringKind:
|
2023-06-30 10:04:17 +00:00
|
|
|
serialized[fieldNumber-1] = fieldValue.String()
|
2023-06-30 09:55:49 +00:00
|
|
|
default:
|
2023-06-30 10:04:17 +00:00
|
|
|
return nil, fmt.Errorf("unsupported field type %s in %s", fieldDescriptor.Kind(), fieldDescriptor.FullName())
|
2023-06-30 09:55:49 +00:00
|
|
|
}
|
|
|
|
}
|
2023-06-30 09:54:08 +00:00
|
|
|
|
2023-06-30 09:55:49 +00:00
|
|
|
return serialized, nil
|
|
|
|
}
|