celogeek/photos-api
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

set restriction, fix print

Browse Source
This commit is contained in:
celogeek 2022-02-05 11:23:56 +01:00
parent 80ccd7101d
commit 433a19a2db
Signed by: celogeek
GPG Key ID: E6B7BDCFC446233A
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
internal/photos/api/main.go
View File

@ -2,7 +2,6 @@ package api
import ( import (
"context" "context"
"errors"
"log" "log"
"math/rand" "math/rand"
"net/http" "net/http"
@ -49,14 +48,15 @@ func New(logger *log.Logger, config *ServiceConfig) *Service {
func (s *Service) SetupRoutes() { func (s *Service) SetupRoutes() {
s.Gin.Use(gin.Logger()) s.Gin.Use(gin.Logger())
s.Gin.Use(s.Recovery) s.Gin.Use(s.Recovery)
s.Gin.Use(s.RequireBody)
ac := s.Gin.Group("/account") ac := s.Gin.Group("/account")
ac.POST("/signup", s.Signup) ac.POST("/signup", s.Signup)
ac.POST("/login", s.Login) ac.POST("/login", s.Login)
ac.POST("/logout", s.Logout) ac.GET("/logout", s.RequireSession, s.Logout)
s.Gin.NoRoute(func(c *gin.Context) { s.Gin.NoRoute(func(c *gin.Context) {
s.Error(c, http.StatusNotFound, errors.New("this route doesn't exists")) s.Error(c, http.StatusNotFound, ErrReqNotFound)
}) })
} }
@ -84,7 +84,7 @@ func (s *Service) Run() error {
// a timeout of 5 seconds. // a timeout of 5 seconds.
signal.Notify(quit, os.Interrupt) signal.Notify(quit, os.Interrupt)
<-quit <-quit
s.Logger.Println("shutdown ...") s.Logger.Print("shutdown ...")
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel() defer cancel()
@ -96,11 +96,11 @@ func (s *Service) Run() error {
if err != nil { if err != nil {
return err return err
} }
s.Logger.Println("closing database ...") s.Logger.Print("closing database ...")
if err := db.Close(); err != nil { if err := db.Close(); err != nil {
return err return err
} }
s.Logger.Println("exiting") s.Logger.Print("exiting")
return runError return runError
} }