/*

  Copyright (c) 2017 Red Hat, Inc. <http://www.redhat.com>

  This file is part of GlusterFS.

  This file is licensed to you under your choice of the GNU Lesser

  General Public License, version 3 or any later version (LGPLv3 or

  later), or the GNU General Public License, version 2 (GPLv2), in all

  cases as published by the Free Software Foundation.

*/

#include "glusterfs/monitoring.h"

#include "glusterfs/xlator.h"

#include "glusterfs/syscall.h"

#include <stdlib.h>

static void

dump_mem_acct_details(xlator_t *xl, int fd)

{

    struct mem_acct_rec *mem_rec;

    int i = 0;

    if (!xl || !xl->mem_acct || (xl->ctx->active != xl->graph))

        return;

    dprintf(fd, "# %s.%s.total.num_types %d\n", xl->type, xl->name,

            xl->mem_acct->num_types);

    dprintf(fd,

            "# type, in-use-size, in-use-units, max-size, "

            "max-units, total-allocs\n");

    for (i = 0; i < xl->mem_acct->num_types; i++) {

        mem_rec = &xl->mem_acct->rec[i];

        if (mem_rec->num_allocs == 0)

            continue;

        dprintf(fd, "# %s, %" PRIu64 ", %u, %" PRIu64 ", %u, %" PRIu64 "\n",

                mem_rec->typestr, mem_rec->size, mem_rec->num_allocs,

                mem_rec->max_size, mem_rec->max_num_allocs,

                mem_rec->total_allocs);

    }

}

static void

dump_global_memory_accounting(int fd)

{

#if MEMORY_ACCOUNTING_STATS

    int i = 0;

    uint64_t count = 0;

    uint64_t tcalloc = GF_ATOMIC_GET(gf_memory_stat_counts.total_calloc);

    uint64_t tmalloc = GF_ATOMIC_GET(gf_memory_stat_counts.total_malloc);

    uint64_t tfree = GF_ATOMIC_GET(gf_memory_stat_counts.total_free);

    dprintf(fd, "memory.total.calloc %lu\n", tcalloc);

    dprintf(fd, "memory.total.malloc %lu\n", tmalloc);

    dprintf(fd, "memory.total.realloc %lu\n",

            GF_ATOMIC_GET(gf_memory_stat_counts.total_realloc));

    dprintf(fd, "memory.total.free %lu\n", tfree);

    dprintf(fd, "memory.total.in-use %lu\n", ((tcalloc + tmalloc) - tfree));

    for (i = 0; i < GF_BLK_MAX_VALUE; i++) {

        count = GF_ATOMIC_GET(gf_memory_stat_counts.blk_size[i]);

        dprintf(fd, "memory.total.blk_size.%s %lu\n",

                gf_mem_stats_blk[i].blk_size_str, count);

    }

    dprintf(fd, "#----\n");

#endif

    /* This is not a metric to be watched in admin guide,

       but keeping it here till we resolve all leak-issues

       would be great */

}

static void

dump_latency_and_count(xlator_t *xl, int fd)

{

    int32_t index = 0;

    uint64_t fop;

    uint64_t cbk;

    uint64_t count;

    if (xl->winds) {

        dprintf(fd, "%s.total.pending-winds.count %" PRIu64 "\n", xl->name,

                xl->winds);

    }

    /* Need 'fuse' data, and don't need all the old graph info */

    if ((xl != xl->ctx->master) && (xl->ctx->active != xl->graph))

        return;

    count = GF_ATOMIC_GET(xl->stats.total.count);

    dprintf(fd, "%s.total.fop-count %" PRIu64 "\n", xl->name, count);

    count = GF_ATOMIC_GET(xl->stats.interval.count);

    dprintf(fd, "%s.interval.fop-count %" PRIu64 "\n", xl->name, count);

    GF_ATOMIC_INIT(xl->stats.interval.count, 0);

    for (index = 0; index < GF_FOP_MAXVALUE; index++) {

        fop = GF_ATOMIC_GET(xl->stats.total.metrics[index].fop);

        if (fop) {

            dprintf(fd, "%s.total.%s.count %" PRIu64 "\n", xl->name,

                    gf_fop_list[index], fop);

        }

        fop = GF_ATOMIC_GET(xl->stats.interval.metrics[index].fop);

        if (fop) {

            dprintf(fd, "%s.interval.%s.count %" PRIu64 "\n", xl->name,

                    gf_fop_list[index], fop);

        }

        cbk = GF_ATOMIC_GET(xl->stats.interval.metrics[index].cbk);

        if (cbk) {

            dprintf(fd, "%s.interval.%s.fail_count %" PRIu64 "\n", xl->name,

                    gf_fop_list[index], cbk);

        }

        if (xl->stats.interval.latencies[index].count != 0.0) {

            dprintf(fd, "%s.interval.%s.latency %lf\n", xl->name,

                    gf_fop_list[index],

                    (xl->stats.interval.latencies[index].total /

                     xl->stats.interval.latencies[index].count));

            dprintf(fd, "%s.interval.%s.max %lf\n", xl->name,

                    gf_fop_list[index],

                    xl->stats.interval.latencies[index].max);

            dprintf(fd, "%s.interval.%s.min %lf\n", xl->name,

                    gf_fop_list[index],

                    xl->stats.interval.latencies[index].min);

        }

        GF_ATOMIC_INIT(xl->stats.interval.metrics[index].cbk, 0);

        GF_ATOMIC_INIT(xl->stats.interval.metrics[index].fop, 0);

    }

    memset(xl->stats.interval.latencies, 0,

           sizeof(xl->stats.interval.latencies));

}

