knowledge-api/api/dao/QuestionQa.go

package dao

import (
	"errors"
	"gorm.io/gorm"
	"knowledge/api/model"
	"knowledge/api/requests"
	"knowledge/global"
	"strings"
	"time"
)

type QuestionQaDao struct {
}

// GetQuestionQaById 获取数据-id
func (r *QuestionQaDao) GetQuestionQaById(qaId int64) (m *model.QuestionQa, err error) {
	err = global.Db.First(&m, qaId).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// DeleteQuestionQa 删除
func (r *QuestionQaDao) DeleteQuestionQa(tx *gorm.DB, maps interface{}) error {
	err := tx.Where(maps).Delete(&model.QuestionQa{}).Error
	if err != nil {
		return err
	}
	return nil
}

// DeleteQuestionQaById 删除-id
func (r *QuestionQaDao) DeleteQuestionQaById(tx *gorm.DB, qaId int64) error {
	if err := tx.Delete(&model.QuestionQa{}, qaId).Error; err != nil {
		return err
	}
	return nil
}

// EditQuestionQa 修改
func (r *QuestionQaDao) EditQuestionQa(tx *gorm.DB, maps interface{}, data interface{}) error {
	err := tx.Model(&model.QuestionQa{}).Where(maps).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// EditQuestionQaById 修改-id
func (r *QuestionQaDao) EditQuestionQaById(tx *gorm.DB, qaId int64, data interface{}) error {
	err := tx.Model(&model.QuestionQa{}).Where("qa_id = ?", qaId).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// GetQuestionQaList 获取列表
func (r *QuestionQaDao) GetQuestionQaList(maps interface{}) (m []*model.QuestionQa, err error) {
	err = global.Db.Where(maps).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetQuestionQaInQaIdList 获取列表-in主键id
func (r *QuestionQaDao) GetQuestionQaInQaIdList(qaIds []string) (m []*model.QuestionQa, err error) {
	err = global.Db.Where("qa_id in ?", qaIds).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// AddQuestionQa 新增
func (r *QuestionQaDao) AddQuestionQa(tx *gorm.DB, model *model.QuestionQa) (*model.QuestionQa, error) {
	if err := tx.Create(model).Error; err != nil {
		return nil, err
	}
	return model, nil
}

// GetQuestionQa 获取
func (r *QuestionQaDao) GetQuestionQa(maps interface{}) (m *model.QuestionQa, err error) {
	err = global.Db.Where(maps).First(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetQuestionQaListByQaExpireTime 获取列表-今天开始时间/过期时间
func (r *QuestionQaDao) GetQuestionQaListByQaExpireTime(maps interface{}, startTime, endTime string) (m []*model.QuestionQa, err error) {
	err = global.Db.Where(maps).Where("qa_expire_time BETWEEN ? AND ?", startTime, endTime).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetQuestionQaPageSearch 获取题目列表-分页
func (r *QuestionQaDao) GetQuestionQaPageSearch(req requests.GetQuestionQaPage, page, pageSize int) (m []*model.QuestionQa, total int64, err error) {
	var totalRecords int64

	// 构建查询条件
	query := global.Db.Model(&model.QuestionQa{})

	// 主键id
	if req.QaId != "" {
		query = query.Where("qa_id = ?", req.QaId)
	}

	// 名称
	if req.QaName != "" {
		query = query.Where("qa_name LIKE ?", "%"+req.QaName+"%")
	}

	// 类型
	query = query.Where("qa_type = ?", req.QaType)

	// 状态
	if req.QaStatus != nil {
		query = query.Where("qa_status = ?", req.QaStatus)
	}

	// 规则解释
	if req.QaRuleContent != "" {
		query = query.Where("qa_rule_content LIKE ?", "%"+req.QaRuleContent+"%")
	}

	// 展示类型
	if req.QaDisplayType != nil {
		query = query.Where("qa_display_type = ?", req.QaDisplayType)
	}

	// 过期时间
	if req.QaExpireTime != "" {
		qaExpireTime := strings.Split(req.QaExpireTime, "&")
		if len(qaExpireTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", qaExpireTime[0])
			endTime, _ := time.Parse("2006-01-02", qaExpireTime[1])

			endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)

			query = query.Where("qa_expire_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 创建时间
	if req.CreatedAt != "" {
		createdAt := strings.Split(req.CreatedAt, "&")
		if len(createdAt) == 2 {
			startTime, _ := time.Parse("2006-01-02", createdAt[0])
			endTime, _ := time.Parse("2006-01-02", createdAt[1])

			endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)

			query = query.Where("created_at BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 修改时间
	if req.UpdatedAt != "" {
		updatedAt := strings.Split(req.UpdatedAt, "&")
		if len(updatedAt) == 2 {
			startTime, _ := time.Parse("2006-01-02", updatedAt[0])
			endTime, _ := time.Parse("2006-01-02", updatedAt[1])

			endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)

			query = query.Where("updated_at BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 排序
	if req.Order != nil {
		// 有效期
		if req.Order.QaExpireTime != "" {
			if req.Order.QaExpireTime != "desc" && req.Order.QaExpireTime != "asc" {
				return nil, 0, errors.New("排序字段错误")
			}

			query = query.Order("qa_expire_time " + req.Order.QaExpireTime)
		}

		// 题目数量
		if req.Order.QaQuantity != "" {
			if req.Order.QaQuantity != "desc" && req.Order.QaQuantity != "asc" {
				return nil, 0, errors.New("排序字段错误")
			}

			query = query.Order("qa_quantity " + req.Order.QaQuantity)
		}

		// 修改时间
		if req.Order.UpdatedAt != "" {
			if req.Order.UpdatedAt != "desc" && req.Order.UpdatedAt != "asc" {
				return nil, 0, errors.New("排序字段错误")
			}

			query = query.Order("updated_at " + req.Order.UpdatedAt)
		}
	}

	// 排序
	query = query.Order("created_at desc")

	// 查询总数量
	if err := query.Count(&totalRecords).Error; err != nil {
		return nil, 0, err
	}

	err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error
	if err != nil {
		return nil, 0, err
	}
	return m, totalRecords, nil
}

// GetQuestionQaListSearch 获取列表
func (r *QuestionQaDao) GetQuestionQaListSearch(req requests.GetQuestionQaList) (m []*model.QuestionQa, err error) {
	// 构建查询条件
	query := global.Db.Model(&model.QuestionQa{})

	// 名称
	if req.QaName != "" {
		query = query.Where("qa_name LIKE ?", "%"+req.QaName+"%")
	}

	// 类型
	query = query.Where("qa_type = ?", req.QaType)

	// 状态
	if req.QaStatus != nil {
		query = query.Where("qa_status = ?", req.QaStatus)
	}

	// 规则解释
	if req.QaRuleContent != "" {
		query = query.Where("qa_rule_content LIKE ?", "%"+req.QaRuleContent+"%")
	}

	// 展示类型
	if req.QaDisplayType != nil {
		query = query.Where("qa_display_type = ?", req.QaDisplayType)
	}

	// 过期时间
	if req.QaExpireTime != "" {
		qaExpireTime := strings.Split(req.QaExpireTime, "&")
		if len(qaExpireTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", qaExpireTime[0])
			endTime, _ := time.Parse("2006-01-02", qaExpireTime[1])

			endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)

			query = query.Where("qa_expire_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 创建时间
	if req.CreatedAt != "" {
		createdAt := strings.Split(req.CreatedAt, "&")
		if len(createdAt) == 2 {
			startTime, _ := time.Parse("2006-01-02", createdAt[0])
			endTime, _ := time.Parse("2006-01-02", createdAt[1])

			endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)

			query = query.Where("created_at BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 修改时间
	if req.UpdatedAt != "" {
		updatedAt := strings.Split(req.UpdatedAt, "&")
		if len(updatedAt) == 2 {
			startTime, _ := time.Parse("2006-01-02", updatedAt[0])
			endTime, _ := time.Parse("2006-01-02", updatedAt[1])

			endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)

			query = query.Where("updated_at BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 排序
	if req.Order != nil {
		// 有效期
		if req.Order.QaExpireTime != "" {
			if req.Order.QaExpireTime != "desc" && req.Order.QaExpireTime != "asc" {
				return nil, errors.New("排序字段错误")
			}

			query = query.Order("qa_expire_time " + req.Order.QaExpireTime)
		}

		// 题目数量
		if req.Order.QaQuantity != "" {
			if req.Order.QaQuantity != "desc" && req.Order.QaQuantity != "asc" {
				return nil, errors.New("排序字段错误")
			}

			query = query.Order("qa_quantity " + req.Order.QaQuantity)
		}

		// 修改时间
		if req.Order.UpdatedAt != "" {
			if req.Order.UpdatedAt != "desc" && req.Order.UpdatedAt != "asc" {
				return nil, errors.New("排序字段错误")
			}

			query = query.Order("updated_at " + req.Order.UpdatedAt)
		}
	}

	// 排序
	query = query.Order("created_at desc")

	err = query.Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// Inc 自增
func (r *QuestionQaDao) Inc(tx *gorm.DB, qaId int64, field string, numeral int) error {
	err := tx.Model(&model.QuestionQa{}).Where("qa_id = ?", qaId).UpdateColumn(field, gorm.Expr(field+" + ?", numeral)).Error
	if err != nil {
		return err
	}
	return nil
}

// Dec 自减
func (r *QuestionQaDao) Dec(tx *gorm.DB, qaId int64, field string, numeral int) error {
	err := tx.Model(&model.QuestionQa{}).Where("qa_id = ?", qaId).UpdateColumn(field, gorm.Expr(field+" - ?", numeral)).Error
	if err != nil {
		return err
	}
	return nil
}