v2
This commit is contained in:
46
internal/erpserver/biz/biz.go
Normal file
46
internal/erpserver/biz/biz.go
Normal file
@@ -0,0 +1,46 @@
|
||||
package biz
|
||||
|
||||
import (
|
||||
db "management/internal/db/sqlc"
|
||||
commonv1 "management/internal/erpserver/biz/v1/common"
|
||||
systemv1 "management/internal/erpserver/biz/v1/system"
|
||||
"management/internal/pkg/redis"
|
||||
"management/internal/pkg/session"
|
||||
)
|
||||
|
||||
// IBiz 定义了业务层需要实现的方法.
|
||||
type IBiz interface {
|
||||
// 获取公共业务接口.
|
||||
CommonV1() commonv1.CommonBiz
|
||||
// 获取系统业务接口.
|
||||
SystemV1() systemv1.SystemBiz
|
||||
}
|
||||
|
||||
// biz 是 IBiz 的一个具体实现.
|
||||
type biz struct {
|
||||
store db.Store
|
||||
redis redis.IRedis
|
||||
session session.ISession
|
||||
}
|
||||
|
||||
// 确保 biz 实现了 IBiz 接口.
|
||||
var _ IBiz = (*biz)(nil)
|
||||
|
||||
// NewBiz 创建一个 IBiz 类型的实例.
|
||||
func NewBiz(store db.Store, redis redis.IRedis, session session.ISession) *biz {
|
||||
return &biz{
|
||||
store: store,
|
||||
redis: redis,
|
||||
session: session,
|
||||
}
|
||||
}
|
||||
|
||||
// CommonV1 返回一个实现了 CommonBiz 接口的实例.
|
||||
func (b *biz) CommonV1() commonv1.CommonBiz {
|
||||
return commonv1.New()
|
||||
}
|
||||
|
||||
// SystemV1 返回一个实现了 SystemBiz 接口的实例.
|
||||
func (b *biz) SystemV1() systemv1.SystemBiz {
|
||||
return systemv1.New(b.store, b.redis, b.session)
|
||||
}
|
||||
36
internal/erpserver/biz/v1/common/captcha.go
Normal file
36
internal/erpserver/biz/v1/common/captcha.go
Normal file
@@ -0,0 +1,36 @@
|
||||
package common
|
||||
|
||||
import (
|
||||
"github.com/mojocn/base64Captcha"
|
||||
)
|
||||
|
||||
// CaptchaBiz 定义处理验证码请求所需的方法.
|
||||
type CaptchaBiz interface {
|
||||
Generate(height int, width int, length int, maxSkew float64, dotCount int) (id, b64s, answer string, err error)
|
||||
Verify(id, answer string, clear bool) bool
|
||||
}
|
||||
|
||||
// captchaBiz 是 CaptchaBiz 接口的实现.
|
||||
type captchaBiz struct{}
|
||||
|
||||
// 确保 captchaBiz 实现了 CaptchaBiz 接口.
|
||||
var _ CaptchaBiz = (*captchaBiz)(nil)
|
||||
|
||||
func NewCaptcha() *captchaBiz {
|
||||
return &captchaBiz{}
|
||||
}
|
||||
|
||||
var captchaStore base64Captcha.Store = base64Captcha.DefaultMemStore
|
||||
|
||||
func (b *captchaBiz) Generate(height int, width int, length int, maxSkew float64, dotCount int) (id, b64s, answer string, err error) {
|
||||
driver := base64Captcha.NewDriverDigit(height, width, length, maxSkew, dotCount)
|
||||
// driver := base64Captcha.NewDriverString(config.File.Captcha.ImgHeight,
|
||||
// config.File.Captcha.ImgWidth,
|
||||
// 6, 1, keyLong, source, nil, nil, nil)
|
||||
cp := base64Captcha.NewCaptcha(driver, captchaStore)
|
||||
return cp.Generate()
|
||||
}
|
||||
|
||||
func (b *captchaBiz) Verify(id, answer string, clear bool) bool {
|
||||
return captchaStore.Verify(id, answer, clear)
|
||||
}
|
||||
17
internal/erpserver/biz/v1/common/common.go
Normal file
17
internal/erpserver/biz/v1/common/common.go
Normal file
@@ -0,0 +1,17 @@
|
||||
package common
|
||||
|
||||
type CommonBiz interface {
|
||||
CaptchaBiz() CaptchaBiz
|
||||
}
|
||||
|
||||
type commonBiz struct{}
|
||||
|
||||
var _ CommonBiz = (*commonBiz)(nil)
|
||||
|
||||
func New() *commonBiz {
|
||||
return &commonBiz{}
|
||||
}
|
||||
|
||||
func (b *commonBiz) CaptchaBiz() CaptchaBiz {
|
||||
return NewCaptcha()
|
||||
}
|
||||
31
internal/erpserver/biz/v1/system/audit.go
Normal file
31
internal/erpserver/biz/v1/system/audit.go
Normal file
@@ -0,0 +1,31 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
db "management/internal/db/sqlc"
|
||||
)
|
||||
|
||||
type AuditBiz interface {
|
||||
Create(ctx context.Context, arg *db.CreateSysAuditLogParams) error
|
||||
|
||||
AuditExpansion
|
||||
}
|
||||
|
||||
type AuditExpansion interface{}
|
||||
|
||||
type auditBiz struct {
|
||||
store db.Store
|
||||
}
|
||||
|
||||
var _ AuditBiz = (*auditBiz)(nil)
|
||||
|
||||
func NewAudit(store db.Store) *auditBiz {
|
||||
return &auditBiz{
|
||||
store: store,
|
||||
}
|
||||
}
|
||||
|
||||
func (b *auditBiz) Create(ctx context.Context, arg *db.CreateSysAuditLogParams) error {
|
||||
return b.store.CreateSysAuditLog(ctx, arg)
|
||||
}
|
||||
63
internal/erpserver/biz/v1/system/config.go
Normal file
63
internal/erpserver/biz/v1/system/config.go
Normal file
@@ -0,0 +1,63 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"time"
|
||||
|
||||
"management/internal/db/model/dto"
|
||||
db "management/internal/db/sqlc"
|
||||
"management/internal/global/keys"
|
||||
"management/internal/global/pearadmin"
|
||||
"management/internal/pkg/redis"
|
||||
"management/internal/pkg/session"
|
||||
)
|
||||
|
||||
type ConfigBiz interface {
|
||||
ConfigExpansion
|
||||
}
|
||||
|
||||
type ConfigExpansion interface {
|
||||
Pear(ctx context.Context) (*dto.PearConfig, error)
|
||||
}
|
||||
|
||||
type configBiz struct {
|
||||
store db.Store
|
||||
redis redis.IRedis
|
||||
session session.ISession
|
||||
}
|
||||
|
||||
var _ ConfigBiz = (*configBiz)(nil)
|
||||
|
||||
func NewConfig(store db.Store, redis redis.IRedis, session session.ISession) *configBiz {
|
||||
return &configBiz{
|
||||
store: store,
|
||||
redis: redis,
|
||||
session: session,
|
||||
}
|
||||
}
|
||||
|
||||
func (b *configBiz) Pear(ctx context.Context) (*dto.PearConfig, error) {
|
||||
// 判断redis是否存储
|
||||
key := keys.GetManageKey(ctx, keys.PearAdmin)
|
||||
bs, err := b.redis.GetBytes(ctx, key)
|
||||
if err == nil {
|
||||
var res *dto.PearConfig
|
||||
if err := json.Unmarshal(bs, &res); err == nil {
|
||||
return res, nil
|
||||
}
|
||||
}
|
||||
|
||||
conf, err := b.store.GetSysConfigByKey(ctx, pearadmin.PearKey)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var pear dto.PearConfig
|
||||
if err := json.Unmarshal(conf.Value, &pear); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
_ = b.redis.Set(ctx, key, conf.Value, time.Hour*6)
|
||||
return &pear, nil
|
||||
}
|
||||
183
internal/erpserver/biz/v1/system/department.go
Normal file
183
internal/erpserver/biz/v1/system/department.go
Normal file
@@ -0,0 +1,183 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"management/internal/db/model/dto"
|
||||
db "management/internal/db/sqlc"
|
||||
"management/internal/erpserver/model/view"
|
||||
"management/internal/global/keys"
|
||||
"management/internal/pkg/redis"
|
||||
"management/internal/pkg/session"
|
||||
)
|
||||
|
||||
type DepartmentBiz interface {
|
||||
Create(ctx context.Context, arg *db.CreateSysDepartmentParams) (*db.SysDepartment, error)
|
||||
Update(ctx context.Context, arg *db.UpdateSysDepartmentParams) (*db.SysDepartment, error)
|
||||
All(ctx context.Context) ([]*db.SysDepartment, error)
|
||||
List(ctx context.Context, q dto.SearchDto) ([]*db.SysDepartment, int64, error)
|
||||
Get(ctx context.Context, id int32) (*db.SysDepartment, error)
|
||||
Refresh(ctx context.Context) ([]*db.SysDepartment, error)
|
||||
RebuildParentPath(ctx context.Context) error
|
||||
|
||||
Tree(ctx context.Context, id int32) ([]*view.LayuiTree, error)
|
||||
XmSelect(ctx context.Context, id int32) ([]*view.XmSelectTree, error)
|
||||
|
||||
DepartmentExpansion
|
||||
}
|
||||
|
||||
type DepartmentExpansion interface{}
|
||||
|
||||
type departmentBiz struct {
|
||||
store db.Store
|
||||
redis redis.IRedis
|
||||
session session.ISession
|
||||
}
|
||||
|
||||
var _ DepartmentBiz = (*departmentBiz)(nil)
|
||||
|
||||
func NewDepartment(store db.Store, redis redis.IRedis, session session.ISession) *departmentBiz {
|
||||
return &departmentBiz{
|
||||
store: store,
|
||||
redis: redis,
|
||||
session: session,
|
||||
}
|
||||
}
|
||||
|
||||
func (b *departmentBiz) All(ctx context.Context) ([]*db.SysDepartment, error) {
|
||||
key := keys.GetManageKey(ctx, keys.AllDepartments)
|
||||
bs, err := redis.GetBytes(ctx, key)
|
||||
if err == nil {
|
||||
var res []*db.SysDepartment
|
||||
if err := json.Unmarshal(bs, &res); err == nil {
|
||||
return res, nil
|
||||
}
|
||||
}
|
||||
|
||||
return b.Refresh(ctx)
|
||||
}
|
||||
|
||||
func (b *departmentBiz) List(ctx context.Context, q dto.SearchDto) ([]*db.SysDepartment, int64, error) {
|
||||
countArg := &db.CountSysDepartmentConditionParams{
|
||||
IsStatus: q.SearchStatus != 9999,
|
||||
Status: int32(q.SearchStatus),
|
||||
IsID: q.SearchID != 0,
|
||||
ID: int32(q.SearchID),
|
||||
IsParentID: q.SearchParentID != 0,
|
||||
ParentID: int32(q.SearchParentID),
|
||||
Name: q.SearchName,
|
||||
}
|
||||
|
||||
dataArg := &db.ListSysDepartmentConditionParams{
|
||||
IsStatus: q.SearchStatus != 9999,
|
||||
Status: int32(q.SearchStatus),
|
||||
IsID: q.SearchID != 0,
|
||||
ID: int32(q.SearchID),
|
||||
IsParentID: q.SearchParentID != 0,
|
||||
ParentID: int32(q.SearchParentID),
|
||||
Name: q.SearchName,
|
||||
Skip: (int32(q.Page) - 1) * int32(q.Rows),
|
||||
Size: int32(q.Rows),
|
||||
}
|
||||
count, err := b.store.CountSysDepartmentCondition(ctx, countArg)
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
departs, err := b.store.ListSysDepartmentCondition(ctx, dataArg)
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
return departs, count, nil
|
||||
}
|
||||
|
||||
func (b *departmentBiz) Get(ctx context.Context, id int32) (*db.SysDepartment, error) {
|
||||
return b.store.GetSysDepartment(ctx, id)
|
||||
}
|
||||
|
||||
func (b *departmentBiz) Create(ctx context.Context, arg *db.CreateSysDepartmentParams) (*db.SysDepartment, error) {
|
||||
return b.store.CreateSysDepartment(ctx, arg)
|
||||
}
|
||||
|
||||
func (b *departmentBiz) Update(ctx context.Context, arg *db.UpdateSysDepartmentParams) (*db.SysDepartment, error) {
|
||||
return b.store.UpdateSysDepartment(ctx, arg)
|
||||
}
|
||||
|
||||
func (b *departmentBiz) Refresh(ctx context.Context) ([]*db.SysDepartment, error) {
|
||||
all, err := b.store.AllSysDepartment(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
bs, err := json.Marshal(all)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
key := keys.GetManageKey(ctx, keys.AllDepartments)
|
||||
err = redis.Set(ctx, key, bs, time.Hour*6)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return all, nil
|
||||
}
|
||||
|
||||
func (h *departmentBiz) RebuildParentPath(ctx context.Context) error {
|
||||
return h.store.SysDepartmentRebuildPath(ctx)
|
||||
}
|
||||
|
||||
func (h *departmentBiz) Tree(ctx context.Context, id int32) ([]*view.LayuiTree, error) {
|
||||
all, err := h.All(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return toLayuiTree(id, all), nil
|
||||
}
|
||||
|
||||
func (h *departmentBiz) XmSelect(ctx context.Context, id int32) ([]*view.XmSelectTree, error) {
|
||||
all, err := h.All(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return toXmSelectTree(id, all), nil
|
||||
}
|
||||
|
||||
func toXmSelectTree(parentId int32, data []*db.SysDepartment) []*view.XmSelectTree {
|
||||
var res []*view.XmSelectTree
|
||||
for _, v := range data {
|
||||
if v.ParentID == parentId {
|
||||
item := view.XmSelectTree{
|
||||
Name: v.Name,
|
||||
Value: strconv.FormatInt(int64(v.ID), 10),
|
||||
Children: toXmSelectTree(v.ID, data),
|
||||
}
|
||||
res = append(res, &item)
|
||||
}
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
func toLayuiTree(parentId int32, data []*db.SysDepartment) []*view.LayuiTree {
|
||||
var res []*view.LayuiTree
|
||||
for _, v := range data {
|
||||
if v.ParentID == parentId {
|
||||
item := view.LayuiTree{}
|
||||
item.ID = strconv.FormatInt(int64(v.ID), 10)
|
||||
item.Title = v.Name
|
||||
item.Children = toLayuiTree(v.ID, data)
|
||||
if v.ParentID == 0 {
|
||||
item.Spread = true
|
||||
}
|
||||
res = append(res, &item)
|
||||
}
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
311
internal/erpserver/biz/v1/system/menu.go
Normal file
311
internal/erpserver/biz/v1/system/menu.go
Normal file
@@ -0,0 +1,311 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"management/internal/db/model/dto"
|
||||
db "management/internal/db/sqlc"
|
||||
"management/internal/global/keys"
|
||||
"management/internal/pkg/redis"
|
||||
"management/internal/pkg/session"
|
||||
)
|
||||
|
||||
type MenuBiz interface {
|
||||
MenuExpansion
|
||||
}
|
||||
|
||||
type MenuExpansion interface {
|
||||
GetSysMenuByUrl(ctx context.Context, url string) (*db.SysMenu, error)
|
||||
ListOwnerMenuByRoleID(ctx context.Context, roleID int32) ([]*dto.OwnerMenuDto, error)
|
||||
RecursiveSysMenus(ctx context.Context, roleID int32) ([]*dto.MenuUIDto, error)
|
||||
SetRecursiveSysMenus(ctx context.Context, roleID int32) ([]*dto.MenuUIDto, error)
|
||||
MapOwnerMenuByRoleID(ctx context.Context, roleID int32) (map[string]*dto.OwnerMenuDto, error)
|
||||
SetOwnerMapMenuByRoleID(ctx context.Context, roleID int32) (map[string]*dto.OwnerMenuDto, error)
|
||||
}
|
||||
|
||||
type menuBiz struct {
|
||||
store db.Store
|
||||
redis redis.IRedis
|
||||
session session.ISession
|
||||
}
|
||||
|
||||
var _ MenuBiz = (*menuBiz)(nil)
|
||||
|
||||
func NewMenu(store db.Store, redis redis.IRedis, session session.ISession) *menuBiz {
|
||||
return &menuBiz{
|
||||
store: store,
|
||||
redis: redis,
|
||||
session: session,
|
||||
}
|
||||
}
|
||||
|
||||
func (b *menuBiz) GetSysMenuByUrl(ctx context.Context, url string) (*db.SysMenu, error) {
|
||||
return b.store.GetSysMenuByUrl(ctx, url)
|
||||
}
|
||||
|
||||
func (b *menuBiz) ListOwnerMenuByRoleID(ctx context.Context, roleID int32) ([]*dto.OwnerMenuDto, error) {
|
||||
// 判断redis是否存储
|
||||
key := keys.GetManageKey(ctx, keys.OwnerMenus, roleID)
|
||||
bs, err := b.redis.GetBytes(ctx, key)
|
||||
if err == nil {
|
||||
var res []*dto.OwnerMenuDto
|
||||
if err := json.Unmarshal(bs, &res); err == nil {
|
||||
return res, nil
|
||||
}
|
||||
}
|
||||
|
||||
return b.SetOwnerListMenuByRoleID(ctx, roleID)
|
||||
}
|
||||
|
||||
func (b *menuBiz) SetOwnerListMenuByRoleID(ctx context.Context, roleID int32) ([]*dto.OwnerMenuDto, error) {
|
||||
menus, err := b.ownerMenusByRoleID(ctx, roleID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var res []*dto.OwnerMenuDto
|
||||
for _, menu := range menus {
|
||||
res = append(res, &dto.OwnerMenuDto{
|
||||
ID: menu.ID,
|
||||
DisplayName: menu.DisplayName,
|
||||
Url: menu.Url,
|
||||
ParentID: menu.ParentID,
|
||||
Avatar: menu.Avatar,
|
||||
Style: menu.Style,
|
||||
IsList: menu.IsList,
|
||||
})
|
||||
}
|
||||
|
||||
bs, err := json.Marshal(res)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
key := keys.GetManageKey(ctx, keys.OwnerMenus, roleID)
|
||||
_ = redis.Set(ctx, key, bs, time.Hour*6)
|
||||
return res, nil
|
||||
}
|
||||
|
||||
func (b *menuBiz) RecursiveSysMenus(ctx context.Context, roleID int32) ([]*dto.MenuUIDto, error) {
|
||||
// 判断redis是否存储
|
||||
key := keys.GetManageKey(ctx, keys.RecursiveMenus, roleID)
|
||||
bs, err := b.redis.GetBytes(ctx, key)
|
||||
if err == nil {
|
||||
var res []*dto.MenuUIDto
|
||||
if err := json.Unmarshal(bs, &res); err == nil {
|
||||
return res, nil
|
||||
}
|
||||
}
|
||||
|
||||
return b.SetRecursiveSysMenus(ctx, roleID)
|
||||
}
|
||||
|
||||
func (b *menuBiz) SetRecursiveSysMenus(ctx context.Context, roleID int32) ([]*dto.MenuUIDto, error) {
|
||||
// 判断当前用户是否有vip角色
|
||||
role, err := b.store.GetSysRole(ctx, roleID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var menus []*db.SysMenu
|
||||
if role.Vip {
|
||||
// vip 用户
|
||||
all, err := b.store.RecursiveSysMenus(ctx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
menus = convertToMenuUIDto(all)
|
||||
|
||||
} else {
|
||||
// not vip
|
||||
all, err := b.store.RecursiveSysMenusByRoleID(ctx, roleID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
menus = convertToMenuUIDto2(all)
|
||||
}
|
||||
menuList := uniqueSysMenus(menus)
|
||||
if len(menuList) == 0 {
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
tree := convertToUITree(menuList, 0)
|
||||
bs, err := json.Marshal(tree)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
key := keys.GetManageKey(ctx, keys.RecursiveMenus, roleID)
|
||||
_ = redis.Set(ctx, key, bs, time.Hour*6)
|
||||
return tree, nil
|
||||
}
|
||||
|
||||
func (b *menuBiz) MapOwnerMenuByRoleID(ctx context.Context, roleID int32) (map[string]*dto.OwnerMenuDto, error) {
|
||||
// 判断redis是否存储
|
||||
key := keys.GetManageKey(ctx, keys.OwnerMenus, roleID)
|
||||
bs, err := b.redis.GetBytes(ctx, key)
|
||||
if err == nil {
|
||||
var res map[string]*dto.OwnerMenuDto
|
||||
if err := json.Unmarshal(bs, &res); err == nil {
|
||||
return res, nil
|
||||
}
|
||||
}
|
||||
|
||||
return b.SetOwnerMapMenuByRoleID(ctx, roleID)
|
||||
}
|
||||
|
||||
func (b *menuBiz) SetOwnerMapMenuByRoleID(ctx context.Context, roleID int32) (map[string]*dto.OwnerMenuDto, error) {
|
||||
result := make(map[string]*dto.OwnerMenuDto)
|
||||
menus, err := b.ownerMenusByRoleID(ctx, roleID)
|
||||
if err != nil {
|
||||
return result, err
|
||||
}
|
||||
|
||||
for _, menu := range menus {
|
||||
result[menu.Url] = &dto.OwnerMenuDto{
|
||||
ID: menu.ID,
|
||||
DisplayName: menu.DisplayName,
|
||||
Url: menu.Url,
|
||||
ParentID: menu.ParentID,
|
||||
Avatar: menu.Avatar,
|
||||
Style: menu.Style,
|
||||
IsList: menu.IsList,
|
||||
}
|
||||
}
|
||||
|
||||
bs, err := json.Marshal(result)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
key := keys.GetManageKey(ctx, keys.OwnerMenus, roleID)
|
||||
_ = redis.Set(ctx, key, bs, time.Hour*6)
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (b *menuBiz) ownerMenusByRoleID(ctx context.Context, roleID int32) ([]*db.SysMenu, error) {
|
||||
// 判断当前用户是否有vip角色
|
||||
role, err := b.store.GetSysRole(ctx, roleID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var e error
|
||||
var menus []*db.SysMenu
|
||||
if role.Vip {
|
||||
// vip 用户
|
||||
menus, e = b.store.AllSysMenu(ctx)
|
||||
if e != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
} else {
|
||||
// not vip
|
||||
menus, e = b.store.ListSysMenuByRoleID(ctx, roleID)
|
||||
if e != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
return menus, nil
|
||||
}
|
||||
|
||||
func convertToMenuUIDto(data []*db.RecursiveSysMenusRow) []*db.SysMenu {
|
||||
var res []*db.SysMenu
|
||||
|
||||
for _, item := range data {
|
||||
temp := &db.SysMenu{
|
||||
ID: item.ID,
|
||||
Name: item.Name,
|
||||
DisplayName: item.DisplayName,
|
||||
Url: item.Url,
|
||||
Type: item.Type,
|
||||
ParentID: item.ParentID,
|
||||
ParentPath: item.ParentPath,
|
||||
Avatar: item.Avatar,
|
||||
Style: item.Style,
|
||||
Visible: item.Visible,
|
||||
IsList: item.IsList,
|
||||
Status: item.Status,
|
||||
Sort: item.Sort,
|
||||
CreatedAt: item.CreatedAt,
|
||||
UpdatedAt: item.UpdatedAt,
|
||||
}
|
||||
res = append(res, temp)
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
func convertToMenuUIDto2(data []*db.RecursiveSysMenusByRoleIDRow) []*db.SysMenu {
|
||||
var res []*db.SysMenu
|
||||
|
||||
for _, item := range data {
|
||||
temp := &db.SysMenu{
|
||||
ID: item.ID,
|
||||
Name: item.Name,
|
||||
DisplayName: item.DisplayName,
|
||||
Url: item.Url,
|
||||
Type: item.Type,
|
||||
ParentID: item.ParentID,
|
||||
ParentPath: item.ParentPath,
|
||||
Avatar: item.Avatar,
|
||||
Style: item.Style,
|
||||
Visible: item.Visible,
|
||||
IsList: item.IsList,
|
||||
Status: item.Status,
|
||||
Sort: item.Sort,
|
||||
CreatedAt: item.CreatedAt,
|
||||
UpdatedAt: item.UpdatedAt,
|
||||
}
|
||||
res = append(res, temp)
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
func convertToUITree(data []*db.SysMenu, parentID int32) []*dto.MenuUIDto {
|
||||
var root []*dto.MenuUIDto
|
||||
for _, item := range data {
|
||||
if item.ParentID == parentID {
|
||||
if item.IsList {
|
||||
temp := &dto.MenuUIDto{
|
||||
ID: strings.ToLower(item.Url),
|
||||
Title: item.DisplayName,
|
||||
Icon: item.Avatar,
|
||||
Type: 1,
|
||||
OpenType: "_iframe",
|
||||
// OpenType: "_component",
|
||||
Href: item.Url,
|
||||
}
|
||||
root = append(root, temp)
|
||||
} else {
|
||||
temp := &dto.MenuUIDto{
|
||||
ID: strconv.Itoa(int(item.ID)),
|
||||
Title: item.DisplayName,
|
||||
Icon: item.Avatar,
|
||||
Type: 0,
|
||||
}
|
||||
temp.Children = convertToUITree(data, item.ID)
|
||||
root = append(root, temp)
|
||||
}
|
||||
}
|
||||
}
|
||||
return root
|
||||
}
|
||||
|
||||
func uniqueSysMenus(sm []*db.SysMenu) []*db.SysMenu {
|
||||
res := make([]*db.SysMenu, 0) // 返回的新切片
|
||||
m1 := make(map[int32]byte) // 用来去重的临时map
|
||||
for _, v := range sm {
|
||||
if _, ok := m1[v.ID]; !ok {
|
||||
m1[v.ID] = 1
|
||||
res = append(res, v)
|
||||
}
|
||||
}
|
||||
return res
|
||||
}
|
||||
51
internal/erpserver/biz/v1/system/system.go
Normal file
51
internal/erpserver/biz/v1/system/system.go
Normal file
@@ -0,0 +1,51 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
db "management/internal/db/sqlc"
|
||||
"management/internal/pkg/redis"
|
||||
"management/internal/pkg/session"
|
||||
)
|
||||
|
||||
type SystemBiz interface {
|
||||
UserBiz() UserBiz
|
||||
MenuBiz() MenuBiz
|
||||
DepartmentBiz() DepartmentBiz
|
||||
AuditBiz() AuditBiz
|
||||
ConfigBiz() ConfigBiz
|
||||
}
|
||||
|
||||
type systemBiz struct {
|
||||
store db.Store
|
||||
redis redis.IRedis
|
||||
session session.ISession
|
||||
}
|
||||
|
||||
var _ SystemBiz = (*systemBiz)(nil)
|
||||
|
||||
func New(store db.Store, redis redis.IRedis, session session.ISession) *systemBiz {
|
||||
return &systemBiz{
|
||||
store: store,
|
||||
redis: redis,
|
||||
session: session,
|
||||
}
|
||||
}
|
||||
|
||||
func (b *systemBiz) UserBiz() UserBiz {
|
||||
return NewUser(b.store, b.session)
|
||||
}
|
||||
|
||||
func (b *systemBiz) MenuBiz() MenuBiz {
|
||||
return NewMenu(b.store, b.redis, b.session)
|
||||
}
|
||||
|
||||
func (b *systemBiz) DepartmentBiz() DepartmentBiz {
|
||||
return NewDepartment(b.store, b.redis, b.session)
|
||||
}
|
||||
|
||||
func (b *systemBiz) AuditBiz() AuditBiz {
|
||||
return NewAudit(b.store)
|
||||
}
|
||||
|
||||
func (b *systemBiz) ConfigBiz() ConfigBiz {
|
||||
return NewConfig(b.store, b.redis, b.session)
|
||||
}
|
||||
117
internal/erpserver/biz/v1/system/user.go
Normal file
117
internal/erpserver/biz/v1/system/user.go
Normal file
@@ -0,0 +1,117 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"time"
|
||||
|
||||
"management/internal/db/model/dto"
|
||||
db "management/internal/db/sqlc"
|
||||
"management/internal/erpserver/model/req"
|
||||
"management/internal/global/know"
|
||||
"management/internal/pkg/crypto"
|
||||
"management/internal/pkg/session"
|
||||
)
|
||||
|
||||
// UserBiz 定义处理用户请求所需的方法.
|
||||
type UserBiz interface {
|
||||
Create(ctx context.Context, req *db.CreateSysUserParams) (*db.SysUser, error)
|
||||
|
||||
UserExpansion
|
||||
}
|
||||
|
||||
// UserExpansion 定义用户操作的扩展方法.
|
||||
type UserExpansion interface {
|
||||
Login(ctx context.Context, req *req.Login) error
|
||||
}
|
||||
|
||||
// userBiz 是 UserBiz 接口的实现.
|
||||
type userBiz struct {
|
||||
store db.Store
|
||||
session session.ISession
|
||||
}
|
||||
|
||||
// 确保 userBiz 实现了 UserBiz 接口.
|
||||
var _ UserBiz = (*userBiz)(nil)
|
||||
|
||||
func NewUser(store db.Store, session session.ISession) *userBiz {
|
||||
return &userBiz{
|
||||
store: store,
|
||||
session: session,
|
||||
}
|
||||
}
|
||||
|
||||
func (b *userBiz) Create(ctx context.Context, req *db.CreateSysUserParams) (*db.SysUser, error) {
|
||||
return b.store.CreateSysUser(ctx, req)
|
||||
}
|
||||
|
||||
func (b *userBiz) Login(ctx context.Context, req *req.Login) error {
|
||||
log := &db.CreateSysUserLoginLogParams{
|
||||
CreatedAt: time.Now(),
|
||||
Email: req.Email,
|
||||
IsSuccess: false,
|
||||
RefererUrl: req.Referrer,
|
||||
Url: req.Url,
|
||||
Os: req.Os,
|
||||
Ip: req.Ip,
|
||||
Browser: req.Browser,
|
||||
}
|
||||
|
||||
user, err := b.store.GetSysUserByEmail(ctx, req.Email)
|
||||
if err != nil {
|
||||
log.Message = err.Error()
|
||||
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
||||
return err
|
||||
}
|
||||
log.UserUuid = user.Uuid
|
||||
log.Username = user.Username
|
||||
|
||||
err = crypto.BcryptComparePassword(user.HashedPassword, req.Password+user.Salt)
|
||||
if err != nil {
|
||||
log.Message = "compare password failed"
|
||||
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
||||
return errors.New(log.Message)
|
||||
}
|
||||
|
||||
// 登陆成功
|
||||
|
||||
if user.RoleID == 0 {
|
||||
log.Message = "账号没有配置角色, 请联系管理员"
|
||||
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
||||
return errors.New(log.Message)
|
||||
}
|
||||
|
||||
sysRole, err := b.store.GetSysRole(ctx, user.RoleID)
|
||||
if err != nil {
|
||||
log.Message = "账号配置的角色错误, 请联系管理员"
|
||||
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
||||
return errors.New(log.Message)
|
||||
}
|
||||
|
||||
auth := dto.AuthorizeUser{
|
||||
ID: user.ID,
|
||||
Uuid: user.Uuid,
|
||||
Email: user.Email,
|
||||
Username: user.Username,
|
||||
RoleID: sysRole.ID,
|
||||
RoleName: sysRole.Name,
|
||||
OS: log.Os,
|
||||
IP: log.Ip,
|
||||
Browser: log.Browser,
|
||||
}
|
||||
|
||||
gob, err := json.Marshal(auth)
|
||||
if err != nil {
|
||||
log.Message = err.Error()
|
||||
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
||||
return err
|
||||
}
|
||||
|
||||
b.session.Put(ctx, know.StoreName, gob)
|
||||
|
||||
log.IsSuccess = true
|
||||
log.Message = "登陆成功"
|
||||
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
||||
return nil
|
||||
}
|
||||
56
internal/erpserver/handler/common/captcha.go
Normal file
56
internal/erpserver/handler/common/captcha.go
Normal file
@@ -0,0 +1,56 @@
|
||||
package common
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"management/internal/config"
|
||||
commonv1 "management/internal/erpserver/biz/v1/common"
|
||||
"management/internal/pkg/tpl"
|
||||
)
|
||||
|
||||
type CaptchaHandler interface {
|
||||
Captcha(w http.ResponseWriter, r *http.Request)
|
||||
}
|
||||
|
||||
// captchaHandler 是 CaptchaHandler 接口的实现.
|
||||
type captchaHandler struct {
|
||||
conf *config.Captcha
|
||||
render tpl.Renderer
|
||||
biz commonv1.CaptchaBiz
|
||||
}
|
||||
|
||||
// 确保 captchaHandler 实现了 CaptchaHandler 接口.
|
||||
var _ CaptchaHandler = (*captchaHandler)(nil)
|
||||
|
||||
func NewCaptchaHandler(conf *config.Captcha, render tpl.Renderer, biz commonv1.CaptchaBiz) *captchaHandler {
|
||||
return &captchaHandler{
|
||||
conf: conf,
|
||||
render: render,
|
||||
biz: biz,
|
||||
}
|
||||
}
|
||||
|
||||
type CaptchaResponse struct {
|
||||
CaptchaID string `json:"captcha_id"`
|
||||
PicPath string `json:"pic_path"`
|
||||
CaptchaLength int `json:"captcha_length"`
|
||||
OpenCaptcha int `json:"open_captcha"`
|
||||
}
|
||||
|
||||
func (h *captchaHandler) Captcha(w http.ResponseWriter, r *http.Request) {
|
||||
keyLong := h.conf.KeyLong
|
||||
oc := h.conf.OpenCaptcha
|
||||
id, b64s, _, err := h.biz.Generate(h.conf.ImgHeight, h.conf.ImgWidth, keyLong, 0.7, 80)
|
||||
if err != nil {
|
||||
h.render.JSON(w, tpl.Response{Success: false, Message: "获取验证码失败"})
|
||||
return
|
||||
}
|
||||
|
||||
rsp := CaptchaResponse{
|
||||
CaptchaID: id,
|
||||
PicPath: b64s,
|
||||
CaptchaLength: keyLong,
|
||||
OpenCaptcha: oc,
|
||||
}
|
||||
h.render.JSON(w, tpl.Response{Success: true, Message: "ok", Data: rsp})
|
||||
}
|
||||
31
internal/erpserver/handler/common/common.go
Normal file
31
internal/erpserver/handler/common/common.go
Normal file
@@ -0,0 +1,31 @@
|
||||
package common
|
||||
|
||||
import (
|
||||
"management/internal/config"
|
||||
commonv1 "management/internal/erpserver/biz/v1/common"
|
||||
"management/internal/pkg/tpl"
|
||||
)
|
||||
|
||||
type CommonHandler interface {
|
||||
CaptchaHandler() CaptchaHandler
|
||||
}
|
||||
|
||||
type commonHandler struct {
|
||||
conf *config.Config
|
||||
render tpl.Renderer
|
||||
biz commonv1.CommonBiz
|
||||
}
|
||||
|
||||
var _ CommonHandler = (*commonHandler)(nil)
|
||||
|
||||
func NewCommonHandler(conf *config.Config, render tpl.Renderer, biz commonv1.CommonBiz) *commonHandler {
|
||||
return &commonHandler{
|
||||
conf: conf,
|
||||
render: render,
|
||||
biz: biz,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *commonHandler) CaptchaHandler() CaptchaHandler {
|
||||
return NewCaptchaHandler(&h.conf.Captcha, h.render, h.biz.CaptchaBiz())
|
||||
}
|
||||
55
internal/erpserver/handler/handler.go
Normal file
55
internal/erpserver/handler/handler.go
Normal file
@@ -0,0 +1,55 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"management/internal/config"
|
||||
"management/internal/erpserver/biz"
|
||||
"management/internal/erpserver/handler/common"
|
||||
"management/internal/erpserver/handler/system"
|
||||
"management/internal/pkg/session"
|
||||
"management/internal/pkg/tpl"
|
||||
)
|
||||
|
||||
// IHandler 定义了Handler需要实现的方法.
|
||||
type IHandler interface {
|
||||
// 获取 Common Handler 接口.
|
||||
CommonHandler() common.CommonHandler
|
||||
|
||||
// 获取首页
|
||||
Home(w http.ResponseWriter, req *http.Request)
|
||||
|
||||
// 获取 System Handler 接口.
|
||||
SystemHandler() system.SystemHandler
|
||||
}
|
||||
|
||||
// handler 是 IHandler 的一个具体实现.
|
||||
type handler struct {
|
||||
conf *config.Config
|
||||
render tpl.Renderer
|
||||
session session.ISession
|
||||
biz biz.IBiz
|
||||
}
|
||||
|
||||
// 确保 handler 实现了 IHandler 接口.
|
||||
var _ IHandler = (*handler)(nil)
|
||||
|
||||
// NewHandler 创建一个 IHandler 类型的实例.
|
||||
func NewHandler(conf *config.Config, render tpl.Renderer, session session.ISession, biz biz.IBiz) *handler {
|
||||
return &handler{
|
||||
conf: conf,
|
||||
render: render,
|
||||
session: session,
|
||||
biz: biz,
|
||||
}
|
||||
}
|
||||
|
||||
// CommonHandler 返回一个实现了 CommonHandler 接口的实例.
|
||||
func (h *handler) CommonHandler() common.CommonHandler {
|
||||
return common.NewCommonHandler(h.conf, h.render, h.biz.CommonV1())
|
||||
}
|
||||
|
||||
// SystemHandler 返回一个实现了 SystemHandler 接口的实例.
|
||||
func (h *handler) SystemHandler() system.SystemHandler {
|
||||
return system.NewSystemHandler(h.render, h.session, h.biz)
|
||||
}
|
||||
7
internal/erpserver/handler/home.go
Normal file
7
internal/erpserver/handler/home.go
Normal file
@@ -0,0 +1,7 @@
|
||||
package handler
|
||||
|
||||
import "net/http"
|
||||
|
||||
func (h *handler) Home(w http.ResponseWriter, r *http.Request) {
|
||||
h.render.HTML(w, r, "home/home.tmpl", nil)
|
||||
}
|
||||
50
internal/erpserver/handler/system/config.go
Normal file
50
internal/erpserver/handler/system/config.go
Normal file
@@ -0,0 +1,50 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"management/internal/erpserver/biz"
|
||||
"management/internal/pkg/session"
|
||||
"management/internal/pkg/tpl"
|
||||
)
|
||||
|
||||
type ConfigHandler interface {
|
||||
// Add(w http.ResponseWriter, r *http.Request)
|
||||
// Edit(w http.ResponseWriter, r *http.Request)
|
||||
// Save(w http.ResponseWriter, r *http.Request)
|
||||
// List(w http.ResponseWriter, r *http.Request)
|
||||
|
||||
ConfigExpansion
|
||||
}
|
||||
|
||||
type ConfigExpansion interface {
|
||||
Pear(w http.ResponseWriter, r *http.Request)
|
||||
}
|
||||
|
||||
// configHandler 是 ConfigHandler 接口的实现.
|
||||
type configHandler struct {
|
||||
render tpl.Renderer
|
||||
session session.ISession
|
||||
biz biz.IBiz
|
||||
}
|
||||
|
||||
// 确保 userHandler 实现了 ConfigHandler 接口.
|
||||
var _ ConfigHandler = (*configHandler)(nil)
|
||||
|
||||
func NewConfigHandler(render tpl.Renderer, session session.ISession, biz biz.IBiz) *configHandler {
|
||||
return &configHandler{
|
||||
render: render,
|
||||
session: session,
|
||||
biz: biz,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *configHandler) Pear(w http.ResponseWriter, r *http.Request) {
|
||||
pear, err := h.biz.SystemV1().ConfigBiz().Pear(r.Context())
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
h.render.JSON(w, pear)
|
||||
}
|
||||
206
internal/erpserver/handler/system/department.go
Normal file
206
internal/erpserver/handler/system/department.go
Normal file
@@ -0,0 +1,206 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"management/internal/db/model/dto"
|
||||
db "management/internal/db/sqlc"
|
||||
"management/internal/erpserver/biz"
|
||||
"management/internal/pkg/convertor"
|
||||
"management/internal/pkg/tpl"
|
||||
)
|
||||
|
||||
type DepartmentHandler interface {
|
||||
List(w http.ResponseWriter, r *http.Request)
|
||||
Add(w http.ResponseWriter, r *http.Request)
|
||||
AddChildren(w http.ResponseWriter, r *http.Request)
|
||||
Edit(w http.ResponseWriter, r *http.Request)
|
||||
Save(w http.ResponseWriter, r *http.Request)
|
||||
Tree(w http.ResponseWriter, r *http.Request)
|
||||
Refresh(w http.ResponseWriter, r *http.Request)
|
||||
RebuildParentPath(w http.ResponseWriter, r *http.Request)
|
||||
}
|
||||
|
||||
type departmentHandler struct {
|
||||
render tpl.Renderer
|
||||
biz biz.IBiz
|
||||
}
|
||||
|
||||
var _ DepartmentHandler = (*departmentHandler)(nil)
|
||||
|
||||
func NewDepartmentHandler(render tpl.Renderer, biz biz.IBiz) *departmentHandler {
|
||||
return &departmentHandler{
|
||||
render: render,
|
||||
biz: biz,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *departmentHandler) List(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method == http.MethodGet {
|
||||
h.render.HTML(w, r, "department/list.tmpl", nil)
|
||||
return
|
||||
} else if r.Method == http.MethodPost {
|
||||
var q dto.SearchDto
|
||||
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 := h.biz.SystemV1().DepartmentBiz().List(r.Context(), q)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
data := tpl.ResponseList{
|
||||
Code: 0,
|
||||
Message: "ok",
|
||||
Count: count,
|
||||
Data: res,
|
||||
}
|
||||
h.render.JSON(w, data)
|
||||
return
|
||||
}
|
||||
|
||||
http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
|
||||
}
|
||||
|
||||
func (h *departmentHandler) Add(w http.ResponseWriter, r *http.Request) {
|
||||
h.render.HTML(w, r, "department/edit.tmpl", map[string]any{
|
||||
"Item": &db.SysDepartment{Sort: 6666},
|
||||
})
|
||||
}
|
||||
|
||||
func (h *departmentHandler) AddChildren(w http.ResponseWriter, r *http.Request) {
|
||||
vars := r.URL.Query()
|
||||
parentID := convertor.QueryInt(vars, "parentID", 0)
|
||||
vm := &db.SysDepartment{ParentID: int32(parentID), Sort: 6666}
|
||||
h.render.HTML(w, r, "department/edit.tmpl", map[string]any{
|
||||
"Item": vm,
|
||||
})
|
||||
}
|
||||
|
||||
func (h *departmentHandler) Edit(w http.ResponseWriter, r *http.Request) {
|
||||
vars := r.URL.Query()
|
||||
id := convertor.QueryInt[int32](vars, "id", 0)
|
||||
vm := &db.SysDepartment{Sort: 6666}
|
||||
if id > 0 {
|
||||
vm, _ = h.biz.SystemV1().DepartmentBiz().Get(r.Context(), id)
|
||||
}
|
||||
h.render.HTML(w, r, "department/edit.tmpl", map[string]any{
|
||||
"Item": vm,
|
||||
})
|
||||
}
|
||||
|
||||
func (h *departmentHandler) Save(w http.ResponseWriter, r *http.Request) {
|
||||
id := convertor.ConvertInt[int32](r.PostFormValue("ID"), 0)
|
||||
ParentID := convertor.ConvertInt[int32](r.PostFormValue("ParentID"), 0)
|
||||
name := r.PostFormValue("Name")
|
||||
sort := convertor.ConvertInt[int32](r.PostFormValue("Sort"), 6666)
|
||||
status := convertor.ConvertInt[int32](r.PostFormValue("Status"), 9999)
|
||||
|
||||
ctx := r.Context()
|
||||
var parent *db.SysDepartment
|
||||
if ParentID > 0 {
|
||||
var err error
|
||||
parent, err = h.biz.SystemV1().DepartmentBiz().Get(ctx, ParentID)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, "父级节点错误")
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if id == 0 {
|
||||
arg := db.CreateSysDepartmentParams{
|
||||
Name: name,
|
||||
ParentID: ParentID,
|
||||
ParentPath: fmt.Sprintf("%s,%d,", parent.ParentPath, parent.ID),
|
||||
Status: status,
|
||||
Sort: sort,
|
||||
CreatedAt: time.Now(),
|
||||
UpdatedAt: time.Now(),
|
||||
}
|
||||
_, err := h.biz.SystemV1().DepartmentBiz().Create(ctx, &arg)
|
||||
if err != nil {
|
||||
if db.IsUniqueViolation(err) {
|
||||
h.render.JSONERR(w, "部门名称已存在")
|
||||
return
|
||||
}
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
h.render.JSONOK(w, "添加成功")
|
||||
} else {
|
||||
res, err := h.biz.SystemV1().DepartmentBiz().Get(ctx, id)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
arg := &db.UpdateSysDepartmentParams{
|
||||
ID: res.ID,
|
||||
Name: name,
|
||||
ParentID: ParentID,
|
||||
ParentPath: fmt.Sprintf("%s,%d,", parent.ParentPath, parent.ID),
|
||||
Status: status,
|
||||
Sort: sort,
|
||||
UpdatedAt: time.Now(),
|
||||
}
|
||||
_, err = h.biz.SystemV1().DepartmentBiz().Update(ctx, arg)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
h.render.JSONOK(w, "更新成功")
|
||||
}
|
||||
}
|
||||
|
||||
func (h *departmentHandler) Tree(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
vars := r.URL.Query()
|
||||
if vars.Get("type") == "xmselect" {
|
||||
res, err := h.biz.SystemV1().DepartmentBiz().XmSelect(ctx, 0)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
h.render.JSON(w, res)
|
||||
return
|
||||
} else {
|
||||
res, err := h.biz.SystemV1().DepartmentBiz().Tree(ctx, 0)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
h.render.JSON(w, res)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func (h *departmentHandler) Refresh(w http.ResponseWriter, r *http.Request) {
|
||||
_, err := h.biz.SystemV1().DepartmentBiz().Refresh(r.Context())
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
h.render.JSONOK(w, "刷新成功")
|
||||
}
|
||||
|
||||
func (h *departmentHandler) RebuildParentPath(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
err := h.biz.SystemV1().DepartmentBiz().RebuildParentPath(ctx)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
h.render.JSONOK(w, "重建成功")
|
||||
}
|
||||
53
internal/erpserver/handler/system/menu.go
Normal file
53
internal/erpserver/handler/system/menu.go
Normal file
@@ -0,0 +1,53 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
|
||||
"management/internal/db/model/dto"
|
||||
"management/internal/erpserver/biz"
|
||||
"management/internal/global/know"
|
||||
"management/internal/pkg/session"
|
||||
"management/internal/pkg/tpl"
|
||||
)
|
||||
|
||||
type MenuHandler interface {
|
||||
MenuExpansion
|
||||
}
|
||||
|
||||
type MenuExpansion interface {
|
||||
Menus(w http.ResponseWriter, r *http.Request)
|
||||
}
|
||||
|
||||
type menuHandler struct {
|
||||
render tpl.Renderer
|
||||
session session.ISession
|
||||
biz biz.IBiz
|
||||
}
|
||||
|
||||
var _ MenuHandler = (*menuHandler)(nil)
|
||||
|
||||
func NewMenuHandler(render tpl.Renderer, session session.ISession, biz biz.IBiz) *menuHandler {
|
||||
return &menuHandler{
|
||||
render: render,
|
||||
session: session,
|
||||
biz: biz,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *menuHandler) Menus(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
b := h.session.GetBytes(ctx, know.StoreName)
|
||||
var u dto.AuthorizeUser
|
||||
if err := json.Unmarshal(b, &u); err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
menus, err := h.biz.SystemV1().MenuBiz().RecursiveSysMenus(ctx, u.RoleID)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
h.render.JSON(w, menus)
|
||||
}
|
||||
46
internal/erpserver/handler/system/system.go
Normal file
46
internal/erpserver/handler/system/system.go
Normal file
@@ -0,0 +1,46 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"management/internal/erpserver/biz"
|
||||
"management/internal/pkg/session"
|
||||
"management/internal/pkg/tpl"
|
||||
)
|
||||
|
||||
type SystemHandler interface {
|
||||
UserHandler() UserHandler
|
||||
MenuHandler() MenuHandler
|
||||
DepartmentHandler() DepartmentHandler
|
||||
ConfigHandler() ConfigHandler
|
||||
}
|
||||
|
||||
type systemHandler struct {
|
||||
render tpl.Renderer
|
||||
session session.ISession
|
||||
biz biz.IBiz
|
||||
}
|
||||
|
||||
var _ SystemHandler = (*systemHandler)(nil)
|
||||
|
||||
func NewSystemHandler(render tpl.Renderer, session session.ISession, biz biz.IBiz) *systemHandler {
|
||||
return &systemHandler{
|
||||
render: render,
|
||||
session: session,
|
||||
biz: biz,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *systemHandler) UserHandler() UserHandler {
|
||||
return NewUserHandler(h.render, h.session, h.biz)
|
||||
}
|
||||
|
||||
func (h *systemHandler) MenuHandler() MenuHandler {
|
||||
return NewMenuHandler(h.render, h.session, h.biz)
|
||||
}
|
||||
|
||||
func (h *systemHandler) DepartmentHandler() DepartmentHandler {
|
||||
return NewDepartmentHandler(h.render, h.biz)
|
||||
}
|
||||
|
||||
func (h *systemHandler) ConfigHandler() ConfigHandler {
|
||||
return NewConfigHandler(h.render, h.session, h.biz)
|
||||
}
|
||||
127
internal/erpserver/handler/system/user.go
Normal file
127
internal/erpserver/handler/system/user.go
Normal file
@@ -0,0 +1,127 @@
|
||||
package system
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"management/internal/db/model/dto"
|
||||
"management/internal/erpserver/biz"
|
||||
"management/internal/erpserver/model/req"
|
||||
"management/internal/global/know"
|
||||
"management/internal/pkg/session"
|
||||
"management/internal/pkg/tpl"
|
||||
|
||||
"github.com/zhang2092/browser"
|
||||
)
|
||||
|
||||
type UserHandler interface {
|
||||
Add(w http.ResponseWriter, r *http.Request)
|
||||
Edit(w http.ResponseWriter, r *http.Request)
|
||||
Save(w http.ResponseWriter, r *http.Request)
|
||||
List(w http.ResponseWriter, r *http.Request)
|
||||
|
||||
UserExpansion
|
||||
}
|
||||
|
||||
type UserExpansion interface {
|
||||
Login(w http.ResponseWriter, r *http.Request)
|
||||
Logout(w http.ResponseWriter, r *http.Request)
|
||||
}
|
||||
|
||||
// userHandler 是 UserHandler 接口的实现.
|
||||
type userHandler struct {
|
||||
render tpl.Renderer
|
||||
session session.ISession
|
||||
biz biz.IBiz
|
||||
}
|
||||
|
||||
// 确保 userHandler 实现了 UserHandler 接口.
|
||||
var _ UserHandler = (*userHandler)(nil)
|
||||
|
||||
func NewUserHandler(render tpl.Renderer, session session.ISession, biz biz.IBiz) *userHandler {
|
||||
return &userHandler{
|
||||
render: render,
|
||||
session: session,
|
||||
biz: biz,
|
||||
}
|
||||
}
|
||||
|
||||
func (h *userHandler) Add(w http.ResponseWriter, r *http.Request) {}
|
||||
|
||||
func (h *userHandler) Edit(w http.ResponseWriter, r *http.Request) {}
|
||||
|
||||
func (h *userHandler) Save(w http.ResponseWriter, r *http.Request) {}
|
||||
|
||||
func (h *userHandler) List(w http.ResponseWriter, r *http.Request) {}
|
||||
|
||||
func (h *userHandler) Login(w http.ResponseWriter, r *http.Request) {
|
||||
ctx := r.Context()
|
||||
if r.Method == http.MethodGet {
|
||||
var user dto.AuthorizeUser
|
||||
u := h.session.GetBytes(ctx, know.StoreName)
|
||||
if err := json.Unmarshal(u, &user); err == nil {
|
||||
// 判断租户是否一致, 一致则刷新令牌,跳转到首页
|
||||
if err := h.session.RenewToken(ctx); err == nil {
|
||||
h.session.Put(ctx, know.StoreName, u)
|
||||
http.Redirect(w, r, "/home.html", http.StatusFound)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
h.session.Destroy(ctx)
|
||||
h.render.HTML(w, r, "oauth/login.tmpl", nil)
|
||||
return
|
||||
} else if r.Method == http.MethodPost {
|
||||
req := &req.Login{
|
||||
Email: strings.TrimSpace(r.PostFormValue("email")),
|
||||
Password: strings.TrimSpace(r.PostFormValue("password")),
|
||||
CaptchaID: strings.TrimSpace(r.PostFormValue("captcha_id")),
|
||||
Captcha: strings.TrimSpace(r.PostFormValue("captcha")),
|
||||
Ip: r.RemoteAddr,
|
||||
Referrer: r.Header.Get("Referer"),
|
||||
Url: r.URL.RequestURI(),
|
||||
}
|
||||
|
||||
if len(req.Email) == 0 {
|
||||
h.render.JSON(w, tpl.Response{Success: false, Message: "请填写邮箱"})
|
||||
return
|
||||
}
|
||||
if len(req.Password) == 0 {
|
||||
h.render.JSON(w, tpl.Response{Success: false, Message: "请填写密码"})
|
||||
return
|
||||
}
|
||||
if len(req.Captcha) == 0 {
|
||||
h.render.JSON(w, tpl.Response{Success: false, Message: "请填写验证码"})
|
||||
return
|
||||
}
|
||||
if !h.biz.CommonV1().CaptchaBiz().Verify(req.CaptchaID, req.Captcha, true) {
|
||||
h.render.JSON(w, tpl.Response{Success: false, Message: "验证码错误"})
|
||||
return
|
||||
}
|
||||
|
||||
br, err := browser.NewBrowser(r.Header.Get("User-Agent"))
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, "平台信息获取错误")
|
||||
return
|
||||
}
|
||||
|
||||
req.Os = br.Platform().Name()
|
||||
req.Browser = br.Name()
|
||||
err = h.biz.SystemV1().UserBiz().Login(ctx, req)
|
||||
if err != nil {
|
||||
h.render.JSONERR(w, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
h.render.JSONOK(w, "login successful")
|
||||
return
|
||||
}
|
||||
|
||||
http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
|
||||
}
|
||||
|
||||
func (h *userHandler) Logout(w http.ResponseWriter, r *http.Request) {
|
||||
h.session.Destroy(r.Context())
|
||||
http.Redirect(w, r, "/", http.StatusFound)
|
||||
}
|
||||
147
internal/erpserver/http.go
Normal file
147
internal/erpserver/http.go
Normal file
@@ -0,0 +1,147 @@
|
||||
package erpserver
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"management/internal/erpserver/handler"
|
||||
mw "management/internal/pkg/middleware"
|
||||
|
||||
"github.com/go-chi/chi/v5"
|
||||
"github.com/go-chi/chi/v5/middleware"
|
||||
)
|
||||
|
||||
func NewRouter(handler handler.IHandler, mw mw.IMiddleware) *chi.Mux {
|
||||
r := chi.NewRouter()
|
||||
|
||||
r.Use(middleware.RequestID)
|
||||
r.Use(middleware.RealIP)
|
||||
// r.Use(middleware.Logger)
|
||||
r.Use(middleware.Recoverer)
|
||||
|
||||
staticServer := http.FileServer(http.Dir("./web/statics/"))
|
||||
r.Handle("/statics/*", http.StripPrefix("/statics", staticServer))
|
||||
|
||||
uploadServer := http.FileServer(http.Dir("./upload/"))
|
||||
r.Handle("/upload/*", http.StripPrefix("/upload", uploadServer))
|
||||
|
||||
r.Group(func(r chi.Router) {
|
||||
r.Use(mw.NoSurf) // CSRF
|
||||
r.Use(mw.LoadSession) // Session
|
||||
|
||||
r.Get("/captcha", handler.CommonHandler().CaptchaHandler().Captcha)
|
||||
|
||||
r.Get("/", handler.SystemHandler().UserHandler().Login)
|
||||
r.Post("/login", handler.SystemHandler().UserHandler().Login)
|
||||
r.Get("/logout", handler.SystemHandler().UserHandler().Logout)
|
||||
|
||||
// r.With(auth.Authorize, mw.Audit).Post("/upload/img", commonhandler.UploadImg)
|
||||
// r.With(auth.Authorize, mw.Audit).Post("/upload/file", commonhandler.UploadFile)
|
||||
// r.With(auth.Authorize, mw.Audit).Post("/upload/mutilfile", commonhandler.UploadMutilFiles)
|
||||
|
||||
r.With(mw.Authorize, mw.Audit).Get("/home.html", handler.Home)
|
||||
r.With(mw.Authorize).Get("/pear.json", handler.SystemHandler().ConfigHandler().Pear)
|
||||
|
||||
r.Route("/system", func(r chi.Router) {
|
||||
r.Use(mw.Authorize)
|
||||
|
||||
// r.Route("/config", func(r chi.Router) {
|
||||
// r.Use(mw.Audit)
|
||||
// r.Get("/list", configHandler.List)
|
||||
// r.Post("/list", configHandler.PostList)
|
||||
// r.Get("/add", configHandler.Add)
|
||||
// r.Get("/edit", configHandler.Edit)
|
||||
// r.Post("/save", configHandler.Save)
|
||||
// r.Post("/reset_pear", configHandler.ResetPear)
|
||||
// r.Post("/refresh", configHandler.Refresh)
|
||||
// })
|
||||
|
||||
r.Route("/department", func(r chi.Router) {
|
||||
r.Use(mw.Audit)
|
||||
r.Get("/list", handler.SystemHandler().DepartmentHandler().List)
|
||||
r.Post("/list", handler.SystemHandler().DepartmentHandler().List)
|
||||
r.Get("/add", handler.SystemHandler().DepartmentHandler().Add)
|
||||
r.Get("/add_children", handler.SystemHandler().DepartmentHandler().AddChildren)
|
||||
r.Get("/edit", handler.SystemHandler().DepartmentHandler().Edit)
|
||||
r.Post("/save", handler.SystemHandler().DepartmentHandler().Save)
|
||||
r.Post("/tree", handler.SystemHandler().DepartmentHandler().Tree)
|
||||
r.Post("/refresh", handler.SystemHandler().DepartmentHandler().Refresh)
|
||||
r.Post("/rebuild_parent_path", handler.SystemHandler().DepartmentHandler().RebuildParentPath)
|
||||
})
|
||||
|
||||
// r.Route("/user", func(r chi.Router) {
|
||||
// r.Use(mw.Audit)
|
||||
// userHandler := systemhandler.NewSysUserHandler()
|
||||
// r.Get("/list", userHandler.List)
|
||||
// r.Post("/list", userHandler.PostList)
|
||||
// r.Get("/add", userHandler.Add)
|
||||
// r.Get("/edit", userHandler.Edit)
|
||||
// r.Post("/save", userHandler.Save)
|
||||
// r.Get("/profile", userHandler.Profile)
|
||||
// r.Post("/xmselect", userHandler.XmSelect)
|
||||
// })
|
||||
|
||||
// r.Route("/login_log", func(r chi.Router) {
|
||||
// // r.Use(mw.Audit)
|
||||
// userLoginLogHandler := systemhandler.NewSysUserLoginLogHandler()
|
||||
// r.Get("/list", userLoginLogHandler.List)
|
||||
// r.Post("/list", userLoginLogHandler.PostList)
|
||||
// })
|
||||
|
||||
// r.Route("/audit_log", func(r chi.Router) {
|
||||
// // r.Use(mw.Audit)
|
||||
// userAuditLogHandler := systemhandler.NewSysAuditLogHandler()
|
||||
// r.Get("/list", userAuditLogHandler.List)
|
||||
// r.Post("/list", userAuditLogHandler.PostList)
|
||||
// })
|
||||
|
||||
// r.Route("/role", func(r chi.Router) {
|
||||
// r.Use(mw.Audit)
|
||||
// roleHandler := systemhandler.NewSysRoleHandler()
|
||||
// r.Get("/list", roleHandler.List)
|
||||
// r.Post("/list", roleHandler.PostList)
|
||||
// r.Get("/add", roleHandler.Add)
|
||||
// r.Get("/edit", roleHandler.Edit)
|
||||
// r.Post("/save", roleHandler.Save)
|
||||
// r.Post("/dtree", roleHandler.DTree)
|
||||
// r.Post("/refresh", roleHandler.Refresh)
|
||||
// r.Post("/rebuild_parent_path", roleHandler.RebuildParentPath)
|
||||
// r.Post("/refresh_role_menus", roleHandler.RefreshRoleMenus)
|
||||
// r.Post("/xm_select", roleHandler.XmSelect)
|
||||
// r.Get("/set_menu", roleHandler.SetMenu)
|
||||
// r.Post("/set_menu", roleHandler.PostSetMenu)
|
||||
// })
|
||||
|
||||
r.Get("/menus", handler.SystemHandler().MenuHandler().Menus)
|
||||
// r.Route("/menu", func(r chi.Router) {
|
||||
// r.Use(mw.Audit)
|
||||
// r.Get("/tree", menuHandler.Tree)
|
||||
// r.Get("/list", menuHandler.List)
|
||||
// r.Post("/list", menuHandler.PostList)
|
||||
// r.Get("/add", menuHandler.Add)
|
||||
// r.Get("/add_children", menuHandler.AddChildren)
|
||||
// r.Get("/edit", menuHandler.Edit)
|
||||
// r.Post("/save", menuHandler.Save)
|
||||
// r.Post("/xm_select_tree", menuHandler.XmSelectTree)
|
||||
// r.Post("/refresh_cache", menuHandler.Refresh)
|
||||
// })
|
||||
|
||||
// // 类别
|
||||
// r.Route("/category", func(r chi.Router) {
|
||||
// r.Use(mw.Audit)
|
||||
// categoryHandler := categoryhandler.NewCategoryHandler()
|
||||
// r.Get("/list", categoryHandler.List)
|
||||
// r.Post("/list", categoryHandler.PostList)
|
||||
// r.Get("/add", categoryHandler.Add)
|
||||
// r.Get("/add_children", categoryHandler.AddChildren)
|
||||
// r.Get("/edit", categoryHandler.Edit)
|
||||
// r.Post("/save", categoryHandler.Save)
|
||||
// r.Post("/dtree", categoryHandler.DTree)
|
||||
// r.Post("/xmselect", categoryHandler.XmSelect)
|
||||
// r.Post("/refresh", categoryHandler.Refresh)
|
||||
// r.Post("/rebuild_parent_path", categoryHandler.RebuildParentPath)
|
||||
// })
|
||||
})
|
||||
})
|
||||
|
||||
return r
|
||||
}
|
||||
15
internal/erpserver/model/req/user.go
Normal file
15
internal/erpserver/model/req/user.go
Normal file
@@ -0,0 +1,15 @@
|
||||
package req
|
||||
|
||||
type Login struct {
|
||||
Email string `json:"email"`
|
||||
Password string `json:"password"`
|
||||
Captcha string `json:"captcha"`
|
||||
CaptchaID string `json:"captcha_id"`
|
||||
|
||||
// 平台信息
|
||||
Os string `json:"os"`
|
||||
Ip string `json:"ip"`
|
||||
Browser string `json:"browser"`
|
||||
Referrer string `json:"referrer"`
|
||||
Url string `json:"url"`
|
||||
}
|
||||
14
internal/erpserver/model/view/system.go
Normal file
14
internal/erpserver/model/view/system.go
Normal file
@@ -0,0 +1,14 @@
|
||||
package view
|
||||
|
||||
type LayuiTree struct {
|
||||
ID string `json:"id"`
|
||||
Title string `json:"title"`
|
||||
Spread bool `json:"spread"`
|
||||
Children []*LayuiTree `json:"children"`
|
||||
}
|
||||
|
||||
type XmSelectTree struct {
|
||||
Name string `json:"name"`
|
||||
Value string `json:"value"`
|
||||
Children []*XmSelectTree `json:"children"`
|
||||
}
|
||||
Reference in New Issue
Block a user