From aaa4fe9adf945d0efe89513da432340091d20d7d Mon Sep 17 00:00:00 2001 From: xdrm-brackets Date: Thu, 2 May 2019 20:55:36 +0200 Subject: [PATCH] use http.Method* instead of strings --- service/auth/auth.go | 3 ++- service/shortener/shortener.go | 8 ++++---- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/service/auth/auth.go b/service/auth/auth.go index 6f7be83..a3d531b 100644 --- a/service/auth/auth.go +++ b/service/auth/auth.go @@ -5,6 +5,7 @@ import ( "database/sql" "encoding/hex" "log" + "net/http" "strconv" "strings" "time" @@ -39,7 +40,7 @@ func New(storage *sql.DB) (*Service, error) { // Wire to the aicra server func (svc *Service) Wire(server *aicra.Server) { log.Printf("[auth] service wired") - server.HandleFunc("POST", "/token", svc.generateToken) + server.HandleFunc(http.MethodPost, "/token", svc.generateToken) } // generateToken generates a token valid for 5 mintes diff --git a/service/shortener/shortener.go b/service/shortener/shortener.go index 926dee3..4796539 100644 --- a/service/shortener/shortener.go +++ b/service/shortener/shortener.go @@ -38,10 +38,10 @@ func New(storage *sql.DB, auth *auth.Service) (*Service, error) { // Wire to the aicra server func (svc *Service) Wire(server *aicra.Server) { log.Printf("[shortener] service wired") - server.HandleFunc("GET", "/", svc.redirect) - server.HandleFunc("POST", "/", svc.authService.CheckToken(svc.register)) - server.HandleFunc("PUT", "/", svc.authService.CheckToken(svc.update)) - server.HandleFunc("DELETE", "/", svc.authService.CheckToken(svc.delete)) + server.HandleFunc(http.MethodGet, "/", svc.redirect) + server.HandleFunc(http.MethodPost, "/", svc.authService.CheckToken(svc.register)) + server.HandleFunc(http.MethodPut, "/", svc.authService.CheckToken(svc.update)) + server.HandleFunc(http.MethodDelete, "/", svc.authService.CheckToken(svc.delete)) } // redirect from a tiny url to the long url