/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2026. All rights reserved.
 * ubs-hcom is licensed under the Mulan PSL v2.
 * You can use this software according to the terms and conditions of the Mulan PSL v2.
 * You may obtain a copy of Mulan PSL v2 at:
 *      http://license.coscl.org.cn/MulanPSL2
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
 * See the Mulan PSL v2 for more details.
 */
#ifndef UBS_COMM_UBSOCKET_PROF_H
#define UBS_COMM_UBSOCKET_PROF_H

#include <cstdint>
#include <cstring>
#include <ctime>

#ifdef __cplusplus
extern "C" {
#endif

extern int ubsocket_prof_enabled;

enum ProfilingTPId : uint32_t
{
    CORE_CONNECT = 0,
    CORE_ACCEPT,
    CORE_WRITE,
    CORE_READ,
    // 当前为了快速分析CTP性能，brpc复用ubsocket打点和cli查询能力, 点位先放一起。后续有需要考虑解耦开
    BRPC_CLIENT_CALL,
    BRPC_SERIALIZE,
    BRPC_WRITEV,
    BRPC_DESERIALIZE,
    BRPC_READV,
    BRPC_SERVER_PROCESS_REQ,
    BRPC_CLIENT_PROCESS_RSP,

    // count the number of ProfilingTPId
    UBSOCKET_PROF_COUNT,
};

typedef struct {
    uint32_t tracepoint_count;  /* how many trace points to be recorded */
    int32_t enable_dump;        /* dump to file or not */
    const char *dump_file_path; /* dump file path */
    uint16_t dump_interval_min; /* dump interval in min */
} ubsocket_prof_option_t;

int ubsocket_prof_init(ubsocket_prof_option_t *option);

int ubsocket_prof_uninit();

int ubsocket_prof_record(uint32_t tracepoint_id, const char *tracepoint_name, uint64_t timestamp, bool good);

int ubsocket_prof_combind(char **out_buf);

void ubsocket_prof_reset();

static __always_inline uint64_t ubsocket_get_timeNs()
{
    struct timespec tpDelay = {0, 0};
    clock_gettime(CLOCK_MONOTONIC, &tpDelay);
    return tpDelay.tv_sec * 1000000000ULL + tpDelay.tv_nsec;
}

#define PROF_START(TP_ID)                           \
    uint64_t tpBegin##TP_ID = 0;                    \
    do {                                            \
        if (ubsocket_prof_enabled == 1) {           \
            tpBegin##TP_ID = ubsocket_get_timeNs(); \
        }                                           \
    } while (0)

#define PROF_END(TP_ID, GOOD)                                                                  \
    do {                                                                                       \
        if (ubsocket_prof_enabled == 1) {                                                      \
            ubsocket_prof_record(TP_ID, #TP_ID, ubsocket_get_timeNs() - tpBegin##TP_ID, GOOD); \
        }                                                                                      \
    } while (0)

#define PROF_RECORD(TP_ID, TP_NUM, GOOD)                       \
    do {                                                       \
        if (ubsocket_prof_enabled == 1) {                      \
            ubsocket_prof_record(TP_ID, #TP_ID, TP_NUM, GOOD); \
        }                                                      \
    } while (0)

#ifdef __cplusplus
}
#endif

#endif // UBS_COMM_UBSOCKET_PROFILING_H