Beego/orm/orm_object.go

package orm

import (
	"database/sql"
	"fmt"
	"reflect"
)

type insertSet struct {
	mi     *modelInfo
	orm    *orm
	stmt   *sql.Stmt
	closed bool
}

func (o *insertSet) Insert(md Modeler) (int64, error) {
	if o.closed {
		return 0, ErrStmtClosed
	}
	val := reflect.ValueOf(md)
	ind := reflect.Indirect(val)
	if val.Type() != o.mi.addrField.Type() {
		panic(fmt.Sprintf("<Inserter.Insert> need type `%s` but found `%s`", o.mi.addrField.Type(), val.Type()))
	}
	id, err := o.orm.alias.DbBaser.InsertStmt(o.stmt, o.mi, ind)
	if err != nil {
		return id, err
	}
	if id > 0 {
		if o.mi.fields.auto != nil {
			ind.Field(o.mi.fields.auto.fieldIndex).SetInt(id)
		}
	}
	return id, nil
}

func (o *insertSet) Close() error {
	o.closed = true
	return o.stmt.Close()
}

func newInsertSet(orm *orm, mi *modelInfo) (Inserter, error) {
	bi := new(insertSet)
	bi.orm = orm
	bi.mi = mi
	st, err := orm.alias.DbBaser.PrepareInsert(orm.db, mi)
	if err != nil {
		return nil, err
	}
	bi.stmt = st
	return bi, nil
}

type object struct {
	ind reflect.Value
	mi  *modelInfo
	orm *orm
}

func (o *object) Insert() (int64, error) {
	id, err := o.orm.alias.DbBaser.Insert(o.orm.db, o.mi, o.ind)
	if err != nil {
		return id, err
	}
	if id > 0 {
		if o.mi.fields.auto != nil {
			o.ind.Field(o.mi.fields.auto.fieldIndex).SetInt(id)
		}
	}
	return id, nil
}

func (o *object) Update() (int64, error) {
	num, err := o.orm.alias.DbBaser.Update(o.orm.db, o.mi, o.ind)
	if err != nil {
		return num, err
	}
	return num, nil
}

func (o *object) Delete() (int64, error) {
	num, err := o.orm.alias.DbBaser.Delete(o.orm.db, o.mi, o.ind)
	if err != nil {
		return num, err
	}
	if num > 0 {
		if o.mi.fields.auto != nil {
			o.ind.Field(o.mi.fields.auto.fieldIndex).SetInt(0)
		}
	}
	return num, nil
}

func newObject(orm *orm, mi *modelInfo, md Modeler) ObjectSeter {
	o := new(object)
	ind := reflect.Indirect(reflect.ValueOf(md))
	o.ind = ind
	o.mi = mi
	o.orm = orm
	return o
}