80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
|
|
"cidservice/model/entity"
|
|
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
var Tenant = tenantDao{}
|
|
|
|
type tenantDao struct{}
|
|
|
|
// Create 创建租户
|
|
func (d *tenantDao) Create(ctx context.Context, tenant *entity.Tenant) (id int64, err error) {
|
|
result, err := g.DB().Model("tenant").Ctx(ctx).Insert(tenant)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
id, _ = result.LastInsertId()
|
|
return id, nil
|
|
}
|
|
|
|
// GetByID 根据ID获取租户
|
|
func (d *tenantDao) GetByID(ctx context.Context, id int64) (tenant *entity.Tenant, err error) {
|
|
err = g.DB().Model("tenant").Ctx(ctx).Where("id = ? AND is_deleted = ?", id, false).Scan(&tenant)
|
|
return
|
|
}
|
|
|
|
// GetByCode 根据编码获取租户
|
|
func (d *tenantDao) GetByCode(ctx context.Context, code string) (tenant *entity.Tenant, err error) {
|
|
err = g.DB().Model("tenant").Ctx(ctx).Where("code = ? AND is_deleted = ?", code, false).Scan(&tenant)
|
|
return
|
|
}
|
|
|
|
// UpdateFields 更新租户字段
|
|
func (d *tenantDao) UpdateFields(ctx context.Context, id int64, updateData *entity.Tenant) (affected int64, err error) {
|
|
result, err := g.DB().Model("tenant").Ctx(ctx).Where("id = ? AND is_deleted = ?", id, false).Update(updateData)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
rowsAffected, _ := result.RowsAffected()
|
|
return rowsAffected, nil
|
|
}
|
|
|
|
// Delete 删除租户
|
|
func (d *tenantDao) Delete(ctx context.Context, id int64) (affected int64, err error) {
|
|
result, err := g.DB().Model("tenant").Ctx(ctx).Where("id = ?", id).Update(gdb.Map{"is_deleted": true})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
rowsAffected, _ := result.RowsAffected()
|
|
return rowsAffected, nil
|
|
}
|
|
|
|
// List 获取租户列表
|
|
func (d *tenantDao) List(ctx context.Context, page, size int, filter map[string]interface{}) (list []*entity.Tenant, total int64, err error) {
|
|
model := g.DB().Model("tenant").Ctx(ctx).Where("is_deleted = ?", false)
|
|
|
|
// 应用过滤条件
|
|
if filter != nil {
|
|
for key, value := range filter {
|
|
model = model.Where(key, value)
|
|
}
|
|
}
|
|
|
|
// 获取总数
|
|
count, err := model.Count()
|
|
if err != nil {
|
|
return
|
|
}
|
|
total = int64(count)
|
|
|
|
// 分页查询
|
|
err = model.Page(page, size).OrderDesc("created_at").Scan(&list)
|
|
return
|
|
}
|