antelope-api-healthcheck/src/api/eosio_contract.go

package api

import (
    "fmt"
    "time"
    "github.com/eosswedenorg-go/haproxy/agentcheck"
    contract_api "github.com/eosswedenorg-go/eos-contract-api-client"
)

type EosioContract struct {
    client contract_api.Client
    block_time float64
    ts time.Time
}

func NewEosioContract(url string, block_time float64) EosioContract {
    return EosioContract{
        client: contract_api.Client{
            Url: url,
        },
        block_time: block_time,
    }
}

func (e EosioContract) LogInfo() LogParams {
    return LogParams{
        "type", "eosio-contract",
        "url", e.client.Url,
        "block_time", e.block_time,
    }
}

func (e *EosioContract) SetTime(t time.Time) {
    e.ts = t
}

func (e EosioContract) GetTime() time.Time {

    if e.ts.IsZero() {
        return time.Now().In(time.UTC)
    }
    return e.ts
}

func (e EosioContract) Call() (agentcheck.Response, string) {

    h, err := e.client.GetHealth()
    if err != nil {
        resp := agentcheck.NewStatusMessageResponse(agentcheck.Failed, "")
        return resp, err.Error()
    }

    // Check HTTP Status Code
    if h.HTTPStatusCode > 299 {
        resp := agentcheck.NewStatusMessageResponse(agentcheck.Down, "")

        msg := "Taking offline because %v was received from backend"
        return resp, fmt.Sprintf(msg, h.HTTPStatusCode)
    }

    // Check postgres
    if h.Data.Postgres.Status != "OK" {

        resp := agentcheck.NewStatusMessageResponse(agentcheck.Down,
            fmt.Sprintf("Postgres: %s", h.Data.Postgres.Status))

        msg := "Taking offline because Postgres reported '%s'"
        return resp, fmt.Sprintf(msg, h.Data.Postgres.Status)
    }

    // Check redis
    if h.Data.Redis.Status != "OK" {
        resp := agentcheck.NewStatusMessageResponse(agentcheck.Down, "")

        msg := "Taking offline because Redis reported '%s'"
        return resp, fmt.Sprintf(msg, h.Data.Redis.Status)
    }

    // Validate head block.
    now  := e.GetTime()
    diff := now.Sub(h.Data.Chain.HeadTime).Seconds()

    if diff > e.block_time {
        resp := agentcheck.NewStatusMessageResponse(agentcheck.Down, "")

        msg := "Taking offline because head block is lagging %.0f seconds"
        return resp, fmt.Sprintf(msg, diff)
    } else if diff < -e.block_time {

        resp := agentcheck.NewStatusMessageResponse(agentcheck.Down, "")

        msg := "Taking offline because head block is %.0f seconds into the future"
        return resp, fmt.Sprintf(msg, diff)
    }

    return agentcheck.NewStatusResponse(agentcheck.Up), "OK"
}