diff --git a/model/gmodel/ldap_department_gen.go b/model/gmodel/ldap_department_gen.go index 77dca379..533341e3 100644 --- a/model/gmodel/ldap_department_gen.go +++ b/model/gmodel/ldap_department_gen.go @@ -1,8 +1,9 @@ package gmodel import ( - "gorm.io/gorm" "time" + + "gorm.io/gorm" ) // ldap_department 部门表 diff --git a/model/gmodel/ldap_department_logic.go b/model/gmodel/ldap_department_logic.go index 8914a76d..2a043b75 100644 --- a/model/gmodel/ldap_department_logic.go +++ b/model/gmodel/ldap_department_logic.go @@ -3,42 +3,45 @@ package gmodel import ( "context" "errors" + "gorm.io/gorm" ) -//获取列表 -func (d *LdapDepartmentModel)GetAll(ctx context.Context,sort string)(resp []LdapDepartment,total int64,err error){ +// 获取列表 +func (d *LdapDepartmentModel) GetAll(ctx context.Context, sort string) (resp []LdapDepartment, total int64, err error) { db := d.db.WithContext(ctx).Model(&LdapDepartment{}) - if sort != ""{ + if sort != "" { db = db.Order(sort) } - if err = db.Count(&total).Error;err != nil{ + if err = db.Count(&total).Error; err != nil { return nil, 0, err } err = db.Find(&resp).Error return resp, total, err } -func (d *LdapDepartmentModel)FindOne(ctx context.Context,id int64)(resp *LdapDepartment,err error){ - err = d.db.WithContext(ctx).Model(&LdapDepartment{}).Where("id = ?",id).Take(&resp).Error - return resp,err +func (d *LdapDepartmentModel) FindOne(ctx context.Context, id int64) (resp *LdapDepartment, err error) { + err = d.db.WithContext(ctx).Model(&LdapDepartment{}).Where("id = ?", id).Take(&resp).Error + return resp, err } -//更新 -func (d *LdapDepartmentModel)Update(ctx context.Context,id int64,data *LdapDepartment)error{ - return d.db.WithContext(ctx).Model(&LdapDepartment{}).Where("id = ?",id).Updates(&data).Error + +// 更新 +func (d *LdapDepartmentModel) Update(ctx context.Context, id int64, data *LdapDepartment) error { + return d.db.WithContext(ctx).Model(&LdapDepartment{}).Where("id = ?", id).Updates(&data).Error } -//创建 -func (d *LdapDepartmentModel)Create(ctx context.Context,data *LdapDepartment)error{ + +// 创建 +func (d *LdapDepartmentModel) Create(ctx context.Context, data *LdapDepartment) error { return d.db.WithContext(ctx).Model(&LdapDepartment{}).Create(&data).Error } -func (d *LdapDepartmentModel)CreateOrUpdate(ctx context.Context,id int64,data *LdapDepartment)error{ - _,err := d.FindOne(ctx,id) - if err != nil{ - if errors.Is(err,gorm.ErrRecordNotFound){ - return d.Create(ctx,data) +func (d *LdapDepartmentModel) CreateOrUpdate(ctx context.Context, id int64, data *LdapDepartment) error { + _, err := d.FindOne(ctx, id) + if err != nil { + if errors.Is(err, gorm.ErrRecordNotFound) { + return d.Create(ctx, data) } return err } - return d.Update(ctx,id,data) -} \ No newline at end of file + return d.Update(ctx, id, data) +} diff --git a/model/gmodel/ldap_menus_logic.go b/model/gmodel/ldap_menus_logic.go index e68225aa..17085f00 100644 --- a/model/gmodel/ldap_menus_logic.go +++ b/model/gmodel/ldap_menus_logic.go @@ -1,2 +1,15 @@ package gmodel -// TODO: 使用model的属性做你想做的 \ No newline at end of file + +import "context" + +// TODO: 使用model的属性做你想做的 + +// 获取全部菜单 +func (d *LdapMenusModel) GetAll(ctx context.Context) (resp []LdapMenus, err error) { + var menus []LdapMenus + result := d.db.Model(&LdapMenus{}).Where("status = ?", 1).Find(&menus) + if result.Error != nil { + return nil, result.Error + } + return menus, nil +} diff --git a/server/ldap-admin/internal/config/config.go b/server/ldap-admin/internal/config/config.go index 72df7c16..0e94c4ca 100644 --- a/server/ldap-admin/internal/config/config.go +++ b/server/ldap-admin/internal/config/config.go @@ -2,6 +2,7 @@ package config import ( "fusenapi/server/ldap-admin/internal/types" + "github.com/zeromicro/go-zero/rest" ) diff --git a/server/ldap-admin/internal/logic/getdepartmentslogic.go b/server/ldap-admin/internal/logic/getdepartmentslogic.go index 2b4ff2f4..34e20d85 100644 --- a/server/ldap-admin/internal/logic/getdepartmentslogic.go +++ b/server/ldap-admin/internal/logic/getdepartmentslogic.go @@ -1,10 +1,11 @@ package logic import ( - "context" "fusenapi/utils/auth" "fusenapi/utils/basic" + "context" + "fusenapi/server/ldap-admin/internal/svc" "fusenapi/server/ldap-admin/internal/types" @@ -30,6 +31,7 @@ func NewGetDepartmentsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Ge // } func (l *GetDepartmentsLogic) GetDepartments(req *types.Request, userinfo *auth.UserInfo) (resp *basic.Response) { +<<<<<<< HEAD //todo 鉴权 。。。。 departList, _, err := l.svcCtx.AllModels.LdapDepartment.GetAll(l.ctx, "sort ASC") if err != nil {