grassrootseconomics/cic-custodial
5
0
Fork 0
mirror of https://github.com/grassrootseconomics/cic-custodial.git synced 2024-11-24 06:56:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: global lock middleware func signature

Browse Source
This commit is contained in:
Mohamed Sohail 2023-03-28 06:55:51 +00:00
parent 5f5678e25b
commit 750fdd9c1f
Signed by: kamikazechaser
GPG Key ID: 7DD45520C01CD85D
1 changed files with 16 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
cmd/service/api.go
View File

@ -40,7 +40,7 @@ func initApiServer(custodialContainer *custodial.Custodial) *echo.Echo {
}) })
} }
apiRoute := server.Group("/api", systemGlobalLock) apiRoute := server.Group("/api", systemGlobalLock(custodialContainer))
apiRoute.POST("/account/create", api.HandleAccountCreate(custodialContainer)) apiRoute.POST("/account/create", api.HandleAccountCreate(custodialContainer))
apiRoute.GET("/account/status/:address", api.HandleNetworkAccountStatus(custodialContainer)) apiRoute.GET("/account/status/:address", api.HandleNetworkAccountStatus(custodialContainer))
@ -78,24 +78,22 @@ func customHTTPErrorHandler(err error, c echo.Context) {
}) })
} }
func systemGlobalLock(next echo.HandlerFunc) echo.HandlerFunc { func systemGlobalLock(cu *custodial.Custodial) echo.MiddlewareFunc {
return func(c echo.Context) error { return func(next echo.HandlerFunc) echo.HandlerFunc {
var ( return func(c echo.Context) error {
cu = c.Get("cu").(*custodial.Custodial) locked, err := cu.RedisClient.Get(c.Request().Context(), systemGlobalLockKey).Bool()
) if err != nil {
return err
}
locked, err := cu.RedisClient.Get(c.Request().Context(), systemGlobalLockKey).Bool() if locked {
if err != nil { return c.JSON(http.StatusServiceUnavailable, api.ErrResp{
return err Ok: false,
Message: "System manually locked.",
})
}
return next(c)
} }
if locked {
return c.JSON(http.StatusServiceUnavailable, api.ErrResp{
Ok: false,
Message: "System manually locked.",
})
}
return next(c)
} }
} }