Compare commits

..

3 commits

View file

@ -36,6 +36,7 @@ func main() {
return return
} }
stmt = sel_stmt stmt = sel_stmt
defer sel_stmt.Close()
fasthttp.ListenAndServe("127.0.0.1:4001", handler) fasthttp.ListenAndServe("127.0.0.1:4001", handler)
} }
@ -59,7 +60,7 @@ func handler(ctx *fasthttp.RequestCtx) {
row := stmt.QueryRow( row := stmt.QueryRow(
ctx.Method(), ctx.Method(),
fmt.Sprintf( fmt.Sprintf(
"%s://%s:%d%s", "%s://%s:%s%s",
scheme, scheme,
host, host,
port, port,
@ -113,15 +114,17 @@ func handler(ctx *fasthttp.RequestCtx) {
func parseHost(host []byte, https bool) ([]byte, []byte) { func parseHost(host []byte, https bool) ([]byte, []byte) {
idx := bytes.LastIndex(host, []byte(":")) idx := bytes.LastIndex(host, []byte(":"))
var port []byte var resHost, port []byte
if idx != -1 { if idx != -1 {
resHost = host[:idx]
port = host[idx+1:] port = host[idx+1:]
} else { } else {
resHost = host
if https { if https {
port = []byte("443") port = []byte("443")
} else { } else {
port = []byte("80") port = []byte("80")
} }
} }
return host[:idx], port return resHost, port
} }