73 lines
2.1 KiB
Go
73 lines
2.1 KiB
Go
package pbcommon
|
|
|
|
import (
|
|
"github.com/hashicorp/consul/agent/structs"
|
|
"google.golang.org/protobuf/types/known/structpb"
|
|
)
|
|
|
|
// ProtobufTypesStructToMapStringInterface converts a protobuf/structpb.Struct into a
|
|
// map[string]interface{}.
|
|
func ProtobufTypesStructToMapStringInterface(s *structpb.Struct) map[string]interface{} {
|
|
if s == nil {
|
|
return nil
|
|
}
|
|
return s.AsMap()
|
|
}
|
|
|
|
// MapStringInterfaceToProtobufTypesStruct converts a map[string]interface{} into a proto.Struct
|
|
func MapStringInterfaceToProtobufTypesStruct(m map[string]interface{}) *structpb.Struct {
|
|
if len(m) < 1 {
|
|
return nil
|
|
}
|
|
// TODO - handle the error better. It probably requires mog to be able to use alternative method signatures though
|
|
s, _ := structpb.NewStruct(m)
|
|
return s
|
|
}
|
|
|
|
// SliceToPBListValue converts a []interface{} into a proto.ListValue. It's used
|
|
// internally by MapStringInterfaceToProtobufTypesStruct when it encouters slices.
|
|
// TODO (remove usage of this struct in favor of structpb.NewListValue)
|
|
func SliceToPBListValue(s []interface{}) *structpb.ListValue {
|
|
if len(s) < 1 {
|
|
return nil
|
|
}
|
|
// TODO - handle the error better. It probably requires mog to use alt method signatures though
|
|
val, _ := structpb.NewList(s)
|
|
return val
|
|
}
|
|
|
|
// EnvoyExtensionsToStructs takes a protobuf EnvoyExtension argument and converts it to the
|
|
// structs EnvoyExtension
|
|
func EnvoyExtensionsToStructs(args []*EnvoyExtension) []structs.EnvoyExtension {
|
|
o := make([]structs.EnvoyExtension, len(args))
|
|
for i := range args {
|
|
var e structs.EnvoyExtension
|
|
if args[i] != nil {
|
|
e = structs.EnvoyExtension{
|
|
Name: args[i].Name,
|
|
Required: args[i].Required,
|
|
Arguments: ProtobufTypesStructToMapStringInterface(args[i].Arguments),
|
|
}
|
|
}
|
|
|
|
o[i] = e
|
|
}
|
|
|
|
return o
|
|
}
|
|
|
|
// EnvoyExtensionsFromStructs takes a structs EnvoyExtension argument and converts it to the
|
|
// protobuf EnvoyExtension
|
|
func EnvoyExtensionsFromStructs(args []structs.EnvoyExtension) []*EnvoyExtension {
|
|
o := make([]*EnvoyExtension, len(args))
|
|
for i, e := range args {
|
|
o[i] = &EnvoyExtension{
|
|
Name: e.Name,
|
|
Required: e.Required,
|
|
Arguments: MapStringInterfaceToProtobufTypesStruct(e.Arguments),
|
|
}
|
|
}
|
|
|
|
return o
|
|
}
|