61 lines
2.1 KiB
Go
61 lines
2.1 KiB
Go
|
package models
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"errors"
|
|||
|
|
|
|||
|
|
"git.apinb.com/bsm-sdk/core/database"
|
|||
|
|
"git.apinb.com/bsm-sdk/core/errcode"
|
|||
|
|
"git.apinb.com/heqiapp/platforms/backend/api/internal/impl"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// OrgGasStation 对应 org_gas_station，表示可燃气体站经营主体。
|
|||
|
|
type OrgGasStation struct {
|
|||
|
|
Entity
|
|||
|
|
StationCode string `gorm:"column:station_code;type:varchar(32);uniqueIndex;not null" json:"station_code"` // 气站编码
|
|||
|
|
Name string `gorm:"column:name;type:varchar(128);not null" json:"name"` // 气站名称
|
|||
|
|
Principal string `gorm:"column:principal;type:varchar(64);not null" json:"principal"` // 负责人
|
|||
|
|
ServiceArea string `gorm:"column:service_area;type:varchar(128);not null" json:"service_area"` // 服务区域
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
database.AppendMigrate(&OrgGasStation{})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TableName 返回与模型、文件名一致的单数数据表名。
|
|||
|
|
func (table *OrgGasStation) TableName() string { return "org_gas_station" }
|
|||
|
|
|
|||
|
|
// CreateOrgGasStation 创建待审核气站。
|
|||
|
|
func CreateOrgGasStation(data *OrgGasStation) error {
|
|||
|
|
if err := impl.DBService.Create(data).Error; err != nil {
|
|||
|
|
return errcode.ErrDB
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ListOrgGasStation 按创建时间倒序查询气站。
|
|||
|
|
func ListOrgGasStation(page, size int) ([]OrgGasStation, int64, error) {
|
|||
|
|
var list []OrgGasStation
|
|||
|
|
var total int64
|
|||
|
|
databaseQuery := impl.DBService.Model(&OrgGasStation{})
|
|||
|
|
if err := databaseQuery.Count(&total).Error; err != nil {
|
|||
|
|
return nil, 0, errcode.ErrDB
|
|||
|
|
}
|
|||
|
|
if err := databaseQuery.Order("created_at desc").Offset((page - 1) * size).Limit(size).Find(&list).Error; err != nil {
|
|||
|
|
return nil, 0, errcode.ErrDB
|
|||
|
|
}
|
|||
|
|
return list, total, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetOrgGasStationByIdentity 查询单个气站。
|
|||
|
|
func GetOrgGasStationByIdentity(identity string) (*OrgGasStation, error) {
|
|||
|
|
var data OrgGasStation
|
|||
|
|
if err := impl.DBService.Where("identity = ?", identity).First(&data).Error; err != nil {
|
|||
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|||
|
|
return nil, errcode.ErrRecordNotFound
|
|||
|
|
}
|
|||
|
|
return nil, errcode.ErrDB
|
|||
|
|
}
|
|||
|
|
return &data, nil
|
|||
|
|
}
|