fork github.com/doug-martin

2025-03-22 23:02:05 +08:00
commit f14642a736
131 changed files with 34555 additions and 0 deletions
@@ -0,0 +1,135 @@
+package util
+
+import (
+	"reflect"
+	"sort"
+	"strings"
+
+	"git.fsdpf.net/go/db/v2/internal/tag"
+)
+
+type (
+	ColumnData struct {
+		ColumnName     string
+		FieldIndex     []int
+		ShouldInsert   bool
+		ShouldUpdate   bool
+		DefaultIfEmpty bool
+		OmitNil        bool
+		OmitEmpty      bool
+		GoType         reflect.Type
+	}
+	ColumnMap map[string]ColumnData
+)
+
+func newColumnMap(t reflect.Type, fieldIndex []int, prefixes []string) ColumnMap {
+	cm, n := ColumnMap{}, t.NumField()
+	var subColMaps []ColumnMap
+	for i := 0; i < n; i++ {
+		f := t.Field(i)
+		if f.Anonymous && (f.Type.Kind() == reflect.Struct || f.Type.Kind() == reflect.Ptr) {
+			dbTag := tag.New("db", f.Tag)
+			if !dbTag.Contains("-") {
+				subColMaps = append(subColMaps, getStructColumnMap(&f, fieldIndex, dbTag.Values(), prefixes))
+			}
+		} else if f.PkgPath == "" {
+			dbTag := tag.New("db", f.Tag)
+			// if PkgPath is empty then it is an exported field
+			columnName := getColumnName(&f, dbTag)
+			if !shouldIgnoreField(dbTag) {
+				if !implementsScanner(f.Type) {
+					subCm := getStructColumnMap(&f, fieldIndex, []string{columnName}, prefixes)
+					if len(subCm) != 0 {
+						subColMaps = append(subColMaps, subCm)
+						continue
+					}
+				}
+				ffTag := tag.New("ff", f.Tag)
+				columnName = strings.Join(append(prefixes, columnName), ".")
+				cm[columnName] = newColumnData(&f, columnName, fieldIndex, ffTag)
+			}
+		}
+	}
+	return cm.Merge(subColMaps)
+}
+
+func (cm ColumnMap) Cols() []string {
+	structCols := make([]string, 0, len(cm))
+	for key := range cm {
+		structCols = append(structCols, key)
+	}
+	sort.Strings(structCols)
+	return structCols
+}
+
+func (cm ColumnMap) Merge(colMaps []ColumnMap) ColumnMap {
+	for _, subCm := range colMaps {
+		for key, val := range subCm {
+			if _, ok := cm[key]; !ok {
+				cm[key] = val
+			}
+		}
+	}
+	return cm
+}
+
+func implementsScanner(t reflect.Type) bool {
+	if IsPointer(t.Kind()) {
+		t = t.Elem()
+	}
+	if reflect.PtrTo(t).Implements(scannerType) {
+		return true
+	}
+	if !IsStruct(t.Kind()) {
+		return true
+	}
+
+	return false
+}
+
+func newColumnData(f *reflect.StructField, columnName string, fieldIndex []int, ffTag tag.Options) ColumnData {
+	return ColumnData{
+		ColumnName:     columnName,
+		ShouldInsert:   !ffTag.Contains(skipInsertTagName),
+		ShouldUpdate:   !ffTag.Contains(skipUpdateTagName),
+		DefaultIfEmpty: ffTag.Contains(defaultIfEmptyTagName),
+		OmitNil:        ffTag.Contains(omitNilTagName),
+		OmitEmpty:      ffTag.Contains(omitEmptyTagName),
+		FieldIndex:     concatFieldIndexes(fieldIndex, f.Index),
+		GoType:         f.Type,
+	}
+}
+
+func getStructColumnMap(f *reflect.StructField, fieldIndex []int, fieldNames, prefixes []string) ColumnMap {
+	subFieldIndexes := concatFieldIndexes(fieldIndex, f.Index)
+	subPrefixes := prefixes
+	subPrefixes = append(subPrefixes, fieldNames...)
+	if f.Type.Kind() == reflect.Ptr {
+		return newColumnMap(f.Type.Elem(), subFieldIndexes, subPrefixes)
+	}
+	return newColumnMap(f.Type, subFieldIndexes, subPrefixes)
+}
+
+func getColumnName(f *reflect.StructField, dbTag tag.Options) string {
+	if dbTag.IsEmpty() {
+		return columnRenameFunction(f.Name)
+	}
+	return dbTag.Values()[0]
+}
+
+func shouldIgnoreField(dbTag tag.Options) bool {
+	if dbTag.Equals("-") {
+		return true
+	} else if dbTag.IsEmpty() && ignoreUntaggedFields {
+		return true
+	}
+
+	return false
+}
+
+// safely concat two fieldIndex slices into one.
+func concatFieldIndexes(fieldIndexPath, fieldIndex []int) []int {
+	fieldIndexes := make([]int, 0, len(fieldIndexPath)+len(fieldIndex))
+	fieldIndexes = append(fieldIndexes, fieldIndexPath...)
+	return append(fieldIndexes, fieldIndex...)
+}