mirror of https://github.com/rocboss/paopao-ce
Michael Li
3 years ago
parent
6650d089b8
commit
4dae86cc9c
@ -1 +1,2 @@
|
||||
### RESTful API for paopao-ce use [go-mir](https://github.com/alimy/mir) to generate service interface code automatic.
|
||||
### Mirc for paopao-ce
|
||||
RESTful API for paopao-ce use [go-mir](https://github.com/alimy/mir) to generate service interface code automatic.
|
||||
|
||||
@ -0,0 +1,127 @@
|
||||
// Code generated by go-mir. DO NOT EDIT.
|
||||
|
||||
package v1
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"github.com/alimy/mir/v3"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
type LoginReq struct {
|
||||
AgentInfo AgentInfo `json:"agent_info"`
|
||||
Name string `json:"name"`
|
||||
Passwd string `json:"passwd"`
|
||||
}
|
||||
|
||||
type AgentInfo struct {
|
||||
Platform string `json:"platform"`
|
||||
UserAgent string `json:"user_agent"`
|
||||
}
|
||||
|
||||
type LoginResp struct {
|
||||
UserInfo
|
||||
ServerInfo ServerInfo `json:"server_info"`
|
||||
JwtToken string `json:"jwt_token"`
|
||||
}
|
||||
|
||||
type ServerInfo struct {
|
||||
ApiVer string `json:"api_ver"`
|
||||
}
|
||||
|
||||
type UserInfo struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
|
||||
type User interface {
|
||||
// Chain provide handlers chain for gin
|
||||
Chain() gin.HandlersChain
|
||||
|
||||
Logout(c *gin.Context) mir.Error
|
||||
Login(c *gin.Context, req *LoginReq) (*LoginResp, mir.Error)
|
||||
|
||||
mustEmbedUnimplementedUserServant()
|
||||
}
|
||||
|
||||
type UserBinding interface {
|
||||
BindLogin(c *gin.Context) (*LoginReq, mir.Error)
|
||||
|
||||
mustEmbedUnimplementedUserBinding()
|
||||
}
|
||||
|
||||
type UserRender interface {
|
||||
RenderLogout(c *gin.Context, err mir.Error)
|
||||
RenderLogin(c *gin.Context, data *LoginResp, err mir.Error)
|
||||
|
||||
mustEmbedUnimplementedUserRender()
|
||||
}
|
||||
|
||||
// RegisterUserServant register User servant to gin
|
||||
func RegisterUserServant(e *gin.Engine, s User, b UserBinding, r UserRender) {
|
||||
router := e.Group("r/v1")
|
||||
// use chain for router
|
||||
middlewares := s.Chain()
|
||||
router.Use(middlewares...)
|
||||
|
||||
// register routes info to router
|
||||
router.Handle("POST", "/user/logout/", func(c *gin.Context) {
|
||||
r.RenderLogout(c, s.Logout(c))
|
||||
})
|
||||
|
||||
router.Handle("POST", "/user/login/", func(c *gin.Context) {
|
||||
req, err := b.BindLogin(c)
|
||||
if err != nil {
|
||||
r.RenderLogin(c, nil, err)
|
||||
}
|
||||
resp, err := s.Login(c, req)
|
||||
r.RenderLogin(c, resp, err)
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
// UnimplementedUserServant can be embedded to have forward compatible implementations.
|
||||
type UnimplementedUserServant struct {
|
||||
}
|
||||
|
||||
func (UnimplementedUserServant) Chain() gin.HandlersChain {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (UnimplementedUserServant) Logout(c *gin.Context) mir.Error {
|
||||
return mir.Errorln(http.StatusNotImplemented, http.StatusText(http.StatusNotImplemented))
|
||||
}
|
||||
|
||||
func (UnimplementedUserServant) Login(c *gin.Context, req *LoginReq) (*LoginResp, mir.Error) {
|
||||
return nil, mir.Errorln(http.StatusNotImplemented, http.StatusText(http.StatusNotImplemented))
|
||||
}
|
||||
|
||||
func (UnimplementedUserServant) mustEmbedUnimplementedUserServant() {}
|
||||
|
||||
// UnimplementedUserRender can be embedded to have forward compatible implementations.
|
||||
type UnimplementedUserRender struct {
|
||||
RenderAny func(*gin.Context, any, mir.Error)
|
||||
}
|
||||
|
||||
func (r *UnimplementedUserRender) RenderLogout(c *gin.Context, err mir.Error) {
|
||||
r.RenderAny(c, nil, err)
|
||||
}
|
||||
|
||||
func (r *UnimplementedUserRender) RenderLogin(c *gin.Context, data *LoginResp, err mir.Error) {
|
||||
r.RenderAny(c, data, err)
|
||||
}
|
||||
|
||||
func (r *UnimplementedUserRender) mustEmbedUnimplementedUserRender() {}
|
||||
|
||||
// UnimplementedUserBinding can be embedded to have forward compatible implementations.
|
||||
type UnimplementedUserBinding struct {
|
||||
BindAny func(*gin.Context, any) mir.Error
|
||||
}
|
||||
|
||||
func (b *UnimplementedUserBinding) BindLogin(c *gin.Context) (*LoginReq, mir.Error) {
|
||||
obj := new(LoginReq)
|
||||
err := b.BindAny(c, obj)
|
||||
return obj, err
|
||||
}
|
||||
|
||||
func (b *UnimplementedUserBinding) mustEmbedUnimplementedUserBinding() {}
|
||||
@ -1,4 +1,4 @@
|
||||
### Manager系列RESTful API相关定义文件
|
||||
本目录包含 Manager后台运维相关API定义文件。
|
||||
### Admin系列RESTful API相关定义文件
|
||||
本目录包含 Admin后台运维相关API定义文件。
|
||||
|
||||
* v1 - v1版本API
|
||||
|
||||
@ -0,0 +1,4 @@
|
||||
### Bot系列RESTful API相关定义文件
|
||||
本目录包含 Bot相关API定义文件。
|
||||
|
||||
* v1 - v1版本API
|
||||
@ -0,0 +1,42 @@
|
||||
package v1
|
||||
|
||||
import (
|
||||
. "github.com/alimy/mir/v3"
|
||||
. "github.com/alimy/mir/v3/engine"
|
||||
)
|
||||
|
||||
func init() {
|
||||
AddEntry(new(User))
|
||||
}
|
||||
|
||||
type AgentInfo struct {
|
||||
Platform string `json:"platform"`
|
||||
UserAgent string `json:"user_agent"`
|
||||
}
|
||||
|
||||
type ServerInfo struct {
|
||||
ApiVer string `json:"api_ver"`
|
||||
}
|
||||
|
||||
type UserInfo struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
|
||||
type LoginReq struct {
|
||||
AgentInfo AgentInfo `json:"agent_info"`
|
||||
Name string `json:"name"`
|
||||
Passwd string `json:"passwd"`
|
||||
}
|
||||
|
||||
type LoginResp struct {
|
||||
UserInfo
|
||||
ServerInfo ServerInfo `json:"server_info"`
|
||||
JwtToken string `json:"jwt_token"`
|
||||
}
|
||||
|
||||
type User struct {
|
||||
Chain Chain `mir:"-"`
|
||||
Group Group `mir:"r/v1"`
|
||||
Login func(Post, LoginReq) LoginResp `mir:"/user/login/"`
|
||||
Logout func(Post) `mir:"/user/logout/"`
|
||||
}
|
||||
@ -0,0 +1,15 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package admin
|
||||
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
api "github.com/rocboss/paopao-ce/internal/mirc/auto/api/m/v1"
|
||||
)
|
||||
|
||||
// RouteWeb register Manager route
|
||||
func RouteManager(e *gin.Engine) {
|
||||
api.RegisterUserServant(e, newUserSrv(), newUserBinding(), newUserRender())
|
||||
}
|
||||
@ -0,0 +1,51 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package admin
|
||||
|
||||
import (
|
||||
api "github.com/rocboss/paopao-ce/internal/mirc/auto/api/m/v1"
|
||||
"github.com/rocboss/paopao-ce/internal/servants/base"
|
||||
)
|
||||
|
||||
var (
|
||||
_ api.User = (*userSrv)(nil)
|
||||
_ api.UserBinding = (*userBinding)(nil)
|
||||
_ api.UserRender = (*userRender)(nil)
|
||||
)
|
||||
|
||||
type userSrv struct {
|
||||
base.BaseServant
|
||||
api.UnimplementedUserServant
|
||||
}
|
||||
|
||||
type userBinding struct {
|
||||
base.BaseBinding
|
||||
*api.UnimplementedUserBinding
|
||||
}
|
||||
|
||||
type userRender struct {
|
||||
base.BaseRender
|
||||
*api.UnimplementedUserRender
|
||||
}
|
||||
|
||||
func newUserSrv() api.User {
|
||||
return &userSrv{}
|
||||
}
|
||||
|
||||
func newUserBinding() api.UserBinding {
|
||||
return &userBinding{
|
||||
UnimplementedUserBinding: &api.UnimplementedUserBinding{
|
||||
BindAny: base.BindAny,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func newUserRender() api.UserRender {
|
||||
return &userRender{
|
||||
UnimplementedUserRender: &api.UnimplementedUserRender{
|
||||
RenderAny: base.RenderAny,
|
||||
},
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,15 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package bot
|
||||
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
api "github.com/rocboss/paopao-ce/internal/mirc/auto/api/r/v1"
|
||||
)
|
||||
|
||||
// RouteBot register Bot route
|
||||
func RouteBot(e *gin.Engine) {
|
||||
api.RegisterUserServant(e, newUserSrv(), newUserBinding(), newUserRender())
|
||||
}
|
||||
@ -0,0 +1,51 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package bot
|
||||
|
||||
import (
|
||||
api "github.com/rocboss/paopao-ce/internal/mirc/auto/api/r/v1"
|
||||
"github.com/rocboss/paopao-ce/internal/servants/base"
|
||||
)
|
||||
|
||||
var (
|
||||
_ api.User = (*userSrv)(nil)
|
||||
_ api.UserBinding = (*userBinding)(nil)
|
||||
_ api.UserRender = (*userRender)(nil)
|
||||
)
|
||||
|
||||
type userSrv struct {
|
||||
base.BaseServant
|
||||
api.UnimplementedUserServant
|
||||
}
|
||||
|
||||
type userBinding struct {
|
||||
base.BaseBinding
|
||||
*api.UnimplementedUserBinding
|
||||
}
|
||||
|
||||
type userRender struct {
|
||||
base.BaseRender
|
||||
*api.UnimplementedUserRender
|
||||
}
|
||||
|
||||
func newUserSrv() api.User {
|
||||
return &userSrv{}
|
||||
}
|
||||
|
||||
func newUserBinding() api.UserBinding {
|
||||
return &userBinding{
|
||||
UnimplementedUserBinding: &api.UnimplementedUserBinding{
|
||||
BindAny: base.BindAny,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func newUserRender() api.UserRender {
|
||||
return &userRender{
|
||||
UnimplementedUserRender: &api.UnimplementedUserRender{
|
||||
RenderAny: base.RenderAny,
|
||||
},
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,51 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package localoss
|
||||
|
||||
import (
|
||||
api "github.com/rocboss/paopao-ce/internal/mirc/auto/api/s/v1"
|
||||
"github.com/rocboss/paopao-ce/internal/servants/base"
|
||||
)
|
||||
|
||||
var (
|
||||
_ api.User = (*userSrv)(nil)
|
||||
_ api.UserBinding = (*userBinding)(nil)
|
||||
_ api.UserRender = (*userRender)(nil)
|
||||
)
|
||||
|
||||
type userSrv struct {
|
||||
base.BaseServant
|
||||
api.UnimplementedUserServant
|
||||
}
|
||||
|
||||
type userBinding struct {
|
||||
base.BaseBinding
|
||||
*api.UnimplementedUserBinding
|
||||
}
|
||||
|
||||
type userRender struct {
|
||||
base.BaseRender
|
||||
*api.UnimplementedUserRender
|
||||
}
|
||||
|
||||
func newUserSrv() api.User {
|
||||
return &userSrv{}
|
||||
}
|
||||
|
||||
func newUserBinding() api.UserBinding {
|
||||
return &userBinding{
|
||||
UnimplementedUserBinding: &api.UnimplementedUserBinding{
|
||||
BindAny: base.BindAny,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func newUserRender() api.UserRender {
|
||||
return &userRender{
|
||||
UnimplementedUserRender: &api.UnimplementedUserRender{
|
||||
RenderAny: base.RenderAny,
|
||||
},
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,15 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package space
|
||||
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
api "github.com/rocboss/paopao-ce/internal/mirc/auto/api/x/v1"
|
||||
)
|
||||
|
||||
// RouteWeb register SpaceX route
|
||||
func RouteSpaceX(e *gin.Engine) {
|
||||
api.RegisterUserServant(e, newUserSrv(), newUserBinding(), newUserRender())
|
||||
}
|
||||
@ -0,0 +1,51 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package space
|
||||
|
||||
import (
|
||||
api "github.com/rocboss/paopao-ce/internal/mirc/auto/api/x/v1"
|
||||
"github.com/rocboss/paopao-ce/internal/servants/base"
|
||||
)
|
||||
|
||||
var (
|
||||
_ api.User = (*userSrv)(nil)
|
||||
_ api.UserBinding = (*userBinding)(nil)
|
||||
_ api.UserRender = (*userRender)(nil)
|
||||
)
|
||||
|
||||
type userSrv struct {
|
||||
base.BaseServant
|
||||
api.UnimplementedUserServant
|
||||
}
|
||||
|
||||
type userBinding struct {
|
||||
base.BaseBinding
|
||||
*api.UnimplementedUserBinding
|
||||
}
|
||||
|
||||
type userRender struct {
|
||||
base.BaseRender
|
||||
*api.UnimplementedUserRender
|
||||
}
|
||||
|
||||
func newUserSrv() api.User {
|
||||
return &userSrv{}
|
||||
}
|
||||
|
||||
func newUserBinding() api.UserBinding {
|
||||
return &userBinding{
|
||||
UnimplementedUserBinding: &api.UnimplementedUserBinding{
|
||||
BindAny: base.BindAny,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func newUserRender() api.UserRender {
|
||||
return &userRender{
|
||||
UnimplementedUserRender: &api.UnimplementedUserRender{
|
||||
RenderAny: base.RenderAny,
|
||||
},
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,86 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
|
||||
"github.com/fatih/color"
|
||||
"github.com/gin-contrib/cors"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/rocboss/paopao-ce/internal/conf"
|
||||
"github.com/rocboss/paopao-ce/internal/servants"
|
||||
)
|
||||
|
||||
type adminService struct {
|
||||
*baseHttpService
|
||||
}
|
||||
|
||||
func (s *adminService) Name() string {
|
||||
return "AdminService"
|
||||
}
|
||||
|
||||
func (s *adminService) Init() error {
|
||||
s.registerRoute(servants.RegisterAdminServants)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *adminService) String() string {
|
||||
return fmt.Sprintf("listen on %s\n", color.GreenString("http://%s:%s", conf.AdminServerSetting.HttpIp, conf.AdminServerSetting.HttpPort))
|
||||
}
|
||||
|
||||
func newAdminEngine() *gin.Engine {
|
||||
e := gin.New()
|
||||
e.HandleMethodNotAllowed = true
|
||||
e.Use(gin.Logger())
|
||||
e.Use(gin.Recovery())
|
||||
|
||||
// 跨域配置
|
||||
corsConfig := cors.DefaultConfig()
|
||||
corsConfig.AllowAllOrigins = true
|
||||
corsConfig.AddAllowHeaders("Authorization")
|
||||
e.Use(cors.New(corsConfig))
|
||||
|
||||
// 默认404
|
||||
e.NoRoute(func(c *gin.Context) {
|
||||
c.JSON(http.StatusNotFound, gin.H{
|
||||
"code": 404,
|
||||
"msg": "Not Found",
|
||||
})
|
||||
})
|
||||
|
||||
// 默认405
|
||||
e.NoMethod(func(c *gin.Context) {
|
||||
c.JSON(http.StatusMethodNotAllowed, gin.H{
|
||||
"code": 405,
|
||||
"msg": "Method Not Allowed",
|
||||
})
|
||||
})
|
||||
|
||||
return e
|
||||
}
|
||||
|
||||
func newAdminService() Service {
|
||||
addr := conf.AdminServerSetting.HttpIp + ":" + conf.AdminServerSetting.HttpPort
|
||||
server := httpServerFrom(addr, func() *httpServer {
|
||||
engine := newAdminEngine()
|
||||
return &httpServer{
|
||||
e: engine,
|
||||
server: &http.Server{
|
||||
Addr: addr,
|
||||
Handler: engine,
|
||||
ReadTimeout: conf.AdminServerSetting.ReadTimeout,
|
||||
WriteTimeout: conf.AdminServerSetting.WriteTimeout,
|
||||
MaxHeaderBytes: 1 << 20,
|
||||
},
|
||||
}
|
||||
})
|
||||
return &adminService{
|
||||
baseHttpService: &baseHttpService{
|
||||
server: server,
|
||||
},
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,86 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
|
||||
"github.com/fatih/color"
|
||||
"github.com/gin-contrib/cors"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/rocboss/paopao-ce/internal/conf"
|
||||
"github.com/rocboss/paopao-ce/internal/servants"
|
||||
)
|
||||
|
||||
type botService struct {
|
||||
*baseHttpService
|
||||
}
|
||||
|
||||
func (s *botService) Name() string {
|
||||
return "BotService"
|
||||
}
|
||||
|
||||
func (s *botService) Init() error {
|
||||
s.registerRoute(servants.RegisterBotServants)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *botService) String() string {
|
||||
return fmt.Sprintf("listen on %s\n", color.GreenString("http://%s:%s", conf.BotServerSetting.HttpIp, conf.BotServerSetting.HttpPort))
|
||||
}
|
||||
|
||||
func newBotEngine() *gin.Engine {
|
||||
e := gin.New()
|
||||
e.HandleMethodNotAllowed = true
|
||||
e.Use(gin.Logger())
|
||||
e.Use(gin.Recovery())
|
||||
|
||||
// 跨域配置
|
||||
corsConfig := cors.DefaultConfig()
|
||||
corsConfig.AllowAllOrigins = true
|
||||
corsConfig.AddAllowHeaders("Authorization")
|
||||
e.Use(cors.New(corsConfig))
|
||||
|
||||
// 默认404
|
||||
e.NoRoute(func(c *gin.Context) {
|
||||
c.JSON(http.StatusNotFound, gin.H{
|
||||
"code": 404,
|
||||
"msg": "Not Found",
|
||||
})
|
||||
})
|
||||
|
||||
// 默认405
|
||||
e.NoMethod(func(c *gin.Context) {
|
||||
c.JSON(http.StatusMethodNotAllowed, gin.H{
|
||||
"code": 405,
|
||||
"msg": "Method Not Allowed",
|
||||
})
|
||||
})
|
||||
|
||||
return e
|
||||
}
|
||||
|
||||
func newBotService() Service {
|
||||
addr := conf.BotServerSetting.HttpIp + ":" + conf.BotServerSetting.HttpPort
|
||||
server := httpServerFrom(addr, func() *httpServer {
|
||||
engine := newBotEngine()
|
||||
return &httpServer{
|
||||
e: engine,
|
||||
server: &http.Server{
|
||||
Addr: addr,
|
||||
Handler: engine,
|
||||
ReadTimeout: conf.BotServerSetting.ReadTimeout,
|
||||
WriteTimeout: conf.BotServerSetting.WriteTimeout,
|
||||
MaxHeaderBytes: 1 << 20,
|
||||
},
|
||||
}
|
||||
})
|
||||
return &botService{
|
||||
baseHttpService: &baseHttpService{
|
||||
server: server,
|
||||
},
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,61 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
|
||||
"github.com/fatih/color"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/rocboss/paopao-ce/internal/conf"
|
||||
"github.com/rocboss/paopao-ce/internal/servants"
|
||||
)
|
||||
|
||||
type localossService struct {
|
||||
*baseHttpService
|
||||
}
|
||||
|
||||
func (s *localossService) Name() string {
|
||||
return "LocalossService"
|
||||
}
|
||||
|
||||
func (s *localossService) Init() error {
|
||||
s.registerRoute(servants.RegisterLocalossServants)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *localossService) String() string {
|
||||
return fmt.Sprintf("listen on %s\n", color.GreenString("http://%s:%s", conf.LocalossServerSetting.HttpIp, conf.LocalossServerSetting.HttpPort))
|
||||
}
|
||||
|
||||
func newLocalossEngine() *gin.Engine {
|
||||
e := gin.New()
|
||||
e.Use(gin.Logger())
|
||||
e.Use(gin.Recovery())
|
||||
return e
|
||||
}
|
||||
|
||||
func newLocalossService() Service {
|
||||
addr := conf.LocalossServerSetting.HttpIp + ":" + conf.LocalossServerSetting.HttpPort
|
||||
server := httpServerFrom(addr, func() *httpServer {
|
||||
engine := newLocalossEngine()
|
||||
return &httpServer{
|
||||
e: engine,
|
||||
server: &http.Server{
|
||||
Addr: addr,
|
||||
Handler: engine,
|
||||
ReadTimeout: conf.LocalossServerSetting.ReadTimeout,
|
||||
WriteTimeout: conf.LocalossServerSetting.WriteTimeout,
|
||||
MaxHeaderBytes: 1 << 20,
|
||||
},
|
||||
}
|
||||
})
|
||||
return &localossService{
|
||||
baseHttpService: &baseHttpService{
|
||||
server: server,
|
||||
},
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,86 @@
|
||||
// Copyright 2022 ROC. All rights reserved.
|
||||
// Use of this source code is governed by a MIT style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
|
||||
"github.com/fatih/color"
|
||||
"github.com/gin-contrib/cors"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/rocboss/paopao-ce/internal/conf"
|
||||
"github.com/rocboss/paopao-ce/internal/servants"
|
||||
)
|
||||
|
||||
type spaceXService struct {
|
||||
*baseHttpService
|
||||
}
|
||||
|
||||
func (s *spaceXService) Name() string {
|
||||
return "WebService"
|
||||
}
|
||||
|
||||
func (s *spaceXService) Init() error {
|
||||
s.registerRoute(servants.RegisterSpaceXServants)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *spaceXService) String() string {
|
||||
return fmt.Sprintf("listen on %s\n", color.GreenString("http://%s:%s", conf.SpaceXServerSetting.HttpIp, conf.SpaceXServerSetting.HttpPort))
|
||||
}
|
||||
|
||||
func newSpaceXEngine() *gin.Engine {
|
||||
e := gin.New()
|
||||
e.HandleMethodNotAllowed = true
|
||||
e.Use(gin.Logger())
|
||||
e.Use(gin.Recovery())
|
||||
|
||||
// 跨域配置
|
||||
corsConfig := cors.DefaultConfig()
|
||||
corsConfig.AllowAllOrigins = true
|
||||
corsConfig.AddAllowHeaders("Authorization")
|
||||
e.Use(cors.New(corsConfig))
|
||||
|
||||
// 默认404
|
||||
e.NoRoute(func(c *gin.Context) {
|
||||
c.JSON(http.StatusNotFound, gin.H{
|
||||
"code": 404,
|
||||
"msg": "Not Found",
|
||||
})
|
||||
})
|
||||
|
||||
// 默认405
|
||||
e.NoMethod(func(c *gin.Context) {
|
||||
c.JSON(http.StatusMethodNotAllowed, gin.H{
|
||||
"code": 405,
|
||||
"msg": "Method Not Allowed",
|
||||
})
|
||||
})
|
||||
|
||||
return e
|
||||
}
|
||||
|
||||
func newSpaceXService() Service {
|
||||
addr := conf.SpaceXServerSetting.HttpIp + ":" + conf.SpaceXServerSetting.HttpPort
|
||||
server := httpServerFrom(addr, func() *httpServer {
|
||||
engine := newSpaceXEngine()
|
||||
return &httpServer{
|
||||
e: engine,
|
||||
server: &http.Server{
|
||||
Addr: addr,
|
||||
Handler: engine,
|
||||
ReadTimeout: conf.SpaceXServerSetting.ReadTimeout,
|
||||
WriteTimeout: conf.SpaceXServerSetting.WriteTimeout,
|
||||
MaxHeaderBytes: 1 << 20,
|
||||
},
|
||||
}
|
||||
})
|
||||
return &spaceXService{
|
||||
baseHttpService: &baseHttpService{
|
||||
server: server,
|
||||
},
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue