55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"compress/gzip"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
var port = flag.String("port", "3000", "listen on port")
|
|
var body = flag.Bool("body", true, "print body")
|
|
var header = flag.Bool("header", true, "print headers")
|
|
|
|
func requestHandler(w http.ResponseWriter, r *http.Request) {
|
|
fmt.Printf("%s %s from %s", r.Method, r.RequestURI, r.RemoteAddr)
|
|
|
|
if *body {
|
|
var requestbody []byte
|
|
|
|
r.Body = http.MaxBytesReader(w, r.Body, 1048576)
|
|
if r.Header.Get("Content-Encoding") == "gzip" {
|
|
gzipreader, _ := gzip.NewReader(r.Body)
|
|
requestbody, _ = io.ReadAll(gzipreader)
|
|
} else {
|
|
requestbody, _ = io.ReadAll(r.Body)
|
|
}
|
|
fmt.Printf("\n--- BODY ---\n%s\n--- ENDBODY ---\n", string(requestbody))
|
|
}
|
|
|
|
if *header {
|
|
fmt.Println(r.Header)
|
|
}
|
|
|
|
fmt.Fprintf(w, "The current time is: %s\n", time.Now())
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
srv := http.Server{
|
|
Addr: ":" + *port,
|
|
WriteTimeout: 10 * time.Second,
|
|
ReadHeaderTimeout: 10 * time.Second,
|
|
Handler: http.HandlerFunc(requestHandler),
|
|
}
|
|
|
|
fmt.Printf("Starting server on port %s\n", *port)
|
|
//srv.HandleFunc("/", requestHandler)
|
|
err := srv.ListenAndServe()
|
|
if err != nil {
|
|
fmt.Println("ListenAndServe:", err)
|
|
}
|
|
}
|