mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
package serialization
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
commonpb "go.temporal.io/api/common/v1"
|
|
enumspb "go.temporal.io/api/enums/v1"
|
|
"github.com/hanzoai/tasks/common/codec"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
// SerializerDataEncodingEnvVar controls which codec is used for encoding DataBlobs.
|
|
//
|
|
// Currently supported values (case-insensitive):
|
|
// - "json"
|
|
// - "proto3"
|
|
//
|
|
// Decoding always support all encodings regardless of this setting.
|
|
//
|
|
// WARNING: This environment variable should only be used for testing; and never set it in production.
|
|
const SerializerDataEncodingEnvVar = "TEMPORAL_TEST_DATA_ENCODING"
|
|
|
|
// EncodingTypeFromEnv returns an EncodingType based on the environment variable `TEMPORAL_TEST_DATA_ENCODING`.
|
|
// It defaults to "ENCODING_TYPE_PROTO3" codec if the environment variable is not set.
|
|
func EncodingTypeFromEnv() enumspb.EncodingType {
|
|
codecType := os.Getenv(SerializerDataEncodingEnvVar)
|
|
switch strings.ToLower(codecType) {
|
|
case "", "proto3":
|
|
return enumspb.ENCODING_TYPE_PROTO3
|
|
case "json":
|
|
return enumspb.ENCODING_TYPE_JSON
|
|
default:
|
|
//nolint:forbidigo // should fail fast and hard if used incorrectly
|
|
panic(fmt.Sprintf("unknown codec %q for environment variable %s", codecType, SerializerDataEncodingEnvVar))
|
|
}
|
|
}
|
|
|
|
// ProtoEncode is kept for backward compatibility.
|
|
func ProtoEncode(m proto.Message) (*commonpb.DataBlob, error) {
|
|
return encodeBlob(m, enumspb.ENCODING_TYPE_PROTO3)
|
|
}
|
|
|
|
func encodeBlob(m proto.Message, encoding enumspb.EncodingType) (*commonpb.DataBlob, error) {
|
|
if m == nil {
|
|
return &commonpb.DataBlob{
|
|
Data: nil,
|
|
EncodingType: encoding,
|
|
}, nil
|
|
}
|
|
|
|
switch encoding {
|
|
case enumspb.ENCODING_TYPE_JSON:
|
|
blob, err := codec.NewJSONPBEncoder().Encode(m)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &commonpb.DataBlob{
|
|
Data: blob,
|
|
EncodingType: enumspb.ENCODING_TYPE_JSON,
|
|
}, nil
|
|
case enumspb.ENCODING_TYPE_PROTO3:
|
|
data, err := proto.Marshal(m)
|
|
if err != nil {
|
|
return nil, NewSerializationError(enumspb.ENCODING_TYPE_PROTO3, err)
|
|
}
|
|
return &commonpb.DataBlob{
|
|
EncodingType: enumspb.ENCODING_TYPE_PROTO3,
|
|
Data: data,
|
|
}, nil
|
|
default:
|
|
return nil, NewUnknownEncodingTypeError(encoding.String(), enumspb.ENCODING_TYPE_JSON, enumspb.ENCODING_TYPE_PROTO3)
|
|
}
|
|
}
|
|
|
|
func Decode(data *commonpb.DataBlob, result proto.Message) error {
|
|
if data == nil {
|
|
return NewDeserializationError(enumspb.ENCODING_TYPE_UNSPECIFIED, errors.New("cannot decode nil"))
|
|
}
|
|
|
|
switch data.EncodingType {
|
|
case enumspb.ENCODING_TYPE_JSON:
|
|
return codec.NewJSONPBEncoder().Decode(data.Data, result)
|
|
case enumspb.ENCODING_TYPE_PROTO3:
|
|
err := proto.Unmarshal(data.Data, result)
|
|
if err != nil {
|
|
return NewDeserializationError(enumspb.ENCODING_TYPE_PROTO3, err)
|
|
}
|
|
return nil
|
|
default:
|
|
return NewUnknownEncodingTypeError(data.EncodingType.String(), enumspb.ENCODING_TYPE_JSON, enumspb.ENCODING_TYPE_PROTO3)
|
|
}
|
|
}
|