sdbc/insert_scope.go

package sdbc

import "context"

type _insert struct {
	scope      *scope
	sdbc       *sdbc
	ctx        context.Context
	selectOpts []interface{}
	omitOpts   []string
}

func (i *_insert) SetContext(ctx context.Context) *_insert {
	i.ctx = ctx
	return i
}

// SetSelect 只插入字段
func (i *_insert) SetSelect(selects ...any) *_insert {
	i.selectOpts = selects
	return i
}

// SetOmit 忽略字段
func (i *_insert) SetOmit(omits ...string) *_insert {
	i.omitOpts = omits
	return i
}

// Insert 插入数据
func (i *_insert) Insert(docs any) error {
	db := i.sdbc.client
	if len(i.selectOpts) != 0 {
		db = db.Select(i.selectOpts[0], i.selectOpts[1:]...)
	}
	if len(i.omitOpts) != 0 {
		db = db.Omit(i.omitOpts...)
	}
	return db.Model(i.scope.model.ptr()).Create(docs).Error
}

// InsertBatch 按批插入数据
func (i *_insert) InsertBatch(docs any) error {
	db := i.sdbc.client
	if len(i.selectOpts) != 0 {
		db = db.Select(i.selectOpts[0], i.selectOpts[1:]...)
	}
	if len(i.omitOpts) != 0 {
		db = db.Omit(i.omitOpts...)
	}
	return db.Model(i.scope.model.ptr()).CreateInBatches(docs, 1000).Error
}