Stowage/rest-server
2
0
Fork 0
mirror of https://github.com/restic/rest-server.git synced 2025-10-19 07:33:21 +00:00
Code Issues Projects Releases Packages Wiki Activity

created a router package

Browse source
This commit is contained in:
Chapuis Bertil 2015-08-11 14:56:47 +02:00
parent 90266e7006
commit 1ab2939041
2 changed files with 81 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

54
router/router.go Normal file
View file

@ -0,0 +1,54 @@
package router
import (
"net/http"
"strings"
)
type Route struct {
method string
pattern string
handler http.Handler
}
type Router struct {
filters []http.Handler
routes []Route
}
func NewRouter() Router {
filters := []http.Handler{}
routes := []Route{}
return Router{filters, routes}
}
func (router *Router) Filter(handler http.Handler) {
router.filters = append(router.filters, handler)
}
func (router *Router) FilterFunc(handlerFunc http.HandlerFunc) {
router.Filter(handlerFunc)
}
func (router *Router) Handle(method string, pattern string, handler http.Handler) {
router.routes = append(router.routes, Route{method, pattern, handler})
}
func (router *Router) HandleFunc(method string, pattern string, handlerFunc http.HandlerFunc) {
router.Handle(method, pattern, handlerFunc)
}
func (router Router) ServeHTTP(w http.ResponseWriter, r *http.Request) {
for i := 0; i < len(router.filters); i++ {
filter := router.filters[i]
filter.ServeHTTP(w, r)
}
for i := 0; i < len(router.routes); i++ {
route := router.routes[i]
if route.method == r.Method && strings.HasPrefix(r.URL.String(), route.pattern) {
route.handler.ServeHTTP(w, r)
return
}
}
http.NotFound(w, r)
}

102
server.go
View file

@ -3,98 +3,50 @@ package main
import (
"io/ioutil"
"net/http"
"strings"
"github.com/bchapuis/restic-server/config"
"github.com/bchapuis/restic-server/handlers"
"github.com/bchapuis/restic-server/router"
)
type Route struct {
method string
pattern string
handler http.Handler
}
type Router struct {
filters []http.Handler
routes []Route
}
func NewRouter() Router {
filters := []http.Handler{}
routes := []Route{}
return Router{filters, routes}
}
func (router *Router) Filter(handler http.Handler) {
router.filters = append(router.filters, handler)
}
func (router *Router) FilterFunc(handlerFunc http.HandlerFunc) {
router.Filter(handlerFunc)
}
func (router *Router) Handle(method string, pattern string, handler http.Handler) {
router.routes = append(router.routes, Route{method, pattern, handler})
}
func (router *Router) HandleFunc(method string, pattern string, handlerFunc http.HandlerFunc) {
router.Handle(method, pattern, handlerFunc)
}
func (router Router) ServeHTTP(w http.ResponseWriter, r *http.Request) {
for i := 0; i < len(router.filters); i++ {
filter := router.filters[i]
filter.ServeHTTP(w, r)
}
for i := 0; i < len(router.routes); i++ {
route := router.routes[i]
if route.method == r.Method && strings.HasPrefix(r.URL.String(), route.pattern) {
route.handler.ServeHTTP(w, r)
return
}
}
http.NotFound(w, r)
}
func main() {
path, _ := ioutil.TempDir("", "restic-repository-")
config.Init(path)
router := NewRouter()
r := router.NewRouter()
router.FilterFunc(handlers.RequestLogger)
r.FilterFunc(handlers.RequestLogger)
router.HandleFunc("HEAD", "/config", handlers.HeadConfig)
router.HandleFunc("GET", "/config", handlers.GetConfig)
router.HandleFunc("POST", "/config", handlers.PostConfig)
r.HandleFunc("HEAD", "/config", handlers.HeadConfig)
r.HandleFunc("GET", "/config", handlers.GetConfig)
r.HandleFunc("POST", "/config", handlers.PostConfig)
router.HandleFunc("HEAD", "/data", handlers.HeadData)
router.HandleFunc("GET", "/data", handlers.GetData)
router.HandleFunc("POST", "/data", handlers.PostData)
router.HandleFunc("DELETE", "/data", handlers.DeleteData)
r.HandleFunc("HEAD", "/data", handlers.HeadData)
r.HandleFunc("GET", "/data", handlers.GetData)
r.HandleFunc("POST", "/data", handlers.PostData)
r.HandleFunc("DELETE", "/data", handlers.DeleteData)
router.HandleFunc("HEAD", "/snapshot", handlers.HeadSnapshot)
router.HandleFunc("GET", "/snapshot", handlers.GetSnapshot)
router.HandleFunc("POST", "/snapshot", handlers.PostSnapshot)
router.HandleFunc("DELETE", "/snapshot", handlers.DeleteSnapshot)
r.HandleFunc("HEAD", "/snapshot", handlers.HeadSnapshot)
r.HandleFunc("GET", "/snapshot", handlers.GetSnapshot)
r.HandleFunc("POST", "/snapshot", handlers.PostSnapshot)
r.HandleFunc("DELETE", "/snapshot", handlers.DeleteSnapshot)
router.HandleFunc("HEAD", "/index", handlers.HeadIndex)
router.HandleFunc("GET", "/index", handlers.GetIndex)
router.HandleFunc("POST", "/index", handlers.PostIndex)
router.HandleFunc("DELETE", "/index", handlers.DeleteIndex)
r.HandleFunc("HEAD", "/index", handlers.HeadIndex)
r.HandleFunc("GET", "/index", handlers.GetIndex)
r.HandleFunc("POST", "/index", handlers.PostIndex)
r.HandleFunc("DELETE", "/index", handlers.DeleteIndex)
router.HandleFunc("HEAD", "/lock", handlers.HeadLock)
router.HandleFunc("GET", "/lock", handlers.GetLock)
router.HandleFunc("POST", "/lock", handlers.PostLock)
router.HandleFunc("DELETE", "/lock", handlers.DeleteLock)
r.HandleFunc("HEAD", "/lock", handlers.HeadLock)
r.HandleFunc("GET", "/lock", handlers.GetLock)
r.HandleFunc("POST", "/lock", handlers.PostLock)
r.HandleFunc("DELETE", "/lock", handlers.DeleteLock)
router.HandleFunc("HEAD", "/key", handlers.HeadKey)
router.HandleFunc("GET", "/key", handlers.GetKey)
router.HandleFunc("POST", "/key", handlers.PostKey)
router.HandleFunc("DELETE", "/key", handlers.DeleteKey)
r.HandleFunc("HEAD", "/key", handlers.HeadKey)
r.HandleFunc("GET", "/key", handlers.GetKey)
r.HandleFunc("POST", "/key", handlers.PostKey)
r.HandleFunc("DELETE", "/key", handlers.DeleteKey)
http.ListenAndServe(":8000", router)
http.ListenAndServe(":8000", r)
}