初始化项目

dao/tenant_dao.go

@@ -0,0 +1,79 @@
+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
+}