123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157 |
- // Copyright 2019 getensh.com. All rights reserved.
- // Use of this source code is governed by getensh.com.
- package model
- import (
- "cp-organization-management/errors"
- "fmt"
- "time"
- "github.com/jaryhe/gopkgs/logger"
- "github.com/jinzhu/gorm"
- "go.uber.org/zap"
- )
- type Station struct {
- Id int64 `json:"id" gorm:"primary_key"`
- StationName string `json:"station_name" gorm:"station_name;commnet:'充电站名';not null;default:''"`
- StationCode string `json:"station_code" gorm:"station_code;commnet:'充电站编码';not null;default:''"`
- DeviceCount int `json:"device_count" gorm:"device_count;commnet:'设备数量';not null;default:0"`
- CreatedAt time.Time `json:"created_at" gorm:"created_at"`
- UpdatedAt time.Time `json:"updated_at" gorm:"updated_at"`
- table string `json:"table"`
- }
- func (p *Station) TableName() string {
- return p.table
- }
- func NewStation(database string) *Station {
- return &Station{table: fmt.Sprintf("%s.%s", database, "t_station")}
- }
- func (p *Station) CreateTable(db *gorm.DB) error {
- err := db.CreateTable(p).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", "create table "+p.TableName()),
- zap.String("fields", ""),
- zap.String("error", err.Error()))
- }
- return nil
- }
- func (p *Station) Find(db *gorm.DB, where map[string]interface{}) error {
- err := db.Table(p.TableName()).Where(where).Find(p).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return err
- }
- func (p *Station) Last(db *gorm.DB) error {
- err := db.Table(p.TableName()).Last(p).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return err
- }
- // Insert 插入一条记录
- func (p *Station) Insert(db *gorm.DB) error {
- timeNow := time.Now()
- p.CreatedAt = timeNow
- p.UpdatedAt = timeNow
- err := db.Create(p).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return err
- }
- func (p *Station) Delete(db *gorm.DB, filter map[string]interface{}) error {
- err := db.Where(filter).Delete(p).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return err
- }
- func (p *Station) Save(db *gorm.DB) error {
- p.UpdatedAt = time.Now()
- err := db.Save(p).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return err
- }
- func (p *Station) Update(db *gorm.DB) error {
- p.UpdatedAt = time.Now()
- err := db.Model(p).Updates(p).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return err
- }
- // 通过结构体变量更新字段值, gorm库会忽略零值字段。就是字段值等于0, nil, "", false这些值会被忽略掉,不会更新。如果想更新零值,可以使用map类型替代结构体。
- func (p *Station) UpdateSome(db *gorm.DB, filed map[string]interface{}) error {
- if filed == nil {
- return errors.ParamsError
- }
- filed["updated_at"] = time.Now()
- err := db.Model(p).Updates(filed).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return err
- }
- func (p *Station) QueryAll(db *gorm.DB, filter map[string]interface{}, out interface{}) error {
- err := db.Table(p.TableName()).Where(filter).Find(out).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return err
- }
- func (p *Station) List(db *gorm.DB, limit, offset int, out interface{}) error {
- err := db.Table(p.TableName()).Limit(limit).Offset(offset).Find(out).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return err
- }
- func (p *Station) Count(db *gorm.DB) (int64, error) {
- var count int64
- err := db.Table(p.TableName()).Count(&count).Error
- if err != nil {
- logger.Error("mysql",
- zap.String("sql", fmt.Sprintf("%+v", db.QueryExpr())),
- zap.String("error", err.Error()))
- }
- return count, err
- }
|