1. package main
    2. import (
    3.     WebApplication "github.com/yoyofx/yoyogo/web"
    4.         ......
    5. )
    7. func main() {
    8.     webHost := CreateCustomBuilder().Build()
    9.     webHost.Run()
    10. }
    13. // 自定义HostBuilder并支持 MVC 和 自动参数绑定功能，简单情况也可以直接使用CreateDefaultBuilder 。
    14. func CreateCustomBuilder() *abstractions.HostBuilder {
    16.     configuration := abstractions.NewConfigurationBuilder().
    17.         AddEnvironment().
    18.         AddYamlFile("config").Build()
    20.     return WebApplication.NewWebHostBuilder().
    21.         UseConfiguration(configuration).
    22.         Configure(func(app *WebApplication.WebApplicationBuilder) {
    23.             app.UseMiddleware(middlewares.NewCORS())
    24.             //WebApplication.UseMiddleware(middlewares.NewRequestTracker())
    25.             app.UseStaticAssets()
    26.             app.UseEndpoints(registerEndpointRouterConfig)
    27.             app.UseMvc(func(builder *mvc.ControllerBuilder) {
    28.                 //builder.AddViews(&view.Option{Path: "./Static/templates"})
    29.                 builder.AddViewsByConfig()
    30.                 builder.AddController(contollers.NewUserController)
    31.                 builder.AddFilter("/v1/user/info", &contollers.TestActionFilter{})
    32.             })
    33.         }).
    34.         ConfigureServices(func(serviceCollection *dependencyinjection.ServiceCollection) {
    35.             serviceCollection.AddTransientByImplements(models.NewUserAction, new(models.IUserAction))
    36.             serviceCollection.AddSingletonByImplementsAndName("db1", datasources.NewMysqlDataSource, new(abstractions.IDataSource))
    37.             serviceCollection.AddSingletonByImplementsAndName("redis1", datasources.NewRedis, new(abstractions.IDataSource))
    39.             //eureka.UseServiceDiscovery(serviceCollection)
    40.             //consul.UseServiceDiscovery(serviceCollection)
    41.             nacos.UseServiceDiscovery(serviceCollection)
    42.         }).
    43.         OnApplicationLifeEvent(getApplicationLifeEvent)
    44. }
    46. //region endpoint 路由绑定函数
    47. func registerEndpoints(rb router.IRouterBuilder) {
    48.     Endpoints.UseHealth(rb)
    49.     Endpoints.UseViz(rb)
    50.     Endpoints.UsePrometheus(rb)
    51.     Endpoints.UsePprof(rb)
    52.     Endpoints.UseJwt(rb)
    54.     rb.GET("/error", func(ctx *context.HttpContext) {
    55.         panic("http get error")
    56.     })
    58.     //POST 请求: /info/:id ?q1=abc&username=123
    59.     rb.POST("/info/:id", func(ctx *context.HttpContext) {
    60.         qs_q1 := ctx.Query("q1")
    61.         pd_name := ctx.Param("username")
    63.         userInfo := &UserInfo{}
    65.         _ = ctx.Bind(userInfo) // 手动绑定请求对象
    67.         strResult := fmt.Sprintf("Name:%s , Q1:%s , bind: %s", pd_name, qs_q1, userInfo)
    69.         ctx.JSON(200, context.H{"info": "hello world", "result": strResult})
    70.     })
    72.     // 路由组功能实现绑定 GET 请求:  /v1/api/info
    73.     rb.Group("/v1/api", func(router *router.RouterGroup) {
    74.         router.GET("/info", func(ctx *context.HttpContext) {
    75.             ctx.JSON(200, context.H{"info": "ok"})
    76.         })
    77.     })
    79.     // GET 请求: HttpContext.RequiredServices获取IOC对象
    80.     rb.GET("/ioc", func(ctx *context.HttpContext) {
    81.         var userAction models.IUserAction
    82.         _ = ctx.RequiredServices.GetService(&userAction)
    83.         ctx.JSON(200, context.H{"info": "ok " + userAction.Login("zhang")})
    84.     })
    85. }
    87. //endregion
    89. //region 请求对象
    90. type UserInfo struct {
    91.     UserName string `param:"username"`
    92.     Number   string `param:"q1"`
    93.     Id       string `param:"id"`
    94. }
    96. // ----------------------------------------- MVC 定义 ------------------------------------------------------
    98. // 定义Controller
    99. type UserController struct {
    100.     *mvc.ApiController
    101.     userAction models.IUserAction // IOC 对象参数
    102. }
    104. // 构造器依赖注入
    105. func NewUserController(userAction models.IUserAction) *UserController {
    106.     return &UserController{userAction: userAction}
    107. }
    109. // 请求对象的参数化绑定
    110. type RegiserRequest struct {
    111.     mvc.RequestBody
    112.     UserName string `param:"username"`
    113.     Password string `param:"password"`
    114. }
    116. // Register函数自动绑定参数
    117. func (this *UserController) Register(ctx *context.HttpContext, request *RegiserRequest) actionresult.IActionResult {
    118.     result := mvc.ApiResult{Success: true, Message: "ok", Data: request}
    119.     return actionresult.Json{Data: result}
    120. }
    122. // use userAction interface by ioc  
    123. func (this *UserController) GetInfo() mvc.ApiResult {
    124.     return this.OK(this.userAction.Login("zhang"))
    125. }
    127. // Web程序的开始与停止事件
    128. func fireApplicationLifeEvent(life *abstractions.ApplicationLife) {
    129.     printDataEvent := func(event abstractions.ApplicationEvent) {
    130.         fmt.Printf("[yoyogo] Topic: %s; Event: %v\n", event.Topic, event.Data)
    131.     }
    132.     for {
    133.         select {
    134.         case ev := <-life.ApplicationStarted:
    135.             go printDataEvent(ev)
    136.         case ev := <-life.ApplicationStopped:
    137.             go printDataEvent(ev)
    138.             break
    139.         }
    140.     }
    141. }