140 lines
3.6 KiB
Go
140 lines
3.6 KiB
Go
package dao
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"hospital-admin-api/api/model"
|
|
"hospital-admin-api/api/requests"
|
|
"hospital-admin-api/global"
|
|
)
|
|
|
|
type HospitalDao struct {
|
|
}
|
|
|
|
// GetHospitalById 获取医院数据-医院id
|
|
func (r *HospitalDao) GetHospitalById(hospitalId int64) (m *model.Hospital, err error) {
|
|
err = global.Db.First(&m, hospitalId).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
// AddHospital 新增医院
|
|
func (r *HospitalDao) AddHospital(tx *gorm.DB, model *model.Hospital) (*model.Hospital, error) {
|
|
if err := tx.Create(model).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return model, nil
|
|
}
|
|
|
|
// GetHospitalList 获取医院列表
|
|
func (r *HospitalDao) GetHospitalList(maps interface{}) (m []*model.Hospital, err error) {
|
|
err = global.Db.Where(maps).Find(&m).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
// DeleteHospitalById 删除医院-医院id
|
|
func (r *HospitalDao) DeleteHospitalById(tx *gorm.DB, hospitalId int64) error {
|
|
if err := tx.Delete(&model.Hospital{}, hospitalId).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// EditHospitalById 修改医院-医院id
|
|
func (r *HospitalDao) EditHospitalById(tx *gorm.DB, hospitalId int64, data interface{}) error {
|
|
err := tx.Model(&model.Hospital{}).Where("hospital_id = ?", hospitalId).Updates(data).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetHospitalLimitByMaps 获取医院列表-限制条数
|
|
func (r *HospitalDao) GetHospitalLimitByMaps(hospitalRequest requests.GetHospitalList) (m []*model.Hospital, err error) {
|
|
result := global.Db
|
|
if hospitalRequest.HospitalName != "" {
|
|
result = result.Where("hospital_name like ?", "%"+hospitalRequest.HospitalName+"%")
|
|
}
|
|
|
|
if hospitalRequest.HospitalLevelName != "" {
|
|
result = result.Where("hospital_level_name like ?", "%"+hospitalRequest.HospitalLevelName+"%")
|
|
}
|
|
|
|
if hospitalRequest.ProvinceId != 0 {
|
|
result = result.Where("province_id = ?", hospitalRequest.ProvinceId)
|
|
}
|
|
|
|
if hospitalRequest.Province != "" {
|
|
result = result.Where("province like ?", "%"+hospitalRequest.Province+"%")
|
|
}
|
|
|
|
if hospitalRequest.CityId != 0 {
|
|
result = result.Where("city_id = ?", hospitalRequest.CityId)
|
|
}
|
|
|
|
if hospitalRequest.City != "" {
|
|
result = result.Where("city like ?", "%"+hospitalRequest.City+"%")
|
|
}
|
|
|
|
if hospitalRequest.CountyId != 0 {
|
|
result = result.Where("county_id = ?", hospitalRequest.CountyId)
|
|
}
|
|
|
|
if hospitalRequest.County != "" {
|
|
result = result.Where("county like ?", "%"+hospitalRequest.County+"%")
|
|
}
|
|
|
|
err = result.Limit(10).Find(&m).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
// GetHospitalPageSearch 获取医院列表-分页
|
|
func (r *HospitalDao) GetHospitalPageSearch(req requests.GetHospitalPage, page, pageSize int) (m []*model.Hospital, total int64, err error) {
|
|
var totalRecords int64
|
|
|
|
// 构建查询条件
|
|
query := global.Db.Model(&model.Hospital{})
|
|
|
|
if req.HospitalName != "" {
|
|
query = query.Where("hospital_name like ?", "%"+req.HospitalName+"%")
|
|
}
|
|
|
|
if req.HospitalLevelName != "" {
|
|
query = query.Where("hospital_level_name like ?", "%"+req.HospitalLevelName+"%")
|
|
}
|
|
|
|
if req.ProvinceId != 0 {
|
|
query = query.Where("province_id = ?", req.ProvinceId)
|
|
}
|
|
|
|
if req.CityId != nil {
|
|
query = query.Where("city_id = ?", req.CityId)
|
|
}
|
|
|
|
if req.CountyId != nil {
|
|
query = query.Where("county_id = ?", req.CountyId)
|
|
}
|
|
|
|
// 排序
|
|
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
|
|
}
|