This commit is contained in:
2025-04-01 14:13:41 +08:00
parent d7183fcc00
commit 7857015405
97 changed files with 1917 additions and 6934 deletions

View File

@@ -6,7 +6,7 @@ import (
"net/http"
"management/internal/db/model/dto"
"management/internal/global/auth"
"management/internal/pkg/know"
)
var defaultMenus = map[string]bool{
@@ -59,8 +59,8 @@ func (m *middleware) Authorize(next http.Handler) http.Handler {
}
func (m *middleware) isLogin(ctx context.Context) (*dto.AuthorizeUser, bool) {
if exists := m.session.Exists(ctx, auth.StoreName); exists {
b := m.session.GetBytes(ctx, auth.StoreName)
if exists := m.session.Exists(ctx, know.StoreName); exists {
b := m.session.GetBytes(ctx, know.StoreName)
var user dto.AuthorizeUser
if err := json.Unmarshal(b, &user); err != nil {
return nil, false
@@ -73,8 +73,8 @@ func (m *middleware) isLogin(ctx context.Context) (*dto.AuthorizeUser, bool) {
func (m *middleware) AuthUser(ctx context.Context) dto.AuthorizeUser {
var user dto.AuthorizeUser
if exists := m.session.Exists(ctx, auth.StoreName); exists {
b := m.session.GetBytes(ctx, auth.StoreName)
if exists := m.session.Exists(ctx, know.StoreName); exists {
b := m.session.GetBytes(ctx, know.StoreName)
_ = json.Unmarshal(b, &user)
}
return user

View File

@@ -1,8 +1,10 @@
package middleware
import (
"context"
"net/http"
"management/internal/db/model/dto"
systemv1 "management/internal/erpserver/biz/v1/system"
"management/internal/pkg/session"
)
@@ -12,6 +14,7 @@ type IMiddleware interface {
NoSurf(next http.Handler) http.Handler
LoadSession(next http.Handler) http.Handler
Authorize(next http.Handler) http.Handler
AuthUser(ctx context.Context) dto.AuthorizeUser
}
type middleware struct {