53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package server
|
|
|
|
import (
|
|
"errors"
|
|
"log"
|
|
"net"
|
|
)
|
|
|
|
type RouteHandler func(req *Request) *Response
|
|
|
|
type Router struct {
|
|
Routes map[string]RouteHandler
|
|
}
|
|
|
|
func NewRouter() *Router {
|
|
return &Router{
|
|
Routes: make(map[string]RouteHandler),
|
|
}
|
|
}
|
|
|
|
func (r *Router) AddRoute(pathname string, handler RouteHandler) {
|
|
if _, ok := r.Routes[pathname]; ok {
|
|
log.Printf("WARN: Rewriting existing route '%s'\n", pathname)
|
|
}
|
|
r.Routes[pathname] = handler
|
|
}
|
|
|
|
func (r *Router) Handle(conn net.Conn) error {
|
|
defer conn.Close()
|
|
|
|
log.Printf("Got new connection from %s\n", conn.RemoteAddr().String())
|
|
|
|
dupl := NewPacketDuplex(conn)
|
|
|
|
// TODO: Ability to keep connection alive
|
|
packet, err := dupl.ReadPacket()
|
|
if err != nil {
|
|
dupl.WritePacket(NewErrorResponse(err).ToPacket())
|
|
return err
|
|
}
|
|
|
|
if route, ok := packet.Headers["route"]; ok {
|
|
log.Printf("[ROUTER] New connection to '%s'", route)
|
|
if handler, ok := r.Routes[route]; ok {
|
|
req := NewRequest(conn, packet)
|
|
res := handler(req)
|
|
_, err := dupl.WritePacket(res.ToPacket())
|
|
return err
|
|
}
|
|
}
|
|
return errors.New("Not Found")
|
|
}
|