You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
406 lines
13 KiB
406 lines
13 KiB
// Copyright (c) Shenyang Leading Edge Intelligent Technology Co., Ltd. All rights reserved.
|
|
|
|
package implments
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/go-xorm/xorm"
|
|
"LEIT_PM/grmi"
|
|
meta "LEIT_PM/meta/base"
|
|
model "LEIT_PM/models/base"
|
|
"xorm.io/core"
|
|
)
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Struct Name : UserDAOImplement
|
|
*-----------------------------------------------------------------------------
|
|
*
|
|
* @Description : User的数据访问对象实现
|
|
*
|
|
* @Author : 代码生成器创建
|
|
*
|
|
* @Date : 2021-08-20 13:39:52
|
|
*
|
|
******************************************************************************/
|
|
type UserDAOImplement struct {
|
|
session *xorm.Session
|
|
meta *grmi.Entity
|
|
plantNr int
|
|
userid string
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Function Name : NewUserDAOImplement
|
|
*-----------------------------------------------------------------------------
|
|
*
|
|
* @Description : 创建一个UserDAOImplement实例
|
|
*
|
|
* @Function Parameters : xorm会话
|
|
*
|
|
* @Function Parameters : 基本主键
|
|
*
|
|
* @Return Value : UserDAOImplement实例
|
|
*
|
|
* @Author : 代码生成器创建
|
|
*
|
|
* @Date : 2021-08-20 13:39:52
|
|
*
|
|
******************************************************************************/
|
|
func NewUserDAOImplement(session *xorm.Session, plantNr int, userid string) *UserDAOImplement {
|
|
return &UserDAOImplement{session, &meta.User, plantNr, userid}
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.InsertOne
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) InsertOne(entity *model.User) error {
|
|
entity.PlantNr = impl.plantNr
|
|
entity.LastUser = impl.userid
|
|
_, err := impl.session.Table(impl.meta.TableName).Insert(entity)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.DeleteOne
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) DeleteOne(userId string) error {
|
|
_, err := impl.session.Table(impl.meta.TableName).ID(core.PK{impl.plantNr, userId}).Delete(impl.meta.Indicator)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.SelectOne
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) SelectOne(userId string) (*model.User, error) {
|
|
var data model.User
|
|
ok, err := impl.session.Table(impl.meta.TableName).ID(core.PK{impl.plantNr, userId}).Get(&data)
|
|
if err != nil {
|
|
return nil, grmi.NewDataBaseError(err)
|
|
}
|
|
if !ok {
|
|
return nil, nil
|
|
}
|
|
return &data, nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.UpdateOne
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) UpdateOne(entity *model.User) error {
|
|
entity.PlantNr = impl.plantNr
|
|
entity.LastUser = impl.userid
|
|
|
|
_, err := impl.session.Table(impl.meta.TableName).Cols("PlantNr,RoleId,Role,NickName,Name,Status,Gender,UserId,Email,Phone").ID(entity.GetKey()).Update(entity)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.UpdatePwd
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) UpdatePwd(entity *model.User) error {
|
|
entity.PlantNr = impl.plantNr
|
|
entity.LastUser = impl.userid
|
|
|
|
_, err := impl.session.Table(impl.meta.TableName).Cols("Password").ID(entity.GetKey()).Update(entity)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.Insert
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) Insert(entities *[]model.User) error {
|
|
for i := 0; i < len(*entities); i++ {
|
|
entity := &(*entities)[i]
|
|
entity.PlantNr = impl.plantNr
|
|
entity.LastUser = impl.userid
|
|
}
|
|
_, err := impl.session.Table(impl.meta.TableName).Insert(entities)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.Delete
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) Delete(entities *[]model.User) error {
|
|
for _, entity := range *entities {
|
|
entity.PlantNr = impl.plantNr
|
|
_, err := impl.session.Table(impl.meta.TableName).ID(entity.GetKey()).Delete(impl.meta.Indicator)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.DeleteWhere
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) DeleteWhere(predicates []grmi.Predicate) error {
|
|
|
|
parameters := []interface{}{impl.plantNr}
|
|
where := fmt.Sprintf("%s = ?", meta.User_PlantNr.ColumnName)
|
|
|
|
session := impl.session.Table(impl.meta.TableName)
|
|
if predicates != nil {
|
|
for _, predicate := range predicates {
|
|
where += predicate.Build()
|
|
parameters = append(parameters, predicate.Values...)
|
|
}
|
|
session = session.Where(where, parameters...)
|
|
} else {
|
|
session = session.Where(where, parameters...)
|
|
}
|
|
|
|
_, err := session.Delete(impl.meta.Indicator)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.Select
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) Select(predicates []grmi.Predicate, orderByFields []grmi.Field) ([]model.User, error) {
|
|
|
|
parameters := []interface{}{impl.plantNr}
|
|
where := fmt.Sprintf("%s = ?", meta.User_PlantNr.ColumnName)
|
|
|
|
session := impl.session.Table(impl.meta.TableName)
|
|
if predicates != nil {
|
|
for _, predicate := range predicates {
|
|
where += predicate.Build()
|
|
parameters = append(parameters, predicate.Values...)
|
|
}
|
|
session = session.Where(where, parameters...)
|
|
} else {
|
|
session = session.Where(where, parameters...)
|
|
}
|
|
|
|
session = session.OrderBy(meta.User_PlantNr.ColumnName)
|
|
if orderByFields != nil {
|
|
for _, field := range orderByFields {
|
|
session = session.OrderBy(field.ColumnName)
|
|
}
|
|
}
|
|
|
|
data := make([]model.User, 0, 10)
|
|
err := session.Find(&data)
|
|
if err != nil {
|
|
return nil, grmi.NewDataBaseError(err)
|
|
}
|
|
//置空密码
|
|
for k, _ := range data {
|
|
data[k].Password = ""
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.SelectAndPaging
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) SelectAndPaging(paging *grmi.Paging, predicates []grmi.Predicate, orderByFields []grmi.Field) (grmi.PagingResult, error) {
|
|
|
|
parameters := []interface{}{impl.plantNr}
|
|
where := fmt.Sprintf("%s = ?", meta.User_PlantNr.ColumnName)
|
|
|
|
session := impl.session.Table(impl.meta.TableName)
|
|
if predicates != nil {
|
|
for _, predicate := range predicates {
|
|
where += predicate.Build()
|
|
parameters = append(parameters, predicate.Values...)
|
|
}
|
|
session = session.Where(where, parameters...)
|
|
} else {
|
|
session = session.Where(where, parameters...)
|
|
}
|
|
|
|
count, err := session.Count(impl.meta.Indicator)
|
|
if err != nil {
|
|
return grmi.EmptyPagingResult, err
|
|
}
|
|
|
|
orderBy := " order by " + meta.User_PlantNr.ColumnName
|
|
if orderByFields != nil {
|
|
for _, field := range orderByFields {
|
|
if orderBy == " order by " {
|
|
orderBy += field.ColumnName
|
|
} else {
|
|
orderBy += ", " + field.ColumnName
|
|
}
|
|
}
|
|
}
|
|
parameters = append(parameters, paging.Offset(), paging.Size)
|
|
data := make([]model.User, 0, 10)
|
|
where = " where " + where
|
|
err = impl.session.Table(impl.meta.TableName).SQL("select * from ["+impl.meta.TableName+"]"+where+orderBy+" offset ? row fetch next ? row only", parameters...).Find(&data)
|
|
if err != nil {
|
|
return grmi.EmptyPagingResult, err
|
|
}
|
|
//置空密码
|
|
for k, _ := range data {
|
|
data[k].Password = ""
|
|
}
|
|
return grmi.PagingResult{Records: data, Count: count, PageNumber: paging.Number, PageSize: paging.Size}, nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.Update
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) Update(entities *[]model.User) error {
|
|
for i := 0; i < len(*entities); i++ {
|
|
entity := &(*entities)[i]
|
|
entity.PlantNr = impl.plantNr
|
|
entity.LastUser = impl.userid
|
|
_, err := impl.session.Table(impl.meta.TableName).Cols(impl.meta.UpdatingMembers...).ID(entity.GetKey()).Update(entity)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.UpdateWhere
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) UpdateWhere(predicates []grmi.Predicate, entity *model.User, columns ...string) error {
|
|
|
|
parameters := []interface{}{impl.plantNr}
|
|
where := fmt.Sprintf("%s = ?", meta.User_PlantNr.ColumnName)
|
|
|
|
session := impl.session.Table(impl.meta.TableName)
|
|
if predicates != nil {
|
|
for _, predicate := range predicates {
|
|
where += predicate.Build()
|
|
parameters = append(parameters, predicate.Values...)
|
|
}
|
|
session = session.Where(where, parameters...)
|
|
} else {
|
|
session = session.Where(where, parameters...)
|
|
}
|
|
|
|
columns = append(columns, meta.User_LastUser.ColumnName)
|
|
entity.LastUser = impl.userid
|
|
_, err := impl.session.Table(impl.meta.TableName).Cols(columns...).Update(entity)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.GetUserByUsername
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) GetUserByUsername(plantNr int, userId string) (*model.User, error) {
|
|
var data model.User
|
|
ok, err := impl.session.Table(impl.meta.TableName).Where("PlantNr = ? and UserId = ?", plantNr, userId).Get(&data)
|
|
if err != nil {
|
|
return nil, grmi.NewDataBaseError(err)
|
|
}
|
|
if !ok {
|
|
return nil, nil
|
|
}
|
|
return &data, nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.GetUserInfo
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) GetUserInfo(plantNr int, userId string) (*model.UserViewInfo, error) {
|
|
var data model.UserViewInfo
|
|
var datalist model.User
|
|
ok, err := impl.session.Table(impl.meta.TableName).Where("PlantNr = ? and UserId = ?", plantNr, userId).Get(&datalist)
|
|
if err != nil {
|
|
return nil, grmi.NewDataBaseError(err)
|
|
}
|
|
if !ok {
|
|
return nil, nil
|
|
}
|
|
data.PlantNr = datalist.PlantNr
|
|
data.RoleId = datalist.RoleId
|
|
data.Role = datalist.Role
|
|
data.NickName = datalist.NickName
|
|
data.Name = datalist.Name
|
|
data.Status = datalist.Status
|
|
data.Gender = datalist.Gender
|
|
data.UserId = datalist.UserId
|
|
data.Email = datalist.Email
|
|
data.Phone = datalist.Phone
|
|
return &data, nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.UpdateStatus
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) UpdateStatus(entity *model.User) error {
|
|
entity.PlantNr = impl.plantNr
|
|
entity.LastUser = impl.userid
|
|
_, err := impl.session.Table(impl.meta.TableName).Cols("Status").ID(entity.GetKey()).Update(entity)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
/******************************************************************************
|
|
*
|
|
* @Reference LEIT_PM/dao/base/UserDAO.UpdatePic
|
|
*
|
|
******************************************************************************/
|
|
func (impl *UserDAOImplement) UpdatePic(entity *model.User) error {
|
|
entity.PlantNr = impl.plantNr
|
|
entity.LastUser = impl.userid
|
|
_, err := impl.session.Table(impl.meta.TableName).Cols("avatar").ID(entity.GetKey()).Update(entity)
|
|
if err != nil {
|
|
return grmi.NewDataBaseError(err)
|
|
}
|
|
return nil
|
|
}
|