From ca37557a26809b6089cb48a11e56b8c624ceebce Mon Sep 17 00:00:00 2001 From: gobenon Date: Thu, 19 Nov 2015 14:30:14 +0200 Subject: [PATCH] Update orm_querym2m.go --- orm/orm_querym2m.go | 30 ++++++++++++++++++++++-------- 1 file changed, 22 insertions(+), 8 deletions(-) diff --git a/orm/orm_querym2m.go b/orm/orm_querym2m.go index 1eaccf72..1d8fea3e 100644 --- a/orm/orm_querym2m.go +++ b/orm/orm_querym2m.go @@ -44,7 +44,15 @@ func (o *queryM2M) Add(mds ...interface{}) (int64, error) { dbase := orm.alias.DbBaser var models []interface{} - + last_md_col_name := mi.fields.dbcols[len(mi.fields.dbcols)-1] + last_md := mds[len(mds)-1] + var v3 interface{} + var names []string + var values []interface{} + if reflect.Indirect(reflect.ValueOf(last_md)).Kind() != reflect.Struct { + v3 = (last_md) + mds = mds[:len(mds)-1] + } for _, md := range mds { val := reflect.ValueOf(md) if val.Kind() == reflect.Slice || val.Kind() == reflect.Array { @@ -63,15 +71,18 @@ func (o *queryM2M) Add(mds ...interface{}) (int64, error) { if exist == false { panic(ErrMissPK) } + if v3 != nil { + names = []string{mfi.column, rfi.column, last_md_col_name} - names := []string{mfi.column, rfi.column} - - values := make([]interface{}, 0, len(models)*2) + values = make([]interface{}, 0, len(models)*3) + } else { + names = []string{mfi.column, rfi.column} + values = make([]interface{}, 0, len(models)*2) + } for _, md := range models { ind := reflect.Indirect(reflect.ValueOf(md)) - var v2 interface{} if ind.Kind() != reflect.Struct { v2 = ind.Interface() @@ -81,14 +92,17 @@ func (o *queryM2M) Add(mds ...interface{}) (int64, error) { panic(ErrMissPK) } } - - values = append(values, v1, v2) + if v3 == nil { + values = append(values, v1, v2) + } else { + values = append(values, v1, v2, v3) + } } - return dbase.InsertValue(orm.db, mi, true, names, values) } + // remove models following the origin model relationship func (o *queryM2M) Remove(mds ...interface{}) (int64, error) { fi := o.fi