代码拉取完成,页面将自动刷新
package echo
import (
"net/http"
"strings"
"github.com/labstack/echo"
. "github.com/rpcxio/rpcx-gateway"
)
type Server struct {
addr string
e *echo.Echo
}
// New returns a server.
func New(addr string) *Server {
return &Server{
addr: addr,
}
}
// NewWithEcho returns a server with preconfigured echo.
func NewWithEcho(addr string, e *echo.Echo) *Server {
return &Server{
addr: addr,
e: e,
}
}
// RegisterHandler configures the handler to handle http rpcx invoke.
// It wraps ServiceHandler into httprouter.Handle.
func (s *Server) RegisterHandler(base string, handler ServiceHandler) {
e := s.e
if e == nil {
e = echo.New()
}
h := wrapServiceHandler(handler)
e.POST(base, h)
e.GET(base, h)
e.PUT(base, h)
s.e = e
}
func wrapServiceHandler(handler ServiceHandler) echo.HandlerFunc {
return func(ctx echo.Context) error {
r := ctx.Request()
w := ctx.Response()
if r.Header.Get(XServicePath) == "" {
servicePath := ctx.Param("servicePath")
if strings.HasPrefix(servicePath, "/") {
servicePath = servicePath[1:]
}
r.Header.Set(XServicePath, servicePath)
}
servicePath := r.Header.Get(XServicePath)
messageID := r.Header.Get(XMessageID)
wh := w.Header()
if messageID != "" {
wh.Set(XMessageID, messageID)
}
meta, payload, err := handler(r, servicePath)
for k, v := range meta {
wh.Set(k, v)
}
if err == nil {
ctx.Blob(http.StatusOK, "application/octet-stream", payload)
return nil
}
rh := r.Header
for k, v := range rh {
if strings.HasPrefix(k, "X-RPCX-") && len(v) > 0 {
wh.Set(k, v[0])
}
}
wh.Set(XMessageStatusType, "Error")
wh.Set(XErrorMessage, err.Error())
ctx.String(http.StatusOK, err.Error())
return nil
}
}
func (s *Server) Serve() error {
return s.e.Start(s.addr)
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。