golang-yitisheng-server/server/modules/yx/mapper/yx_school_major_mapper.go

124 lines
3.5 KiB
Go

// Package mapper 数据访问层
package mapper
import (
"server/config"
"server/modules/yx/dto"
"server/modules/yx/entity"
"gorm.io/gorm/clause"
)
type YxSchoolMajorMapper struct{}
func NewYxSchoolMajorMapper() *YxSchoolMajorMapper {
return &YxSchoolMajorMapper{}
}
func (m *YxSchoolMajorMapper) FindAll(page, size int) ([]entity.YxSchoolMajor, int64, error) {
var items []entity.YxSchoolMajor
var total int64
config.DB.Model(&entity.YxSchoolMajor{}).Count(&total)
err := config.DB.Offset((page - 1) * size).Limit(size).Find(&items).Error
return items, total, err
}
func (m *YxSchoolMajorMapper) FindByID(id string) (*entity.YxSchoolMajor, error) {
var item entity.YxSchoolMajor
err := config.DB.First(&item, "id = ?", id).Error
return &item, err
}
func (m *YxSchoolMajorMapper) Create(item *entity.YxSchoolMajor) error {
return config.DB.Create(item).Error
}
func (m *YxSchoolMajorMapper) Update(item *entity.YxSchoolMajor) error {
return config.DB.Save(item).Error
}
func (m *YxSchoolMajorMapper) UpdateFields(id string, fields map[string]interface{}) error {
return config.DB.Model(&entity.YxSchoolMajor{}).Where("id = ?", id).Updates(fields).Error
}
func (m *YxSchoolMajorMapper) Delete(id string) error {
return config.DB.Delete(&entity.YxSchoolMajor{}, "id = ?", id).Error
}
func (m *YxSchoolMajorMapper) BatchCreate(items []entity.YxSchoolMajor, batchSize int) error {
return config.DB.CreateInBatches(items, batchSize).Error
}
func (m *YxSchoolMajorMapper) BatchUpdate(items []entity.YxSchoolMajor) error {
return config.DB.Save(items).Error
}
func (m *YxSchoolMajorMapper) BatchUpsert(items []entity.YxSchoolMajor, updateColumns []string) error {
return config.DB.Clauses(clause.OnConflict{
Columns: []clause.Column{{Name: "id"}},
DoUpdates: clause.AssignmentColumns(updateColumns),
}).CreateInBatches(items, 100).Error
}
func (m *YxSchoolMajorMapper) BatchDelete(ids []string) error {
return config.DB.Delete(&entity.YxSchoolMajor{}, "id IN ?", ids).Error
}
func (m *YxSchoolMajorMapper) SelectSchoolMajor(query dto.SchoolMajorQuery) ([]dto.SchoolMajorDTO, error) {
var items []dto.SchoolMajorDTO
sql := `
SELECT
sm.school_code,
sc.school_name,
sm.major_code,
sm.major_name,
sm.major_type,
sm.major_type_child,
sm.plan_num,
sm.main_subjects,
sm.limitation,
sm.chinese_score_limitation,
sm.english_score_limitation,
sm.cultural_score_limitation,
sm.professional_score_limitation,
sm.enrollment_code,
sm.tuition,
sm.detail,
sm.category,
sm.batch,
sm.rules_enroll_probability,
sm.probability_operator,
sm.private_rules_enroll_probability,
sm.private_probability_operator,
sm.rules_enroll_probability_sx,
sm.kslx,
sm.state
FROM yx_school_major sm
LEFT JOIN (SELECT school_id,school_name,school_code FROM yx_school_child group by school_code) sc ON sc.school_code = sm.school_code
LEFT JOIN yx_school s ON s.id = sc.school_id
WHERE 1=1 AND sm.state > 0
`
params := []interface{}{}
if query.MajorType != "" {
sql += " AND sm.major_type = ?"
params = append(params, query.MajorType)
}
if query.Category != "" {
sql += " AND sm.category = ?"
params = append(params, query.Category)
}
if len(query.MajorTypeChildren) > 0 {
sql += " AND sm.major_type_child IN ?"
params = append(params, query.MajorTypeChildren)
}
if query.MainSubjects != "" {
sql += " AND sm.main_subjects = ?"
params = append(params, query.MainSubjects)
}
err := config.DB.Raw(sql, params...).Scan(&items).Error
return items, err
}