[feat] job 和 grpc 支持

controller.go

@@ -8,50 +8,143 @@ import (
 )
 
 type Controller interface {
-	AuthDB() ResAuthDB
-	Execute(GlobalParams) any
-	ExecuteWs(WsClient, GlobalParams) error
-	WsClientId(GlobalParams) WsClientID
-	WsClientGroup(GlobalParams) WsClientGroup
-	Call(code string, params map[string]any, category ...RouteCategory) (HttpResponse, error)
+	Container() *do.Injector
+	// 派遣一个任务
+	Dispatch(job string, payload any, u User) error
 }
 
 type GRpController interface {
+	Controller
+	// 获取 gRPC 服务的描述信息
 	GetGRpcServiceDesc() *grpc.ServiceDesc
 }
 
+type JobController interface {
+	Controller
+	// 任务处理
+	Handle(any) 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)
+}
+
+type WsController interface {
+	HttpController
+	// 获取 Ws 客户端
+	WsClient() WsClient
+	// 获取 Ws 标识
+	WsClientId(GlobalParams) WsClientID
+	// 获取 Ws 分组
+	WsClientGroup(GlobalParams) WsClientGroup
+}
+
 type BaseController struct {
-	User      User
-	Container *do.Injector
-	Request   *http.Request
-	request   *http.Request
-	Route     Route
+	container *do.Injector
+}
+
+type GRpcBaseController struct {
+	Controller
+}
+
+type JobBaseController struct {
+	Controller
+}
+
+type HttpBaseController struct {
+	Controller
+	request *http.Request
+}
+
+type WsBaseController struct {
+	HttpController
+	ws WsClient
 }
 
 var defaultWsClientGroup WsClientGroup = "__DEFAULT__"
 var wsClientID WsClientID = 0
 
-func (BaseController) AuthDB() ResAuthDB {
-	return ResAuthOff
+func (this BaseController) Container() *do.Injector {
+	return this.container
 }
 
-func (BaseController) Execute(params GlobalParams) any {
+func (this BaseController) Dispatch(job string, payload any, u User) error {
+	return do.MustInvoke[Job](this.Container()).Dispatch(job, payload, u)
+}
+
+func NewHttpBaseController(container *do.Injector, request *http.Request) HttpController {
+	return &HttpBaseController{
+		Controller: &BaseController{container},
+		request:    request,
+	}
+}
+
+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 (BaseController) ExecuteWs(wc WsClient, params GlobalParams) error {
-	return nil
+func NewWsBaseController(ws WsClient, container *do.Injector, request *http.Request) WsController {
+	return &WsBaseController{
+		HttpController: NewHttpBaseController(container, request),
+		ws:             ws,
+	}
 }
 
-func (BaseController) WsClientId(GlobalParams) WsClientID {
+func (this WsBaseController) WsClient() WsClient {
+	return this.ws
+}
+
+func (WsBaseController) WsClientId(GlobalParams) WsClientID {
 	wsClientID++
 	return wsClientID
 }
 
-func (BaseController) WsClientGroup(GlobalParams) WsClientGroup {
+func (WsBaseController) WsClientGroup(GlobalParams) WsClientGroup {
 	return defaultWsClientGroup
 }
 
-func (this BaseController) Call(code string, params map[string]any, category ...RouteCategory) (HttpResponse, error) {
-	return do.MustInvoke[Router](this.Container).Call(this.Request, code, params, category...)
+func NewGRpBaseController(container *do.Injector) JobController {
+	return &JobBaseController{
+		Controller: &BaseController{container},
+	}
+}
+
+func (JobBaseController) Handle(any) error {
+	return nil
+}
+
+func NewJobBaseController(container *do.Injector) GRpController {
+	return &GRpcBaseController{
+		Controller: &BaseController{container},
+	}
+}
+
+func (GRpcBaseController) GetGRpcServiceDesc() *grpc.ServiceDesc {
+	return nil
 }