98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
package dao
|
|
|
|
import (
|
|
"context"
|
|
|
|
"cid/model/entity"
|
|
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
// applicationDao 应用DAO
|
|
type applicationDao struct{}
|
|
|
|
var Application = &applicationDao{}
|
|
|
|
// Ctx 获取数据库上下文
|
|
func (d *applicationDao) Ctx(ctx context.Context) *gdb.Model {
|
|
return g.DB().Model("application").Ctx(ctx)
|
|
}
|
|
|
|
// Create 创建应用
|
|
func (d *applicationDao) Create(ctx context.Context, app *entity.Application) (int64, error) {
|
|
result, err := d.Ctx(ctx).Insert(app)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
id, _ := result.LastInsertId()
|
|
return id, nil
|
|
}
|
|
|
|
// GetByID 根据ID获取应用
|
|
func (d *applicationDao) GetByID(ctx context.Context, id int64) (*entity.Application, error) {
|
|
var app *entity.Application
|
|
err := d.Ctx(ctx).Where("id", id).Scan(&app)
|
|
return app, err
|
|
}
|
|
|
|
// GetByTenantID 根据租户ID获取应用列表
|
|
func (d *applicationDao) GetByTenantID(ctx context.Context, tenantID int64) ([]*entity.Application, error) {
|
|
var apps []*entity.Application
|
|
err := d.Ctx(ctx).Where("tenant_id", tenantID).Scan(&apps)
|
|
return apps, err
|
|
}
|
|
|
|
// GetByAPIKey 根据API密钥获取应用
|
|
func (d *applicationDao) GetByAPIKey(ctx context.Context, apiKey string) (*entity.Application, error) {
|
|
var app *entity.Application
|
|
err := d.Ctx(ctx).Where("app_key", apiKey).Scan(&app)
|
|
return app, err
|
|
}
|
|
|
|
// Update 更新应用
|
|
func (d *applicationDao) Update(ctx context.Context, app *entity.Application) error {
|
|
_, err := d.Ctx(ctx).Where("id", app.Id).Update(app)
|
|
return err
|
|
}
|
|
|
|
// Delete 删除应用
|
|
func (d *applicationDao) Delete(ctx context.Context, id int64) error {
|
|
_, err := d.Ctx(ctx).Where("id", id).Delete()
|
|
return err
|
|
}
|
|
|
|
// List 应用列表
|
|
func (d *applicationDao) List(ctx context.Context, tenantID int64, page, pageSize int) ([]*entity.Application, int, error) {
|
|
model := d.Ctx(ctx)
|
|
if tenantID > 0 {
|
|
model = model.Where("tenant_id", tenantID)
|
|
}
|
|
|
|
var apps []*entity.Application
|
|
err := model.Page(page, pageSize).OrderDesc("created_at").Scan(&apps)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
total, err := model.Count()
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
return apps, total, nil
|
|
}
|
|
|
|
// GetByName 根据名称获取应用
|
|
func (d *applicationDao) GetByName(ctx context.Context, name string) (*entity.Application, error) {
|
|
var app *entity.Application
|
|
err := d.Ctx(ctx).Where("name", name).Scan(&app)
|
|
return app, err
|
|
}
|
|
|
|
// UpdateFields 更新应用部分字段
|
|
func (d *applicationDao) UpdateFields(ctx context.Context, id int64, data *entity.Application) error {
|
|
_, err := d.Ctx(ctx).Where("id", id).Update(data)
|
|
return err
|
|
}
|