static inline void

dump_call_stack_details(glusterfs_ctx_t *ctx, int fd)

{

    dprintf(fd, "total.stack.count %" PRIu64 "\n",

            GF_ATOMIC_GET(ctx->pool->total_count));

    dprintf(fd, "total.stack.in-flight %" PRIu64 "\n", ctx->pool->cnt);

}

static inline void

dump_dict_details(glusterfs_ctx_t *ctx, int fd)

{

    uint64_t total_dicts = 0;

    uint64_t total_pairs = 0;

    total_dicts = GF_ATOMIC_GET(ctx->stats.total_dicts_used);

    total_pairs = GF_ATOMIC_GET(ctx->stats.total_pairs_used);

    dprintf(fd, "total.dict.max-pairs-per %" PRIu64 "\n",

            GF_ATOMIC_GET(ctx->stats.max_dict_pairs));

    dprintf(fd, "total.dict.pairs-used %" PRIu64 "\n", total_pairs);

    dprintf(fd, "total.dict.used %" PRIu64 "\n", total_dicts);

    dprintf(fd, "total.dict.average-pairs %" PRIu64 "\n",

            (total_pairs / total_dicts));

}

static void

dump_inode_stats(glusterfs_ctx_t *ctx, int fd)

{

}

static void

dump_global_metrics(glusterfs_ctx_t *ctx, int fd)

{

    struct timeval tv;

    time_t nowtime;

    struct tm *nowtm;

    char tmbuf[64] = {

        0,

    };

    gettimeofday(&tv, NULL);

    nowtime = tv.tv_sec;

    nowtm = localtime(&nowtime);

    strftime(tmbuf, sizeof tmbuf, "%Y-%m-%d %H:%M:%S", nowtm);

    /* Let every file have information on which process dumped info */

    dprintf(fd, "## %s\n", ctx->cmdlinestr);

    dprintf(fd, "### %s\n", tmbuf);

    dprintf(fd, "### BrickName: %s\n", ctx->cmd_args.brick_name);

    dprintf(fd, "### MountName: %s\n", ctx->cmd_args.mount_point);

    dprintf(fd, "### VolumeName: %s\n", ctx->cmd_args.volume_name);

    /* Dump memory accounting */

    dump_global_memory_accounting(fd);

    dprintf(fd, "# -----\n");

    dump_call_stack_details(ctx, fd);

    dump_dict_details(ctx, fd);

    dprintf(fd, "# -----\n");

    dump_inode_stats(ctx, fd);

    dprintf(fd, "# -----\n");

}

static void

dump_xl_metrics(glusterfs_ctx_t *ctx, int fd)

{

    xlator_t *xl;

    xl = ctx->active->top;

    while (xl) {

        dump_latency_and_count(xl, fd);

        dump_mem_acct_details(xl, fd);

        if (xl->dump_metrics)

            xl->dump_metrics(xl, fd);

        xl = xl->next;

    }

    if (ctx->master) {

        xl = ctx->master;

        dump_latency_and_count(xl, fd);

        dump_mem_acct_details(xl, fd);

        if (xl->dump_metrics)

            xl->dump_metrics(xl, fd);

    }

    return;

}

char *

gf_monitor_metrics(glusterfs_ctx_t *ctx)

{

    int ret = -1;

    int fd = 0;

    char *filepath = NULL, *dumppath = NULL;

    dumppath = ctx->config.metrics_dumppath;

    if (dumppath == NULL) {

        dumppath = GLUSTER_METRICS_DIR;

    }

    ret = mkdir_p(dumppath, 0755, true);

    if (ret) {

        /* EEXIST is handled in mkdir_p() itself */

        gf_msg("monitoring", GF_LOG_ERROR, 0, LG_MSG_STRDUP_ERROR,

               "failed to create metrics dir %s (%s)", dumppath,

               strerror(errno));

        return NULL;

    }

    ret = gf_asprintf(&filepath, "%s/gmetrics.XXXXXX", dumppath);

    if (ret < 0) {

        return NULL;

    }

    /* coverity[secure_temp] mkstemp uses 0600 as the mode and is safe */

    fd = mkstemp(filepath);

    if (fd < 0) {

        gf_msg("monitoring", GF_LOG_ERROR, 0, LG_MSG_STRDUP_ERROR,

               "failed to open tmp file %s (%s)", filepath, strerror(errno));

        GF_FREE(filepath);

        return NULL;

    }

    dump_global_metrics(ctx, fd);

    dump_xl_metrics(ctx, fd);

    /* This below line is used just to capture any errors with dprintf() */

    ret = dprintf(fd, "\n# End of metrics\n");

    if (ret < 0) {

        gf_msg("monitoring", GF_LOG_WARNING, 0, LG_MSG_STRDUP_ERROR,

               "dprintf() failed: %s", strerror(errno));

    }

    ret = sys_fsync(fd);

    if (ret < 0) {

        gf_msg("monitoring", GF_LOG_WARNING, 0, LG_MSG_STRDUP_ERROR,

               "fsync() failed: %s", strerror(errno));

    }

    sys_close(fd);

    /* Figure this out, not happy with returning this string */

    return filepath;

}