package orm import ( "reflect" ) type queryM2M struct { md interface{} mi *modelInfo fi *fieldInfo qs *querySet ind reflect.Value } func (o *queryM2M) Add(mds ...interface{}) (int64, error) { fi := o.fi mi := fi.relThroughModelInfo mfi := fi.reverseFieldInfo rfi := fi.reverseFieldInfoTwo orm := o.qs.orm dbase := orm.alias.DbBaser var models []interface{} for _, md := range mds { val := reflect.ValueOf(md) if val.Kind() == reflect.Slice || val.Kind() == reflect.Array { for i := 0; i < val.Len(); i++ { v := val.Index(i) if v.CanInterface() { models = append(models, v.Interface()) } } } else { models = append(models, md) } } _, v1, exist := getExistPk(o.mi, o.ind) if exist == false { panic(ErrMissPK) } names := []string{mfi.column, rfi.column} var nums int64 for _, md := range models { ind := reflect.Indirect(reflect.ValueOf(md)) var v2 interface{} if ind.Kind() != reflect.Struct { v2 = ind.Interface() } else { _, v2, exist = getExistPk(fi.relModelInfo, ind) if exist == false { panic(ErrMissPK) } } values := []interface{}{v1, v2} _, err := dbase.InsertValue(orm.db, mi, names, values) if err != nil { return nums, err } nums += 1 } return nums, nil } func (o *queryM2M) Remove(mds ...interface{}) (int64, error) { fi := o.fi qs := o.qs.Filter(fi.reverseFieldInfo.name, o.md) nums, err := qs.Filter(fi.reverseFieldInfoTwo.name+ExprSep+"in", mds).Delete() if err != nil { return nums, err } return nums, nil } func (o *queryM2M) Exist(md interface{}) bool { fi := o.fi return o.qs.Filter(fi.reverseFieldInfo.name, o.md). Filter(fi.reverseFieldInfoTwo.name, md).Exist() } func (o *queryM2M) Clear() (int64, error) { fi := o.fi return o.qs.Filter(fi.reverseFieldInfo.name, o.md).Delete() } func (o *queryM2M) Count() (int64, error) { fi := o.fi return o.qs.Filter(fi.reverseFieldInfo.name, o.md).Count() } var _ QueryM2Mer = new(queryM2M) func newQueryM2M(md interface{}, o *orm, mi *modelInfo, fi *fieldInfo, ind reflect.Value) QueryM2Mer { qm2m := new(queryM2M) qm2m.md = md qm2m.mi = mi qm2m.fi = fi qm2m.ind = ind qm2m.qs = newQuerySet(o, fi.relThroughModelInfo).(*querySet) return qm2m }