fix

model/gmodel/ldap_department_gen.go

@@ -1,8 +1,9 @@
 package gmodel
 
 import (
-	"gorm.io/gorm"
 	"time"
+
+	"gorm.io/gorm"
 )
 
 // ldap_department 部门表

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)
-}
+	return d.Update(ctx, id, data)
+}

model/gmodel/ldap_menus_logic.go

@@ -1,2 +1,15 @@
 package gmodel
-// TODO: 使用model的属性做你想做的
+
+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
+}