dbmap/dbmap.go at master · AlexanderMatveev/dbmap · GitHub

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
package dbmap

import (
	"reflect"
)

const tagName = "db"

func Columns(t reflect.Type) []string {
	var vs []string
	for i := 0; i < t.NumField(); i++ {
		tag := t.Field(i).Tag.Get(tagName)
		if tag == "" {
			continue
		}
		vs = append(vs, tag)
	}
	return vs
}

func Values(s any) []any {
	v := reflect.ValueOf(s)
	t := reflect.TypeOf(s)
	var vs []any
	for i := 0; i < v.NumField(); i++ {
		if t.Field(i).Tag.Get(tagName) == "" {
			continue
		}
		vs = append(vs, v.Field(i).Interface())
	}
	return vs
}

func Scan(s any) []any {
	v := reflect.ValueOf(s).Elem()
	t := reflect.TypeOf(s).Elem()
	var vs []any
	for i := 0; i < v.NumField(); i++ {
		if t.Field(i).Tag.Get(tagName) == "" {
			continue
		}
		vs = append(vs, v.Field(i).Addr().Interface())
	}
	return vs
}