148 lines
3.1 KiB
Go
148 lines
3.1 KiB
Go
// Copyright 2014 beego Author. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package orm
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"reflect"
|
|
)
|
|
|
|
// single model info
|
|
type modelInfo struct {
|
|
pkg string
|
|
name string
|
|
fullName string
|
|
table string
|
|
model interface{}
|
|
fields *fields
|
|
manual bool
|
|
addrField reflect.Value
|
|
uniques []string
|
|
isThrough bool
|
|
}
|
|
|
|
// new model info
|
|
func newModelInfo(val reflect.Value) (info *modelInfo) {
|
|
var (
|
|
err error
|
|
fi *fieldInfo
|
|
sf reflect.StructField
|
|
)
|
|
|
|
info = &modelInfo{}
|
|
info.fields = newFields()
|
|
|
|
ind := reflect.Indirect(val)
|
|
typ := ind.Type()
|
|
|
|
info.addrField = val
|
|
|
|
info.name = typ.Name()
|
|
info.fullName = getFullName(typ)
|
|
|
|
for i := 0; i < ind.NumField(); i++ {
|
|
field := ind.Field(i)
|
|
sf = ind.Type().Field(i)
|
|
if sf.PkgPath != "" {
|
|
continue
|
|
}
|
|
fi, err = newFieldInfo(info, field, sf)
|
|
|
|
if err != nil {
|
|
if err == errSkipField {
|
|
err = nil
|
|
continue
|
|
}
|
|
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 = fi
|
|
}
|
|
}
|
|
|
|
fi.fieldIndex = i
|
|
fi.mi = info
|
|
fi.inModel = true
|
|
}
|
|
|
|
if err != nil {
|
|
fmt.Println(fmt.Errorf("field: %s.%s, %s", ind.Type(), sf.Name, err))
|
|
os.Exit(2)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// combine related model info to new model info.
|
|
// prepare for relation models query.
|
|
func newM2MModelInfo(m1, m2 *modelInfo) (info *modelInfo) {
|
|
info = new(modelInfo)
|
|
info.fields = newFields()
|
|
info.table = m1.table + "_" + m2.table + "s"
|
|
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
|
|
fa.name = "Id"
|
|
fa.column = "id"
|
|
fa.fullName = info.fullName + "." + fa.name
|
|
|
|
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 = fa
|
|
|
|
info.uniques = []string{f1.column, f2.column}
|
|
return
|
|
}
|