// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.api.application.v4.model;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ObjectNode;

import java.util.HashMap;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;

public class SearchNodeMetrics {

    private static final ObjectMapper jsonMapper = new ObjectMapper();

    private static final Logger logger = Logger.getLogger(SearchNodeMetrics.class.getName());

    public static final String DOCUMENTS_ACTIVE_COUNT = "documentsActiveCount";
    public static final String DOCUMENTS_READY_COUNT = "documentsReadyCount";
    public static final String DOCUMENTS_TOTAL_COUNT = "documentsTotalCount";
    public static final String DOCUMENT_DISK_USAGE = "documentDiskUsage";
    public static final String RESOURCE_DISK_USAGE_AVERAGE = "resourceDiskUsageAverage";
    public static final String RESOURCE_MEMORY_USAGE_AVERAGE = "resourceMemoryUsageAverage";

    private final String clusterId;
    private final Map<String, Double> metrics;

    public SearchNodeMetrics(String clusterId) {
        this.clusterId = clusterId;
        metrics = new HashMap<>();
    }

    public String getClusterId() { return clusterId; }

    public double documentsActiveCount() { return metrics.get(DOCUMENTS_ACTIVE_COUNT); }

    public double documentsReadyCount() { return metrics.get(DOCUMENTS_READY_COUNT); }

    public double documentsTotalCount() { return metrics.get(DOCUMENTS_TOTAL_COUNT); }

    public double documentDiskUsage() { return metrics.get(DOCUMENT_DISK_USAGE); }

    public double resourceDiskUsageAverage() { return metrics.get(RESOURCE_DISK_USAGE_AVERAGE); }

    public double resourceMemoryUsageAverage() { return metrics.get(RESOURCE_MEMORY_USAGE_AVERAGE); }

    public SearchNodeMetrics addMetric(String name, double value) {
        metrics.put(name, value);
        return this;
    }

    public JsonNode toJson() {
        try {
            ObjectNode protonMetrics = jsonMapper.createObjectNode();
            protonMetrics.put("clusterId", clusterId);
            ObjectNode jsonMetrics = jsonMapper.createObjectNode();
            for (Map.Entry<String, Double> entry : metrics.entrySet()) {
                jsonMetrics.put(entry.getKey(), entry.getValue());
            }
            protonMetrics.set("metrics", jsonMetrics);
            return protonMetrics;
        } catch (Exception e) {
            logger.log(Level.SEVERE, "Unable to convert Proton Metrics to JSON Object: " + e.getMessage(), e);
        }
        return jsonMapper.createObjectNode();
    }
}