91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package rest
|
|
|
|
import (
|
|
"github.com/jinzhu/gorm"
|
|
_ "github.com/jinzhu/gorm/dialects/mysql"
|
|
"github.com/nu7hatch/gouuid"
|
|
"time"
|
|
)
|
|
|
|
type SecurityVisitDao struct {
|
|
BaseDao
|
|
}
|
|
|
|
//按照Id查询文件
|
|
func (this *SecurityVisitDao) FindByUuid(uuid string) *SecurityVisit {
|
|
|
|
// Read
|
|
var securityVisit SecurityVisit
|
|
db := this.context.DB.Where(&SecurityVisit{Base: Base{Uuid: uuid}}).First(&securityVisit)
|
|
if db.Error != nil {
|
|
return nil
|
|
}
|
|
return &securityVisit
|
|
}
|
|
|
|
//按照Id查询文件
|
|
func (this *SecurityVisitDao) CheckByUuid(uuid string) *SecurityVisit {
|
|
|
|
// Read
|
|
var securityVisit SecurityVisit
|
|
db := this.context.DB.Where(&SecurityVisit{Base: Base{Uuid: uuid}}).First(&securityVisit)
|
|
this.PanicError(db.Error)
|
|
|
|
return &securityVisit
|
|
|
|
}
|
|
|
|
//按分页条件获取分页
|
|
func (this *SecurityVisitDao) Page(page int, pageSize int, userUuid string, sortArray []OrderPair) *Pager {
|
|
|
|
var wp = &WherePair{}
|
|
|
|
if userUuid != "" {
|
|
wp = wp.And(&WherePair{Query: "user_uuid = ?", Args: []interface{}{userUuid}})
|
|
}
|
|
|
|
var conditionDB *gorm.DB
|
|
conditionDB = this.context.DB.Model(&SecurityVisit{}).Where(wp.Query, wp.Args...)
|
|
|
|
count := 0
|
|
db := conditionDB.Count(&count)
|
|
this.PanicError(db.Error)
|
|
|
|
var securityVisits []*SecurityVisit
|
|
db = conditionDB.Order(this.GetSortString(sortArray)).Offset(page * pageSize).Limit(pageSize).Find(&securityVisits)
|
|
this.PanicError(db.Error)
|
|
pager := NewPager(page, pageSize, count, securityVisits)
|
|
|
|
return pager
|
|
}
|
|
|
|
//创建
|
|
func (this *SecurityVisitDao) Create(securityVisit *SecurityVisit) *SecurityVisit {
|
|
|
|
timeUUID, _ := uuid.NewV4()
|
|
securityVisit.Uuid = string(timeUUID.String())
|
|
securityVisit.CreateTime = time.Now()
|
|
securityVisit.UpdateTime = time.Now()
|
|
db := this.context.DB.Create(securityVisit)
|
|
this.PanicError(db.Error)
|
|
|
|
return securityVisit
|
|
}
|
|
|
|
//修改一条记录
|
|
func (this *SecurityVisitDao) Save(securityVisit *SecurityVisit) *SecurityVisit {
|
|
|
|
securityVisit.UpdateTime = time.Now()
|
|
db := this.context.DB.Save(securityVisit)
|
|
this.PanicError(db.Error)
|
|
|
|
return securityVisit
|
|
}
|
|
|
|
//删除一条记录
|
|
func (this *SecurityVisitDao) Delete(securityVisit *SecurityVisit) {
|
|
|
|
db := this.context.DB.Delete(&securityVisit)
|
|
this.PanicError(db.Error)
|
|
}
|