1
This commit is contained in:
parent
c767718aa4
commit
faa5f840af
89
context.go
89
context.go
@ -1,89 +0,0 @@
|
||||
package f5
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"q5"
|
||||
)
|
||||
|
||||
type Context struct {
|
||||
s *HttpServer
|
||||
w *http.ResponseWriter
|
||||
r *http.Request
|
||||
c *middleware
|
||||
l *q5.ListHead
|
||||
aborted bool
|
||||
suspended bool
|
||||
userData map[string]interface{}
|
||||
}
|
||||
|
||||
func (this *Context) do() {
|
||||
this.l.ForEach(
|
||||
func(data interface{}) bool {
|
||||
m := data.(*middleware)
|
||||
this.c = m
|
||||
m.handlerFunc(this)
|
||||
return !this.aborted && !this.suspended
|
||||
})
|
||||
}
|
||||
|
||||
func (this *Context) Abort() {
|
||||
this.aborted = true
|
||||
}
|
||||
|
||||
func (this *Context) Method() string {
|
||||
return this.r.Method
|
||||
}
|
||||
|
||||
func (this *Context) GetRemoteAddr() string {
|
||||
return q5.GetRequestRemoteAddr(this.r)
|
||||
}
|
||||
|
||||
func (this *Context) Request(name string) string {
|
||||
return q5.Request(this.r, name)
|
||||
}
|
||||
|
||||
func (this *Context) GetBody() string {
|
||||
return q5.GetPostBody(this.r)
|
||||
}
|
||||
|
||||
func (this *Context) Header(name string) string {
|
||||
if val, ok := this.r.Header[name]; ok {
|
||||
return val[0]
|
||||
} else {
|
||||
return ""
|
||||
}
|
||||
}
|
||||
|
||||
func (this *Context) Response(data string) {
|
||||
q5.Response(this.w, data)
|
||||
}
|
||||
|
||||
func (this *Context)ResponseErr(errCode int32, errMsg string) {
|
||||
q5.ResponseErr(this.w, errCode, errMsg)
|
||||
}
|
||||
|
||||
func (this *Context) ResponseOk() {
|
||||
q5.ResponseOk(this.w)
|
||||
}
|
||||
|
||||
func (this *Context) ResponseInt32Ok(key string, val int32) {
|
||||
q5.ResponseInt32Ok(this.w, key, val)
|
||||
}
|
||||
|
||||
func (this *Context) Set(key string, val interface{}) {
|
||||
if this.userData == nil {
|
||||
this.userData = make(map[string]interface{})
|
||||
}
|
||||
this.userData[key] = val
|
||||
}
|
||||
|
||||
func (this *Context) Get(key string) interface{} {
|
||||
if this.userData == nil {
|
||||
return nil
|
||||
}
|
||||
if val, ok := this.userData[key]; ok {
|
||||
return val
|
||||
} else {
|
||||
return nil
|
||||
}
|
||||
}
|
132
httpserver.go
132
httpserver.go
@ -1,132 +0,0 @@
|
||||
package f5
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
"fmt"
|
||||
"q5"
|
||||
)
|
||||
|
||||
type HttpServer struct {
|
||||
totalRequestTimes int64
|
||||
okTimes int64
|
||||
pageNotFoundTimes int64
|
||||
maxHandleTime int64
|
||||
startTimes int64
|
||||
|
||||
handlersMutex sync.RWMutex
|
||||
handlers map[string]HandlerFunc
|
||||
|
||||
restHandlersMutex sync.RWMutex
|
||||
restHandlers map[string]*q5.ListHead
|
||||
}
|
||||
|
||||
func (this *HttpServer) Init(serviceName string, logOutputTime int32) *HttpServer {
|
||||
this.handlers = make(map[string]HandlerFunc)
|
||||
this.restHandlers = make(map[string]*q5.ListHead)
|
||||
this.RegisterRestHandle("/webapp/index.php", this.dispatchRequest)
|
||||
this.RegisterHandle("Ops", "selfChecking",
|
||||
func (c *Context) {
|
||||
c.Response(`{"errcode":0, "errmsg":"", "healthy":1, "max_rundelay":10}`)
|
||||
})
|
||||
GetSysLog().Info("HttpServer.Init")
|
||||
if logOutputTime > 0 {
|
||||
GetTimer().SetInterval(
|
||||
logOutputTime,
|
||||
func (ev int32, params *q5.Args) {
|
||||
if ev == q5.TIMER_EXEC_EVENT {
|
||||
GetSysLog().Info("%s maxHandleTime:%d totalRequestTimes:%d okTimes:%d pageNotFoundTimes:%d",
|
||||
serviceName,
|
||||
this.maxHandleTime,
|
||||
this.totalRequestTimes,
|
||||
this.okTimes,
|
||||
this.pageNotFoundTimes)
|
||||
this.maxHandleTime = 0
|
||||
this.okTimes = 0
|
||||
this.pageNotFoundTimes = 0
|
||||
}
|
||||
})
|
||||
}
|
||||
return this
|
||||
}
|
||||
|
||||
func (this *HttpServer) UnInit() {
|
||||
GetSysLog().Info("HttpServer.UnInit")
|
||||
}
|
||||
|
||||
func (this *HttpServer) Start(listen_port int32) {
|
||||
atomic.AddInt64(&this.totalRequestTimes, 1)
|
||||
GetSysLog().Info("HttpServer.Start listen_port:%d", listen_port)
|
||||
go http.ListenAndServe(fmt.Sprintf("0.0.0.0:%d", listen_port), nil)
|
||||
}
|
||||
|
||||
func (this *HttpServer) dispatchRequest(c *Context) {
|
||||
atomic.AddInt64(&this.totalRequestTimes, 1)
|
||||
handleName := c.Request("c") + "$" + c.Request("a")
|
||||
handle := this.getHandle(handleName)
|
||||
if handle != nil {
|
||||
beginTick := q5.GetTickCount()
|
||||
defer func() {
|
||||
endTick := q5.GetTickCount()
|
||||
if oldVal := atomic.LoadInt64(&this.maxHandleTime); beginTick - endTick > oldVal {
|
||||
atomic.CompareAndSwapInt64(&this.maxHandleTime, oldVal, endTick - beginTick)
|
||||
}
|
||||
atomic.AddInt64(&this.okTimes, 1)
|
||||
if err := recover(); err != nil {
|
||||
GetSysLog().Error("handle name:%s error:%s", handleName, err)
|
||||
c.Response(`{"errcode":500, "errmsg":"server internal error"}`)
|
||||
}
|
||||
}()
|
||||
handle(c)
|
||||
} else {
|
||||
c.Response(`{"errcode":404, "errmsg":"interface not found"}`)
|
||||
atomic.AddInt64(&this.pageNotFoundTimes, 1)
|
||||
}
|
||||
}
|
||||
|
||||
func (this *HttpServer) getHandle(handleName string) HandlerFunc {
|
||||
this.handlersMutex.Lock()
|
||||
defer this.handlersMutex.Unlock()
|
||||
if handle, ok := this.handlers[handleName]; ok {
|
||||
return handle
|
||||
} else {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
func (this *HttpServer) RegisterHandle(c string, a string, handle HandlerFunc) {
|
||||
this.handlersMutex.Lock()
|
||||
defer this.handlersMutex.Unlock()
|
||||
handleName := c + "$" + a
|
||||
this.handlers[handleName] = handle
|
||||
}
|
||||
|
||||
func (this *HttpServer) RegisterRestHandle(pattern string,
|
||||
handle HandlerFunc) {
|
||||
if this.totalRequestTimes > 0 {
|
||||
panic("")
|
||||
}
|
||||
this.restHandlersMutex.Lock()
|
||||
defer this.restHandlersMutex.Unlock()
|
||||
|
||||
p := new(middleware)
|
||||
p.entry.Init(p)
|
||||
p.handlerFunc = handle
|
||||
if l, ok := this.restHandlers[pattern]; ok {
|
||||
l.AddTail(&p.entry)
|
||||
} else {
|
||||
l := q5.NewListHead()
|
||||
l.AddTail(&p.entry)
|
||||
this.restHandlers[pattern] = l
|
||||
http.HandleFunc(pattern,
|
||||
func (w http.ResponseWriter, r *http.Request) {
|
||||
c := new(Context)
|
||||
c.s = this
|
||||
c.w = &w
|
||||
c.r = r
|
||||
c.l = l
|
||||
c.do()
|
||||
})
|
||||
}
|
||||
}
|
7
types.go
7
types.go
@ -34,7 +34,6 @@ type StreamPagination struct {
|
||||
Remaining int32
|
||||
}
|
||||
|
||||
type HandlerFunc func(*Context)
|
||||
type GinHandlerFunc func(*gin.Context)
|
||||
|
||||
type QueryResultCb func(error, *DataSet)
|
||||
@ -42,9 +41,3 @@ type QueryOneCb func(error, *DataSet)
|
||||
type PageQueryCb func(error, *Pagination)
|
||||
type SteamPageQueryCb func(error, *StreamPagination)
|
||||
type ExecResultCb func(error, int64, int64)
|
||||
|
||||
type middleware struct {
|
||||
middlewareType int32
|
||||
handlerFunc HandlerFunc
|
||||
entry q5.ListHead
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user