youngee_b_api/app/dao/selection_info_dao.go

package dao

import (
	"time"
	"youngee_b_api/app/entity"
)

type SelectionInfoDAO struct{}

func (SelectionInfoDAO) GetSelectionInfoById(id string) (entity.SelectionInfo, error) {
	var selectionInfo entity.SelectionInfo
	err := Db.Where("selection_id = ?", id).First(&selectionInfo).Error
	return selectionInfo, err
}

func (SelectionInfoDAO) UpdateSelectionInfoById(id int, enterpriseId string) {
	Db.Model(&entity.SelectionInfo{}).Where("id = ?", id).Update("enterprise_id", enterpriseId)
}

func (SelectionInfoDAO) DeleteSelectionInfoById(id int) error {
	err := Db.Delete(&entity.SelectionInfo{}, id).Error
	return err
}

// 根据enterpriseId查询指定某天的所有带货数据
func (SelectionInfoDAO) GetSelectionInfoListOfDay(enterpriseId string, date time.Time) ([]entity.SelectionInfo, error) {
	var selectionInfos []entity.SelectionInfo
	// 构建查询
	query := Db.Model(&entity.SelectionInfo{})
	if enterpriseId != "" {
		query = query.Where("enterprise_id = ?", enterpriseId)
	}
	// 将日期部分提取出来进行匹配
	query = query.Where("DATE(created_at) = ?", date.Format("2006-01-02"))
	err := query.Find(&selectionInfos).Error
	return selectionInfos, err
}