forked from aws/amazon-ecs-agent
-
Notifications
You must be signed in to change notification settings - Fork 0
/
types.go
59 lines (50 loc) · 2.06 KB
/
types.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
// Copyright 2014-2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may
// not use this file except in compliance with the License. A copy of the
// License is located at
//
// http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file is distributed
// on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
// express or implied. See the License for the specific language governing
// permissions and limitations under the License.
package wsclient
import "reflect"
// TypeDecoder interface defines methods to decode ecs types.
type TypeDecoder interface {
// NewOfType returns an object of a recognized type for a given type name.
// It additionally returns a boolean value which is set to false for an
// unrecognized type.
NewOfType(string) (interface{}, bool)
// GetRecognizedTypes returns a map of type-strings (as passed in acs/tcs messages as
// the 'type' field) to a pointer to the corresponding struct type this type should
// be marshalled/unmarshalled to/from.
GetRecognizedTypes() map[string]reflect.Type
}
// TypeDecoderImpl is an implementation for general use between ACS and
// TCS clients
type TypeDecoderImpl struct {
typeMappings map[string]reflect.Type
}
// BuildTypeDecoder takes a list of interfaces and stores them internally as a
// list of typeMappings in the format below.
// "MyMessage": TypeOf(ecstcs.MyMessage)
func BuildTypeDecoder(recognizedTypes []interface{}) TypeDecoder {
typeMappings := make(map[string]reflect.Type)
for _, recognizedType := range recognizedTypes {
typeMappings[reflect.TypeOf(recognizedType).Name()] = reflect.TypeOf(recognizedType)
}
return &TypeDecoderImpl{typeMappings: typeMappings}
}
func (d *TypeDecoderImpl) NewOfType(typeString string) (interface{}, bool) {
rtype, ok := d.typeMappings[typeString]
if !ok {
return nil, false
}
return reflect.New(rtype).Interface(), true
}
func (d *TypeDecoderImpl) GetRecognizedTypes() map[string]reflect.Type {
return d.typeMappings
}