gorm/callback_create.go

package gorm

import (
	"fmt"
	"strings"
	"time"
)

func BeforeCreate(scope *Scope) {
	scope.CallMethod("BeforeSave")
	scope.CallMethod("BeforeCreate")
}

func UpdateTimeStampWhenCreate(scope *Scope) {
	if !scope.HasError() {
		scope.SetColumn("CreatedAt", time.Now())
		scope.SetColumn("UpdatedAt", time.Now())
	}
}

func Create(scope *Scope) {
	defer scope.Trace(time.Now())

	if !scope.HasError() {
		// set create sql
		var sqls, columns []string

		for _, field := range scope.Fields() {
			if field.DBName != scope.PrimaryKey() && len(field.SqlTag) > 0 && !field.IsIgnored {
				columns = append(columns, scope.quote(field.DBName))
				sqls = append(sqls, scope.AddToVars(field.Value))
			}
		}

		scope.Raw(fmt.Sprintf(
			"INSERT INTO %v (%v) VALUES (%v) %v",
			scope.TableName(),
			strings.Join(columns, ","),
			strings.Join(sqls, ","),
			scope.Dialect().ReturningStr(scope.PrimaryKey()),
		))

		// execute create sql
		var id interface{}
		if scope.Dialect().SupportLastInsertId() {
			if sql_result, err := scope.DB().Exec(scope.Sql, scope.SqlVars...); scope.Err(err) == nil {
				id, err = sql_result.LastInsertId()
				scope.Err(err)
			}
		} else {
			scope.Err(scope.DB().QueryRow(scope.Sql, scope.SqlVars...).Scan(&id))
		}

		if !scope.HasError() {
			scope.SetColumn(scope.PrimaryKey(), id)
		}
	}
}

func AfterCreate(scope *Scope) {
	scope.CallMethod("AfterCreate")
	scope.CallMethod("AfterSave")
}

func init() {
	DefaultCallback.Create().Register("begin_transaction", BeginTransaction)
	DefaultCallback.Create().Register("before_create", BeforeCreate)
	DefaultCallback.Create().Register("save_before_associations", SaveBeforeAssociations)
	DefaultCallback.Create().Register("update_time_stamp_when_create", UpdateTimeStampWhenCreate)
	DefaultCallback.Create().Register("create", Create)
	DefaultCallback.Create().Register("save_after_associations", SaveAfterAssociations)
	DefaultCallback.Create().Register("after_create", AfterCreate)
	DefaultCallback.Create().Register("commit_or_rollback_transaction", CommitOrRollbackTransaction)
}