Removed ws. Added /v1/head endpoint for stats

cmd/api/handlers/head.go

@@ -0,0 +1,76 @@
+package handlers
+
+import (
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+// GetHead godoc
+// @Summary Show indexer head
+// @Description Get indexer head for each network
+// @Tags head
+// @ID get-indexer-head
+// @Accept json
+// @Produce json
+// @Success 200 {array} HeadResponse
+// @Failure 500 {object} Error
+// @Router /v1/head [get]
+func (ctx *Context) GetHead(c *gin.Context) {
+	item, err := ctx.Cache.Fetch("head", time.Second*30, ctx.getHead)
+	if ctx.handleError(c, err, 0) {
+		return
+	}
+
+	c.JSON(http.StatusOK, item.Value())
+}
+
+func (ctx *Context) getHead() (interface{}, error) {
+	blocks, err := ctx.Blocks.LastByNetworks()
+	if err != nil {
+		return nil, err
+	}
+
+	var network string
+	if len(blocks) == 1 {
+		network = blocks[0].Network
+	}
+	callCounts, err := ctx.Storage.GetCallsCountByNetwork(network)
+	if err != nil {
+		return nil, err
+	}
+	contractStats, err := ctx.Storage.GetContractStatsByNetwork(network)
+	if err != nil {
+		return nil, err
+	}
+	faCount, err := ctx.Storage.GetFACountByNetwork(network)
+	if err != nil {
+		return nil, err
+	}
+	body := make([]HeadResponse, len(blocks))
+	for i := range blocks {
+		body[i] = HeadResponse{
+			Network:   blocks[i].Network,
+			Level:     blocks[i].Level,
+			Timestamp: blocks[i].Timestamp,
+			Protocol:  blocks[i].Protocol,
+		}
+		calls, ok := callCounts[blocks[i].Network]
+		if ok {
+			body[i].ContractCalls = calls
+		}
+		fa, ok := faCount[blocks[i].Network]
+		if ok {
+			body[i].FACount = fa
+		}
+		stats, ok := contractStats[blocks[i].Network]
+		if ok {
+			body[i].Total = stats.Total
+			body[i].TotalBalance = stats.Balance
+			body[i].UniqueContracts = stats.SameCount
+		}
+	}
+
+	return body, nil
+}

cmd/api/handlers/responses.go

@@ -746,3 +746,17 @@ type TZIPResponse struct {
 	tzip.TZIP16
 	tzip.TZIP20
 }
+
+// HeadResponse -
+type HeadResponse struct {
+	Network         string    `json:"network"`
+	Level           int64     `json:"level"`
+	Timestamp       time.Time `json:"time"`
+	Protocol        string    `json:"protocol"`
+	Total           int64     `json:"total"`
+	ContractCalls   int64     `json:"contract_calls"`
+	UniqueContracts int64     `json:"unique_contracts"`
+	TotalBalance    int64     `json:"total_balance"`
+	TotalWithdrawn  int64     `json:"total_withdrawn"`
+	FACount         int64     `json:"fa_count"`
+}

cmd/api/main.go

@@ -9,7 +9,6 @@ import (
 	"github.com/baking-bad/bcdhub/cmd/api/handlers"
 	"github.com/baking-bad/bcdhub/cmd/api/seed"
 	"github.com/baking-bad/bcdhub/cmd/api/validations"
-	"github.com/baking-bad/bcdhub/cmd/api/ws"
 	"github.com/baking-bad/bcdhub/internal/config"
 	"github.com/baking-bad/bcdhub/internal/helpers"
 	"github.com/baking-bad/bcdhub/internal/logger"
@@ -21,7 +20,6 @@ import (
 
 type app struct {
 	Router  *gin.Engine
-	Hub     *ws.Hub
 	Context *handlers.Context
 }
 
@@ -53,7 +51,6 @@ func newApp() *app {
 	}
 
 	api := &app{
-		Hub:     ws.DefaultHub(ctx),
 		Context: ctx,
 	}
 
@@ -92,8 +89,8 @@ func (api *app) makeRouter() {
 	v1 := r.Group("v1")
 	{
 		v1.GET("swagger.json", api.Context.GetSwaggerDoc)
-		v1.GET("ws", func(c *gin.Context) { ws.Handler(c, api.Hub) })
 
+		v1.GET("head", api.Context.GetHead)
 		v1.GET("opg/:hash", api.Context.GetOperation)
 		v1.GET("operation/:id/error_location", api.Context.GetOperationErrorLocation)
 		v1.GET("pick_random", api.Context.GetRandomContract)
@@ -258,11 +255,9 @@ func (api *app) makeRouter() {
 
 func (api *app) Close() {
 	api.Context.Close()
-	api.Hub.Stop()
 }
 
 func (api *app) Run() {
-	api.Hub.Run()
 	if err := api.Router.Run(api.Context.Config.API.Bind); err != nil {
 		logger.Error(err)
 		helpers.CatchErrorSentry(err)