change router struct

internal/erpserver/handler/system/department/department.go

@@ -0,0 +1,152 @@
+package department
+
+import (
+	"net/http"
+
+	"management/internal/erpserver/model/dto"
+	"management/internal/erpserver/model/form"
+	systemmodel "management/internal/erpserver/model/system"
+	v1 "management/internal/erpserver/service/v1"
+	"management/internal/erpserver/templ/system/department"
+	"management/internal/pkg/binding"
+	"management/internal/pkg/convertor"
+	"management/internal/pkg/render"
+)
+
+type App struct {
+	render            render.Renderer
+	departmentService v1.DepartmentService
+}
+
+func NewApp(render render.Renderer, departmentService v1.DepartmentService) *App {
+	return &App{
+		render:            render,
+		departmentService: departmentService,
+	}
+}
+
+func (a *App) list(w http.ResponseWriter, r *http.Request) {
+	switch r.Method {
+	case http.MethodGet:
+		ctx := r.Context()
+		a.render.Render(ctx, w, department.List(ctx))
+	case http.MethodPost:
+		var q dto.SearchDto
+		q.SearchTimeBegin, q.SearchTimeEnd = convertor.DefaultStartTimeAndEndTime(r.PostFormValue("timeBegin"), r.PostFormValue("timeEnd"))
+		q.SearchStatus = convertor.ConvertInt(r.PostFormValue("status"), 9999)
+		q.SearchParentID = convertor.ConvertInt(r.PostFormValue("parentId"), 0)
+		q.SearchName = r.PostFormValue("name")
+		q.SearchID = convertor.ConvertInt[int64](r.PostFormValue("id"), 0)
+		q.Page = convertor.ConvertInt(r.PostFormValue("page"), 1)
+		q.Rows = convertor.ConvertInt(r.PostFormValue("rows"), 10)
+		res, count, err := a.departmentService.List(r.Context(), q)
+		if err != nil {
+			a.render.JSONErr(w, err.Error())
+			return
+		}
+
+		data := render.ResponseList{
+			Code:    0,
+			Message: "ok",
+			Count:   count,
+			Data:    res,
+		}
+		a.render.JSON(w, data)
+	default:
+		http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
+	}
+}
+
+func (a *App) add(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	a.render.Render(ctx, w, department.Edit(ctx, &systemmodel.Department{Sort: 6666}))
+}
+
+func (a *App) addChildren(w http.ResponseWriter, r *http.Request) {
+	vars := r.URL.Query()
+	parentID := convertor.QueryInt[int32](vars, "parentID", 0)
+	vm := &systemmodel.Department{ParentID: parentID, Sort: 6666}
+	ctx := r.Context()
+	a.render.Render(ctx, w, department.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.Department{Sort: 6666}
+	if id > 0 {
+		vm, _ = a.departmentService.Get(r.Context(), id)
+	}
+	a.render.Render(ctx, w, department.Edit(ctx, vm))
+}
+
+func (a *App) save(w http.ResponseWriter, r *http.Request) {
+	var req form.Department
+	if err := binding.Form.Bind(r, &req); err != nil {
+		a.render.JSONErr(w, binding.ValidatorErrors(err))
+		return
+	}
+
+	ctx := r.Context()
+	if *req.ID == 0 {
+		err := a.departmentService.Create(ctx, &req)
+		if err != nil {
+			a.render.JSONErr(w, err.Error())
+			return
+		}
+		a.render.JSONOk(w, "添加成功")
+	} else {
+		err := a.departmentService.Update(ctx, &req)
+		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.departmentService.Tree(ctx, 0)
+		if err != nil {
+			a.render.JSONErr(w, err.Error())
+			return
+		}
+
+		a.render.JSON(w, res)
+	} else if t == "xm_select_tree" {
+		res, err := a.departmentService.XmSelectTree(ctx, 0)
+		if err != nil {
+			a.render.JSONErr(w, err.Error())
+			return
+		}
+
+		a.render.JSON(w, res)
+	}
+}
+
+func (a *App) refreshCache(w http.ResponseWriter, r *http.Request) {
+	err := a.departmentService.RefreshCache(r.Context())
+	if err != nil {
+		a.render.JSONErr(w, err.Error())
+		return
+	}
+
+	a.render.JSONOk(w, "缓存刷新成功")
+}
+
+func (a *App) rebuildParentPath(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	err := a.departmentService.RebuildParentPath(ctx)
+	if err != nil {
+		a.render.JSONErr(w, err.Error())
+		return
+	}
+
+	a.render.JSONOk(w, "重建成功")
+}

internal/erpserver/handler/system/department/route.go

@@ -0,0 +1,36 @@
+package department
+
+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
+	DepartmentService v1.DepartmentService
+}
+
+func Routes(r chi.Router, cfg Config) {
+	app := NewApp(cfg.Render, cfg.DepartmentService)
+
+	r.Route("/department", 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)
+		r.Post("/rebuild_parent_path", app.rebuildParentPath)
+	})
+}