File: ensure_slice.go

package info (click to toggle)
golang-github-cheekybits-genny 1.0.0-8
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye, forky, sid, trixie
  • size: 336 kB
  • sloc: makefile: 17; sh: 3
file content (24 lines) | stat: -rw-r--r-- 471 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
package slice

import (
	"log"
	"reflect"

	"github.com/cheekybits/genny/generic"
)

type MyType generic.Type

func EnsureSlice(objectOrSlice interface{}) []MyType {
	log.Printf("%v", reflect.TypeOf(objectOrSlice))
	switch obj := objectOrSlice.(type) {
	case []MyType:
		log.Println("  returning it untouched")
		return obj
	case MyType, *MyType:
		log.Println("  wrapping in slice")
		return []MyType{obj}
	default:
		panic("ensure slice needs MyType or []MyType")
	}
}