diff options
Diffstat (limited to 'vendor/github.com/xlgmokha/x/pkg/mapper/mapper.go')
| -rw-r--r-- | vendor/github.com/xlgmokha/x/pkg/mapper/mapper.go | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/vendor/github.com/xlgmokha/x/pkg/mapper/mapper.go b/vendor/github.com/xlgmokha/x/pkg/mapper/mapper.go new file mode 100644 index 00000000..2057f719 --- /dev/null +++ b/vendor/github.com/xlgmokha/x/pkg/mapper/mapper.go @@ -0,0 +1,46 @@ +package mapper + +import ( + "fmt" + "reflect" +) + +type Mapping[TInput any, TOutput any] func(TInput) TOutput + +var mappings map[string]interface{} + +func init() { + mappings = map[string]interface{}{} +} + +func Register[Input any, Output any](mapping Mapping[Input, Output]) { + mappings[keyFor[Input, Output]()] = mapping +} + +func MapFrom[Input any, Output any](input Input) Output { + if mapping, ok := mappings[keyFor[Input, Output]()]; ok { + return mapping.(Mapping[Input, Output])(input) + } + var output Output + return output +} + +func MapEachFrom[Input any, Output any](input []Input) []Output { + var zero Output + zeroValue := reflect.Zero(reflect.TypeOf(zero)) + + results := []Output{} + for _, item := range input { + tmp := MapFrom[Input, Output](item) + if zeroValue != reflect.ValueOf(tmp) { + results = append(results, tmp) + } + } + return results +} + +func keyFor[Input any, Output any]() string { + var input Input + var output Output + return fmt.Sprintf("%v-%v", reflect.TypeOf(input), reflect.TypeOf(output)) +} |
