fusenapi/model/gmodel/fs_user_stock_logic.go

package gmodel

import "context"

// TODO: 使用model的属性做你想做的

func (s *FsUserStockModel) GetUserAllStockByIds(ctx context.Context, ids []int64, userId int64) (resp []FsUserStock, err error) {
	if len(ids) == 0 {
		return nil, nil
	}
	if userId <= 0 {
		return nil, nil
	}
	err = s.db.WithContext(ctx).Model(&FsUserStock{}).Where("`id` in (?) and `user_id` = ?", ids, userId).Find(&resp).Error
	return resp, err
}
func (s *FsUserStockModel) FindOne(ctx context.Context, id int64, userId int64, lock ...bool) (resp *FsUserStock, err error) {
	db := s.db.WithContext(ctx).Model(&FsUserStock{}).Where("`id`  = ? and `user_id` = ?", id, userId)
	if len(lock) != 0 && lock[0] {
		db = db.Set("gorm:query_option", "FOR UPDATE")
	}
	err = db.First(&resp).Error
	return resp, err
}