overflow_gateway_websocket/constants.go

package overflow_gateway_websocket

import (
	"encoding/json"
	"fmt"

	cuc "git.loafle.net/commons_go/util/context"
)

var (
	ServletSocketKey = cuc.ContextKey("ServletSocket")

	GRPCMetadataKey = cuc.ContextKey("GRPCMetadata")
	ClientTypeKey   = cuc.ContextKey("ClientType")
	SocketIDKey     = cuc.ContextKey("SocketID")
	TargetIDKey     = cuc.ContextKey("TargetID")
)

const (
	GRPCClientTypeKey = "OVERFLOW_GRPC_CLIENT_TYPE"
	GRPCSessionIDKey  = "OVERFLOW_GRPC_SESSION_ID"
	GRPCTargetIDKey   = "OVERFLOW_GRPC_TARGET_ID"
)

type ClientType int

const (
	MEMBER ClientType = iota
	PROBE
)

var (
	clientTypeID = map[ClientType]string{
		MEMBER: "MEMBER",
		PROBE:  "PROBE",
	}

	clientTypeName = map[string]ClientType{
		"MEMBER": MEMBER,
		"PROBE":  PROBE,
	}
)

func (ct ClientType) String() string {
	return clientTypeID[ct]
}

func (ct *ClientType) MarshalJSON() ([]byte, error) {
	value, ok := clientTypeID[*ct]
	if !ok {
		return nil, fmt.Errorf("Invalid EnumType[%s] value", *ct)
	}
	return json.Marshal(value)
}

func (ct *ClientType) UnmarshalJSON(b []byte) error {
	// unmarshal as string
	var s string
	err := json.Unmarshal(b, &s)
	if err != nil {
		return err
	}

	value, ok := clientTypeName[s]
	if !ok {
		return fmt.Errorf("Invalid EnumType[%s] value", s)
	}
	*ct = value
	return nil
}