141 lines
2.9 KiB
Go
141 lines
2.9 KiB
Go
package contracts
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/samber/do"
|
|
)
|
|
|
|
type Controller interface {
|
|
// 初始化
|
|
Init() error
|
|
// 获取 Container
|
|
Container() *do.Injector
|
|
// 派遣一个任务
|
|
Dispatch(job string, payload any, u User) error
|
|
}
|
|
|
|
type HttpController interface {
|
|
Controller
|
|
|
|
// 获取用户信息
|
|
User() User
|
|
|
|
// 获取请求信息
|
|
Request() *http.Request
|
|
|
|
// 请求处理
|
|
Execute(GlobalParams) any
|
|
|
|
// 路由信息
|
|
Route() Route
|
|
|
|
// 内部调用
|
|
Call(code string, params map[string]any, category ...RouteCategory) (HttpResponse, error)
|
|
// 内部调用
|
|
// Invoke()
|
|
|
|
// 数据库权限
|
|
AuthDB() ResAuthDB
|
|
}
|
|
|
|
type WsController interface {
|
|
HttpController
|
|
// 请求处理
|
|
ExecuteWS(GlobalParams) error
|
|
// 获取 Ws 客户端
|
|
WsClient() WsClient
|
|
// 获取 Ws 标识
|
|
WsClientId(GlobalParams) WsClientID
|
|
// 获取 Ws 分组
|
|
WsClientGroup(GlobalParams) WsClientGroup
|
|
}
|
|
|
|
type BaseController struct {
|
|
container *do.Injector
|
|
}
|
|
|
|
type HttpBaseController struct {
|
|
Controller
|
|
request *http.Request
|
|
}
|
|
|
|
type WsBaseController struct {
|
|
HttpController
|
|
ws WsClient
|
|
}
|
|
|
|
var defaultWsClientGroup WsClientGroup = "__DEFAULT__"
|
|
var wsClientID WsClientID = 0
|
|
|
|
func (this BaseController) Container() *do.Injector {
|
|
return this.container
|
|
}
|
|
|
|
func (BaseController) Init() error {
|
|
return nil
|
|
}
|
|
|
|
func (this BaseController) Dispatch(job string, payload any, u User) error {
|
|
return do.MustInvoke[JobService](this.Container()).Dispatch(job, payload, u)
|
|
}
|
|
|
|
func (this HttpBaseController) Request() *http.Request {
|
|
return this.request
|
|
}
|
|
|
|
func (this HttpBaseController) Route() Route {
|
|
return this.Request().Context().Value(RouteCtx{Name: "Route"}).(Route)
|
|
}
|
|
|
|
func (this HttpBaseController) User() User {
|
|
return this.Request().Context().Value(RouteCtx{Name: "User"}).(User)
|
|
}
|
|
|
|
func (this HttpBaseController) Call(code string, params map[string]any, category ...RouteCategory) (HttpResponse, error) {
|
|
return do.MustInvoke[Router](this.Container()).Call(this.Request(), code, params, category...)
|
|
}
|
|
|
|
func (HttpBaseController) Execute(params GlobalParams) any {
|
|
return nil
|
|
}
|
|
|
|
func (HttpBaseController) AuthDB() ResAuthDB {
|
|
return ResAuthOn
|
|
}
|
|
|
|
func (WsBaseController) ExecuteWS(params GlobalParams) error {
|
|
return nil
|
|
}
|
|
|
|
func (this WsBaseController) WsClient() WsClient {
|
|
return this.ws
|
|
}
|
|
|
|
func (WsBaseController) WsClientId(GlobalParams) WsClientID {
|
|
wsClientID++
|
|
return wsClientID
|
|
}
|
|
|
|
func (WsBaseController) WsClientGroup(GlobalParams) WsClientGroup {
|
|
return defaultWsClientGroup
|
|
}
|
|
|
|
func NewBaseController(container *do.Injector) Controller {
|
|
return &BaseController{container}
|
|
}
|
|
|
|
func NewHttpBaseController(container *do.Injector, request *http.Request) HttpController {
|
|
return &HttpBaseController{
|
|
Controller: &BaseController{container},
|
|
request: request,
|
|
}
|
|
}
|
|
|
|
func NewWsBaseController(ws WsClient, ctr HttpController) WsController {
|
|
return &WsBaseController{
|
|
HttpController: ctr,
|
|
ws: ws,
|
|
}
|
|
}
|