Subversion Repositories SmartDukaan

package com.smartdukaan.cron.monitored;

import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;

import io.micrometer.core.instrument.Gauge;
import io.micrometer.core.instrument.MeterRegistry;
import org.springframework.stereotype.Component;

/**
 * Publishes the two prepaid balance signals -- SMS credits and the ThinkWalnut
 * recharge wallet -- on /actuator/prometheus, which alloy scrapes off
 * 127.0.0.1:8081 on this host.
 *
 * Values are pushed in by {@link BalanceMonitorTasks}; -1 means "not read yet"
 * so a scrape before the first run is distinguishable from a genuine zero
 * balance.
 */
@Component
public class BalanceGauges {

    private static final int NOT_READ_YET = -1;

    private final AtomicInteger smsBalance = new AtomicInteger(NOT_READ_YET);
    private final AtomicReference<Double> rechargeWalletBalance =
            new AtomicReference<>((double) NOT_READ_YET);

    public BalanceGauges(MeterRegistry meterRegistry) {
        Gauge.builder("sms_balance_remaining", smsBalance, AtomicInteger::get)
                .description("SMS credits remaining with the messaging gateway; -1 = not read yet")
                .register(meterRegistry);

        Gauge.builder("recharge_wallet_balance_rupees", rechargeWalletBalance, AtomicReference::get)
                .description("ThinkWalnut recharge wallet balance in rupees; -1 = not read yet")
                .register(meterRegistry);
    }

    public void setSmsBalance(int count) {
        smsBalance.set(count);
    }

    public void setRechargeWalletBalance(double rupees) {
        rechargeWalletBalance.set(rupees);
    }
}