-
Notifications
You must be signed in to change notification settings - Fork 0
/
hrq_adapter.go
52 lines (45 loc) · 1.15 KB
/
hrq_adapter.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
package hrq
import (
"context"
beecontext "github.com/beego/beego/v2/server/web/context"
"github.com/gin-gonic/gin"
"github.com/labstack/echo/v4"
"net/http"
)
type adapter interface {
Abort() interface{}
Next() *httpError
}
type defaultAdapter struct {
ctx context.Context
hrq *hrq
reqRsp *reqrsq
}
func (d *defaultAdapter) Abort() interface{} {
return nil
}
func (d *defaultAdapter) Next() *httpError {
hander, _, _ := d.hrq.router.Lookup(d.reqRsp.method, d.reqRsp.path)
if hander != nil {
hander(d.reqRsp.rsp, d.reqRsp.req, nil)
d.reqRsp.done <- true
} else {
d.reqRsp.rsp.WriteHeader(http.StatusNotFound)
d.reqRsp.rsp.Write([]byte("not found"))
d.reqRsp.done <- true
}
return nil
}
var _ adapter = (*defaultAdapter)(nil)
func (h *hrq) getAdapter(ctx interface{}, reqRsp *reqrsq, hrq *hrq) adapter {
switch ctx.(type) {
case *gin.Context:
return &ginAdapter{ctx.(*gin.Context), hrq, reqRsp}
case *beecontext.Context:
return &beegoAdapter{ctx.(*beecontext.Context), hrq, reqRsp}
case echo.Context:
return &echoAdapter{ctx.(echo.Context), hrq, reqRsp}
default:
return &defaultAdapter{ctx.(context.Context), hrq, reqRsp}
}
}