realloc/frostfs-rest-gw
1
0
Fork 0
forked from TrueCloudLab/frostfs-rest-gw
Code Issues Pull requests Projects Releases Packages Wiki Activity

[#33] Add route to get NeoFS balance

Browse source 
Signed-off-by: Denis Kirillov <denis@nspcc.ru>
This commit is contained in:
Denis Kirillov 2022-07-28 19:57:40 +03:00 • committed by Alex Vanin
parent aaf7433c7b
commit 6e01a0ead7
9 changed files with 530 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

37
handlers/balance.go Normal file
View file

@ -0,0 +1,37 @@
package handlers
import (
"strconv"
"github.com/go-openapi/runtime/middleware"
"github.com/nspcc-dev/neofs-rest-gw/gen/models"
"github.com/nspcc-dev/neofs-rest-gw/gen/restapi/operations"
"github.com/nspcc-dev/neofs-rest-gw/internal/util"
"github.com/nspcc-dev/neofs-sdk-go/pool"
"github.com/nspcc-dev/neofs-sdk-go/user"
)
// Balance handler that get balance from NeoFS.
func (a *API) Balance(params operations.GetBalanceParams) middleware.Responder {
var ownerID user.ID
if err := ownerID.DecodeString(params.Address); err != nil {
resp := a.logAndGetErrorResponse("parse address", err)
return operations.NewGetBalanceBadRequest().WithPayload(resp)
}
var prm pool.PrmBalanceGet
prm.SetAccount(ownerID)
neofsBalance, err := a.pool.Balance(params.HTTPRequest.Context(), prm)
if err != nil {
resp := a.logAndGetErrorResponse("get balance", err)
return operations.NewGetBalanceBadRequest().WithPayload(resp)
}
var resp models.Balance
resp.Address = util.NewString(params.Address)
resp.Value = util.NewString(strconv.FormatInt(neofsBalance.Value(), 10))
resp.Precision = util.NewInteger(int64(neofsBalance.Precision()))
return operations.NewGetBalanceOK().WithPayload(&resp)
}