feat: add ContextLoadFunc for custom context loading in templates and configuration

- Introduced `ContextLoadFunc` to `config.go`, `template.tmpl`, and `template.go`.
- Updated `main.go` to allow `context_load_func` as a configurable flag.

Author: tbxark

generate/http/config.go

@@ -18,6 +18,7 @@ type Config struct {
 	DataRespType  protogen.GoIdent
 
 	ServerHandlerFunc protogen.GoIdent
+	ContextLoadFunc   string
 }
 
 type GenConfig struct {
@@ -37,6 +38,7 @@ func NewGenConf(g *protogen.GeneratedFile, conf *Config) *GenConfig {
 		DataResponseType:  g.QualifiedGoIdent(conf.DataRespType),
 
 		ServerHandlerWrapperFunc: g.QualifiedGoIdent(conf.ServerHandlerFunc),
+		ContextLoadFunc:          conf.ContextLoadFunc,
 	}
 	genConf := &GenConfig{
 		omitempty:       conf.Omitempty,

generate/template/template.go

@@ -73,6 +73,7 @@ type PackageDesc struct {
 	ValidateFunc    string
 
 	ServerHandlerWrapperFunc string
+	ContextLoadFunc          string
 }
 
 func (s *ServiceDesc) Execute() (string, error) {

generate/template/template.tmpl

@@ -54,7 +54,7 @@ func _{{$svrType}}_{{.Name}}{{.Num}}_HTTP_Handler(srv {{$svrType}}HTTPServer) {{
             return nil, err
         }
         {{- end}}
-		out, err := srv.{{.Name}}(ctx, &in)
+		out, err := srv.{{.Name}}(ctx{{$packageDesc.ContextLoadFunc}}, &in)
 		if err != nil {
 			return nil, err
 		}

main.go

@@ -27,9 +27,10 @@ var (
 	templateFile      = flag.String("template_file", "", "template file, if not set, use default template")
 	swaggerAuthHeader = flag.String("swagger_auth_header", swaggerAuthComment, "swagger auth header")
 
-	routerType  = flag.String("router_type", defaultHTTPxPackage+";Router", "router type")
-	contextType = flag.String("context_type", defaultHTTPxPackage+";Context", "context type")
-	handlerType = flag.String("handler_type", defaultHTTPxPackage+";Handler", "handler type")
+	routerType      = flag.String("router_type", defaultHTTPxPackage+";Router", "router type")
+	contextType     = flag.String("context_type", defaultHTTPxPackage+";Context", "context type")
+	handlerType     = flag.String("handler_type", defaultHTTPxPackage+";Handler", "handler type")
+	contextLoadFunc = flag.String("context_load_func", ".Context()", "context load func")
 
 	errorRespType     = flag.String("error_resp_type", defaultHTTPzPackage+";ErrorResponse", "error response type")
 	dataRespType      = flag.String("data_resp_type", defaultHTTPzPackage+";DataResponse", "data response type, must support generic")
@@ -119,6 +120,7 @@ func extractConfig() (*http.Config, error) {
 		DataRespType:  _dataRespType,
 
 		ServerHandlerFunc: _serverHandlerFunc,
+		ContextLoadFunc:   *contextLoadFunc,
 	}
 	return conf, nil
 }