antelope-api-healthcheck/internal/api/eosio_v2.go

package api

import (
	"fmt"

	"github.com/eosswedenorg-go/eosapi"
	"github.com/eosswedenorg-go/haproxy/agentcheck"
	"github.com/eosswedenorg/eosio-api-healthcheck/internal/utils"
)

type EosioV2 struct {
	client eosapi.Client
	offset int64
}

func EosioV2Factory(args ApiArguments) ApiInterface {
	return NewEosioV2(args.Url, args.Host, int64(args.NumBlocks))
}

func NewEosioV2(url string, host string, offset int64) EosioV2 {
	api := EosioV2{
		client: *eosapi.New(url),
		offset: offset,
	}

	api.client.Host = host

	return api
}

func (e EosioV2) LogInfo() LogParams {
	p := LogParams{
		"type", "eosio-v2",
		"url", e.client.Url,
	}

	if len(e.client.Host) > 0 {
		p.Add("host", e.client.Host)
	}

	p.Add("offset", e.offset)

	return p
}

func (e EosioV2) Call() (agentcheck.Response, string) {
	health, err := e.client.GetHealth()
	if err != nil {
		resp := agentcheck.NewStatusMessageResponse(agentcheck.Failed, "")
		return resp, err.Error()
	}

	// Fetch elasticsearch and nodeos block numbers from json.
	var es_block int64 = 0
	var node_block int64 = 0

	for _, v := range health.Health {
		if v.Name == "Elasticsearch" {
			es_block = utils.JsonGetInt64(v.Data["last_indexed_block"])
		} else if v.Name == "NodeosRPC" {
			node_block = utils.JsonGetInt64(v.Data["head_block_num"])
		}
	}

	// Error out if ether or both are zero.
	if es_block == 0 || node_block == 0 {
		msg := fmt.Sprintf("Failed to get Elasticsearch and/or nodeos "+
			"block numbers (es: %d, eos: %d)", es_block, node_block)

		resp := agentcheck.NewStatusMessageResponse(agentcheck.Failed, "")
		return resp, msg
	}

	// Check if ES is behind or in the future.
	diff := node_block - es_block
	if diff > e.offset {
		resp := agentcheck.NewStatusMessageResponse(agentcheck.Down, "")
		return resp, fmt.Sprintf("Taking offline because Elastic is %d blocks behind", diff)
	} else if diff < -e.offset {
		resp := agentcheck.NewStatusMessageResponse(agentcheck.Down, "")
		return resp, fmt.Sprintf("Taking offline because Elastic is %d blocks into the future", -1*diff)
	}
	return agentcheck.NewStatusResponse(agentcheck.Up), "OK"
}