Nulo/kawipiko
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[server] Add statistics counters and make sure the HTTP status code was properly initialized.

Browse source
This commit is contained in:
Ciprian Dorin Craciun 2021-12-18 11:35:58 +02:00
parent fc30b8eb78
commit de4234e2d7
1 changed files with 66 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

69
sources/cmd/server/server.go
View file

@ -21,6 +21,7 @@ import "runtime/pprof"
import "strconv" import "strconv"
import "strings" import "strings"
import "sync" import "sync"
import "sync/atomic"
import "syscall" import "syscall"
import "time" import "time"
import "unsafe" import "unsafe"
@ -66,7 +67,7 @@ type server struct {
func (_server *server) Serve (_context *fasthttp.RequestCtx) () { func (_server *server) ServeUnwrapped (_context *fasthttp.RequestCtx) () {
if _server.dummy { if _server.dummy {
@ -450,8 +451,56 @@ func (_server *server) ServeDummy (_context *fasthttp.RequestCtx) () {
func (_server *server) ServeWrapped (_context *fasthttp.RequestCtx) () {
_context.Response.SetStatusCode (-1)
_server.ServeUnwrapped (_context)
_status := _context.Response.StatusCode ()
_invalid := false
switch {
case _status < 100 :
_invalid = true
case _status < 200 :
atomic.AddUint64 (&_statsRequests1xx, 1)
case _status < 300 :
atomic.AddUint64 (&_statsRequests2xx, 1)
case _status < 400 :
atomic.AddUint64 (&_statsRequests3xx, 1)
case _status < 500 :
atomic.AddUint64 (&_statsRequests4xx, 1)
case _status < 600 :
atomic.AddUint64 (&_statsRequests5xx, 1)
default :
_invalid = true
}
if _invalid {
if !_server.quiet {
log.Printf ("[ee] [3db6b217] [http-x..] invalid status code `%d`!\n", _status)
}
_context.Response.Reset ()
_context.Response.SetStatusCode (http.StatusInternalServerError)
atomic.AddUint64 (&_statsRequests5xx, 1)
}
}
func (_server *server) ServeFast (_context *fasthttp.RequestCtx) () {
atomic.AddUint64 (&_statsRequestsFast, 1)
atomic.AddUint64 (&_statsRequestsTotal, 1)
_server.ServeWrapped (_context)
}
func (_server *server) ServeHTTP (_response http.ResponseWriter, _request *http.Request) () { func (_server *server) ServeHTTP (_response http.ResponseWriter, _request *http.Request) () {
atomic.AddUint64 (&_statsRequestsSlow, 1)
atomic.AddUint64 (&_statsRequestsTotal, 1)
_requestProtoUnsupported := false _requestProtoUnsupported := false
switch _request.ProtoMajor { switch _request.ProtoMajor {
case 1 : case 1 :
@ -478,6 +527,7 @@ func (_server *server) ServeHTTP (_response http.ResponseWriter, _request *http.
if !_server.quiet { if !_server.quiet {
log.Printf ("[ww] [4c44e3c0] [go-http.] protocol HTTP/%d not supported for `%s`!", _request.ProtoMajor, _request.URL.Path) log.Printf ("[ww] [4c44e3c0] [go-http.] protocol HTTP/%d not supported for `%s`!", _request.ProtoMajor, _request.URL.Path)
} }
atomic.AddUint64 (&_statsRequests5xx, 1)
return return
} }
@ -493,6 +543,7 @@ func (_server *server) ServeHTTP (_response http.ResponseWriter, _request *http.
if _server.dummyDelay != 0 { if _server.dummyDelay != 0 {
time.Sleep (_server.dummyDelay) time.Sleep (_server.dummyDelay)
} }
atomic.AddUint64 (&_statsRequests2xx, 1)
return return
} }
@ -512,7 +563,7 @@ func (_server *server) ServeHTTP (_response http.ResponseWriter, _request *http.
_context.Response.Reset () _context.Response.Reset ()
_server.Serve (_context) _server.ServeWrapped (_context)
_responseHeaders := NewHttpResponseWriterHeadersBuffer (_context.Response.Header.StatusCode ()) _responseHeaders := NewHttpResponseWriterHeadersBuffer (_context.Response.Header.StatusCode ())
@ -1390,7 +1441,7 @@ func main_0 () (error) {
_httpPlain1Server := & fasthttp.Server { _httpPlain1Server := & fasthttp.Server {
Name : "kawipiko", Name : "kawipiko",
Handler : _server.Serve, Handler : _server.ServeFast,
GetOnly : true, GetOnly : true,
NoDefaultServerHeader : true, NoDefaultServerHeader : true,
@ -1972,6 +2023,18 @@ func (_listener *splitListener) Addr () (net.Addr) {
var _statsRequestsTotal uint64
var _statsRequestsFast uint64
var _statsRequestsSlow uint64
var _statsRequests1xx uint64
var _statsRequests2xx uint64
var _statsRequests3xx uint64
var _statsRequests4xx uint64
var _statsRequests5xx uint64
//go:embed usage.txt //go:embed usage.txt
var usageText string var usageText string