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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
package main
import (
"bufio"
"fmt"
"io"
"net"
"regexp"
)
type Router struct {
routes []Route
}
func (r *Router) handler(request *Request) func(*Request) *Response {
for _, route := range r.routes {
if route.method == request.Method() && route.path.MatchString(request.Target()) {
return route.handler
}
}
return func(request *Request) *Response {
return NewResponse(NotFound, nil, nil)
}
}
func (r *Router) write(writer io.Writer, response *Response, request *Request) (err error) {
if request.Header("Accept-Encoding") == "gzip" {
response.headers["Content-Encoding"] = "gzip"
}
_, err = writer.Write(response.StatusLine())
if err != nil {
fmt.Println("Error writing status:", err)
return
}
_, err = writer.Write(response.Headers())
if err != nil {
fmt.Println("Error writing headers:", err)
return
}
_, err = writer.Write(response.Body())
if err != nil {
fmt.Println("Error writing body:", err)
return
}
return
}
func (r *Router) Handle(conn net.Conn) error {
defer conn.Close()
request, err := NewRequest(bufio.NewReader(conn))
if err != nil {
return err
}
return r.write(conn, r.handler(request)(request), request)
}
type Route struct {
method string
path *regexp.Regexp
handler func(*Request) *Response
}
type RouterBuilder struct {
routes []Route
}
func (r *RouterBuilder) Add(method string, path *regexp.Regexp, handler func(*Request) *Response) {
r.routes = append(r.routes, Route{
method: method,
path: path,
handler: handler,
})
}
func (r *RouterBuilder) Build() *Router {
return &Router{
routes: r.routes,
}
}
func NewRouterBuilder() *RouterBuilder {
return &RouterBuilder{
routes: make([]Route, 0),
}
}
|
