Beego/orm/models_info_m.go

package orm

import (
	"errors"
	"fmt"
	"os"
	"reflect"
)

type modelInfo struct {
	pkg       string
	name      string
	fullName  string
	table     string
	model     Modeler
	fields    *fields
	manual    bool
	addrField reflect.Value
}

func newModelInfo(model Modeler) (info *modelInfo) {
	var (
		err error
		fi  *fieldInfo
		sf  reflect.StructField
	)

	info = &modelInfo{}
	info.fields = newFields()

	val := reflect.ValueOf(model)
	ind := reflect.Indirect(val)
	typ := ind.Type()

	info.addrField = ind.Addr()

	info.name = typ.Name()
	info.fullName = typ.PkgPath() + "." + typ.Name()

	for i := 0; i < ind.NumField(); i++ {
		field := ind.Field(i)
		sf = ind.Type().Field(i)
		if field.CanAddr() {
			addr := field.Addr()
			if _, ok := addr.Interface().(*Manager); ok {
				continue
			}
		}
		fi, err = newFieldInfo(info, field, sf)
		if err != nil {
			break
		}
		added := info.fields.Add(fi)
		if added == false {
			err = errors.New(fmt.Sprintf("duplicate column name: %s", fi.column))
			break
		}
		if fi.pk {
			if info.fields.pk != nil {
				err = errors.New(fmt.Sprintf("one model must have one pk field only"))
				break
			} else {
				info.fields.pk.Add(fi)
			}
		}
		if fi.auto {
			info.fields.auto = fi
		}
		fi.fieldIndex = i
		fi.mi = info
	}

	if _, ok := info.fields.pk.Exist(info.fields.auto); info.fields.auto != nil && ok == false {
		err = errors.New(fmt.Sprintf("when auto field exists, you cannot set other pk field"))
		goto end
	}

	if err != nil {
		fmt.Println(fmt.Errorf("field: %s.%s, %s", ind.Type(), sf.Name, err))
		os.Exit(2)
	}

end:
	if err != nil {
		fmt.Println(err)
		os.Exit(2)
	}
	return
}

func newM2MModelInfo(m1, m2 *modelInfo) (info *modelInfo) {
	info = new(modelInfo)
	info.fields = newFields()
	info.table = m1.table + "_" + m2.table + "_rel"
	info.name = camelString(info.table)
	info.fullName = m1.pkg + "." + info.name

	fa := new(fieldInfo)
	f1 := new(fieldInfo)
	f2 := new(fieldInfo)
	fa.fieldType = TypeBigIntegerField
	fa.auto = true
	fa.pk = true
	fa.dbcol = true

	f1.dbcol = true
	f2.dbcol = true
	f1.fieldType = RelForeignKey
	f2.fieldType = RelForeignKey
	f1.name = camelString(m1.table)
	f2.name = camelString(m2.table)
	f1.fullName = info.fullName + "." + f1.name
	f2.fullName = info.fullName + "." + f2.name
	f1.column = m1.table + "_id"
	f2.column = m2.table + "_id"
	f1.rel = true
	f2.rel = true
	f1.relTable = m1.table
	f2.relTable = m2.table
	f1.relModelInfo = m1
	f2.relModelInfo = m2
	f1.mi = info
	f2.mi = info

	info.fields.Add(fa)
	info.fields.Add(f1)
	info.fields.Add(f2)
	info.fields.pk.Add(fa)
	return
}