change router struct
This commit is contained in:
227
internal/erpserver/handler/system/menu/menu.go
Normal file
227
internal/erpserver/handler/system/menu/menu.go
Normal file
@@ -0,0 +1,227 @@
|
||||
package menu
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
systemmodel "management/internal/erpserver/model/system"
|
||||
v1 "management/internal/erpserver/service/v1"
|
||||
"management/internal/erpserver/templ/system/menu"
|
||||
"management/internal/pkg/convertor"
|
||||
"management/internal/pkg/database"
|
||||
"management/internal/pkg/mid"
|
||||
"management/internal/pkg/render"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
const style = "layui-btn-primary layui-btn-sm"
|
||||
|
||||
type App struct {
|
||||
render render.Renderer
|
||||
menuService v1.MenuService
|
||||
}
|
||||
|
||||
func NewApp(render render.Renderer, menuService v1.MenuService) *App {
|
||||
return &App{
|
||||
render: render,
|
||||
menuService: menuService,
|
||||
}
|
||||
}
|
||||
|
||||
func (a *App) menus(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
user := mid.GetUser(ctx)
|
||||
menus, err := a.menuService.OwerMenus(ctx, user.RoleID)
|
||||
if err != nil {
|
||||
a.render.JSONErr(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
a.render.JSON(w, menus)
|
||||
}
|
||||
|
||||
func (a *App) list(w http.ResponseWriter, r *http.Request) {
|
||||
switch r.Method {
|
||||
case http.MethodGet:
|
||||
ctx := r.Context()
|
||||
a.render.Render(ctx, w, menu.List(ctx))
|
||||
case http.MethodPost:
|
||||
res, err := a.menuService.ListMenuTree(r.Context())
|
||||
if err != nil {
|
||||
a.render.JSONErr(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
data := render.ResponseList{
|
||||
Code: 0,
|
||||
Message: "ok",
|
||||
Count: 0,
|
||||
Data: res,
|
||||
}
|
||||
a.render.JSON(w, data)
|
||||
default:
|
||||
a.render.JSONErr(w, "method not allowed")
|
||||
}
|
||||
}
|
||||
|
||||
func (a *App) add(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
a.render.Render(ctx, w, menu.Edit(ctx, &systemmodel.Menu{Style: style, Visible: true, Sort: 6666}))
|
||||
}
|
||||
|
||||
func (a *App) addChildren(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
vars := r.URL.Query()
|
||||
parentID := convertor.QueryInt[int32](vars, "parentID", 0)
|
||||
vm := &systemmodel.Menu{ParentID: parentID, Style: style, Visible: true, Sort: 6666}
|
||||
parent, err := a.menuService.Get(ctx, parentID)
|
||||
if err == nil {
|
||||
if parent.Type == "node" {
|
||||
vm.Type = "menu"
|
||||
} else if parent.Type == "menu" {
|
||||
vm.Type = "btn"
|
||||
}
|
||||
}
|
||||
a.render.Render(ctx, w, menu.Edit(ctx, vm))
|
||||
}
|
||||
|
||||
func (a *App) edit(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
vars := r.URL.Query()
|
||||
id := convertor.QueryInt[int32](vars, "id", 0)
|
||||
vm := &systemmodel.Menu{Style: style, Sort: 6666}
|
||||
if id > 0 {
|
||||
vm, _ = a.menuService.Get(r.Context(), id)
|
||||
}
|
||||
a.render.Render(ctx, w, menu.Edit(ctx, vm))
|
||||
}
|
||||
|
||||
func (a *App) save(w http.ResponseWriter, r *http.Request) {
|
||||
id := convertor.ConvertInt[int32](r.PostFormValue("ID"), 0)
|
||||
name := r.PostFormValue("Name")
|
||||
displayName := r.PostFormValue("DisplayName")
|
||||
t := r.PostFormValue("Type")
|
||||
url := r.PostFormValue("Url")
|
||||
if len(url) == 0 {
|
||||
url = uuid.Must(uuid.NewRandom()).String()
|
||||
}
|
||||
avatar := r.PostFormValue("Avatar")
|
||||
style := r.PostFormValue("Style")
|
||||
parentID := convertor.ConvertInt[int32](r.PostFormValue("ParentID"), 0)
|
||||
visible := convertor.ConvertBool(r.PostFormValue("Visible"), false)
|
||||
isList := convertor.ConvertBool(r.PostFormValue("IsList"), false)
|
||||
sort := convertor.ConvertInt[int32](r.PostFormValue("Sort"), 6666)
|
||||
status := convertor.ConvertInt[int32](r.PostFormValue("Status"), 0)
|
||||
|
||||
ctx := r.Context()
|
||||
if len(avatar) > 0 && !strings.Contains(avatar, "layui-icon ") {
|
||||
avatar = "layui-icon " + avatar
|
||||
}
|
||||
|
||||
parentPath := ""
|
||||
if parentID > 0 {
|
||||
parent, err := a.menuService.Get(ctx, parentID)
|
||||
if err != nil {
|
||||
a.render.JSONErr(w, err.Error())
|
||||
return
|
||||
}
|
||||
parentPath = parent.ParentPath + "," + strconv.Itoa(int(parentID)) + ","
|
||||
parentPath = strings.ReplaceAll(parentPath, ",,", ",")
|
||||
}
|
||||
|
||||
if id == 0 {
|
||||
arg := &systemmodel.Menu{
|
||||
Name: name,
|
||||
DisplayName: displayName,
|
||||
Url: url,
|
||||
Type: t,
|
||||
ParentID: parentID,
|
||||
ParentPath: parentPath,
|
||||
Avatar: avatar,
|
||||
Style: style,
|
||||
Visible: visible,
|
||||
IsList: isList,
|
||||
Status: status,
|
||||
Sort: sort,
|
||||
CreatedAt: time.Now(),
|
||||
UpdatedAt: time.Now(),
|
||||
}
|
||||
err := a.menuService.Create(ctx, arg)
|
||||
if err != nil {
|
||||
if database.IsUniqueViolation(err) {
|
||||
a.render.JSONErr(w, "菜单已存在")
|
||||
return
|
||||
}
|
||||
a.render.JSONErr(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
a.render.JSONOk(w, "添加成功")
|
||||
} else {
|
||||
res, err := a.menuService.Get(ctx, id)
|
||||
if err != nil {
|
||||
a.render.JSONErr(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
res.Name = name
|
||||
res.DisplayName = displayName
|
||||
res.Url = url
|
||||
res.Type = t
|
||||
res.ParentID = parentID
|
||||
res.ParentPath = parentPath
|
||||
res.Avatar = avatar
|
||||
res.Style = style
|
||||
res.Visible = visible
|
||||
res.IsList = isList
|
||||
res.Status = status
|
||||
res.Sort = sort
|
||||
res.UpdatedAt = time.Now()
|
||||
err = a.menuService.Update(ctx, res)
|
||||
if err != nil {
|
||||
a.render.JSONErr(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
a.render.JSONOk(w, "更新成功")
|
||||
}
|
||||
}
|
||||
|
||||
func (a *App) data(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
vars := r.URL.Query()
|
||||
t := vars.Get("type")
|
||||
if t == "tree" {
|
||||
res, err := a.menuService.Tree(ctx, 0)
|
||||
if err != nil {
|
||||
a.render.JSONErr(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
a.render.JSON(w, res)
|
||||
return
|
||||
} else if t == "xm_select_tree" {
|
||||
res, err := a.menuService.XmSelectTree(ctx, 0)
|
||||
if err != nil {
|
||||
a.render.JSONErr(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
a.render.JSON(w, res)
|
||||
return
|
||||
}
|
||||
a.render.JSON(w, nil)
|
||||
}
|
||||
|
||||
func (a *App) refreshCache(w http.ResponseWriter, r *http.Request) {
|
||||
err := a.menuService.RefreshCache(r.Context())
|
||||
if err != nil {
|
||||
a.render.JSONErr(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
a.render.JSONOk(w, "缓存刷新成功")
|
||||
}
|
||||
35
internal/erpserver/handler/system/menu/route.go
Normal file
35
internal/erpserver/handler/system/menu/route.go
Normal file
@@ -0,0 +1,35 @@
|
||||
package menu
|
||||
|
||||
import (
|
||||
v1 "management/internal/erpserver/service/v1"
|
||||
"management/internal/pkg/mid"
|
||||
"management/internal/pkg/render"
|
||||
"management/internal/pkg/session"
|
||||
|
||||
"github.com/drhin/logger"
|
||||
"github.com/go-chi/chi/v5"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Log *logger.Logger
|
||||
Sm session.Manager
|
||||
Render render.Renderer
|
||||
MenuService v1.MenuService
|
||||
}
|
||||
|
||||
func Routes(r chi.Router, cfg Config) {
|
||||
app := NewApp(cfg.Render, cfg.MenuService)
|
||||
|
||||
r.Get("/menus", app.menus)
|
||||
r.Route("/menu", func(r chi.Router) {
|
||||
r.Use(mid.Audit(cfg.Sm, cfg.Log))
|
||||
r.Get("/list", app.list)
|
||||
r.Post("/list", app.list)
|
||||
r.Get("/add", app.add)
|
||||
r.Get("/add_children", app.addChildren)
|
||||
r.Get("/edit", app.edit)
|
||||
r.Post("/save", app.save)
|
||||
r.Post("/data", app.data)
|
||||
r.Post("/refresh_cache", app.refreshCache)
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user