示例#1
0
func getFields(typ reflect.Type) fields {
	num := typ.NumField()
	fs := newFields(num)

	for i := 0; i < num; i++ {
		f := typ.Field(i)

		if f.Anonymous {
			typ := f.Type
			if typ.Kind() == reflect.Ptr {
				typ = typ.Elem()
			}
			for _, ff := range getFields(typ).List {
				ff.index = append(f.Index, ff.index...)
				fs.Add(ff)
			}
			continue
		}

		if f.PkgPath != "" && !f.Anonymous {
			continue
		}

		name, opts := parseTag(f.Tag.Get("pg"))
		if name == "-" {
			continue
		}
		if name == "" {
			name = pgutil.Underscore(f.Name)
		}

		fieldType := indirectType(f.Type)
		if fieldType.Kind() == reflect.Struct {
			for _, ff := range getFields(fieldType).List {
				ff.PGName = name + "." + ff.Name
				ff.Name = name + "__" + ff.Name
				ff.index = append(f.Index, ff.index...)
				fs.Add(ff)
			}
		}

		var flags int8
		if opts.Contains("nullempty") {
			flags |= nullEmpty
		}

		field := &field{
			Name:  name,
			index: f.Index,
			flags: flags,

			appender: getAppender(fieldType),
			decoder:  getDecoder(fieldType),
		}
		fs.Add(field)
	}

	return fs
}
示例#2
0
文件: types.go 项目: koheeba/pg
func fields(typ reflect.Type) map[string]*pgValue {
	num := typ.NumField()
	dst := make(map[string]*pgValue, num)
	for i := 0; i < num; i++ {
		f := typ.Field(i)

		if f.Anonymous {
			typ := f.Type
			if typ.Kind() == reflect.Ptr {
				typ = typ.Elem()
			}
			for name, ff := range fields(typ) {
				dst[name] = newPGValue(ff.Source, append(f.Index, ff.Index...))
			}
			continue
		}

		if f.PkgPath != "" {
			continue
		}

		name, opts := parseTag(f.Tag.Get("pg"))
		if name == "-" {
			continue
		}
		if name == "" {
			name = pgutil.Underscore(f.Name)
		}

		tt := indirectType(f.Type)
		if tt.Kind() == reflect.Struct {
			for subname, ff := range fields(tt) {
				dst[name+"__"+subname] = newPGValue(ff.Source, append(f.Index, ff.Index...))
			}
		}

		val := newPGValue(f, f.Index)
		if opts.Contains("nullempty") {
			val.NullEmpty = true
		}
		dst[name] = val
	}
	return dst
}