/** @file

  Public RecProcess declarations

  @section license License

  Licensed to the Apache Software Foundation (ASF) under one

  or more contributor license agreements.  See the NOTICE file

  distributed with this work for additional information

  regarding copyright ownership.  The ASF licenses this file

  to you under the Apache License, Version 2.0 (the

  "License"); you may not use this file except in compliance

  with the License.  You may obtain a copy of the License at

      http://www.apache.org/licenses/LICENSE-2.0

  Unless required by applicable law or agreed to in writing, software

  distributed under the License is distributed on an "AS IS" BASIS,

  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  See the License for the specific language governing permissions and

  limitations under the License.

 */

#pragma once

#include "iocore/eventsystem/EThread.h"

#include "records/RecDefs.h"

//-------------------------------------------------------------------------

// RawStat Registration

//-------------------------------------------------------------------------

using RecRawStatBlockAllocator = RecRawStatBlock *(*)(int num_stats);

void             SetRecAllocateRawStatBlockAllocator(RecRawStatBlockAllocator);

RecRawStatBlock *RecAllocateRawStatBlock(int num_stats);

int _RecRegisterRawStat(RecRawStatBlock *rsb, RecT rec_type, const char *name, RecDataT data_type, RecPersistT persist_type, int id,

                        RecRawStatSyncCb sync_cb);

#define RecRegisterRawStat(rsb, rec_type, name, data_type, persist_type, id, sync_cb) \

  _RecRegisterRawStat((rsb), (rec_type), (name), (data_type), REC_PERSISTENCE_TYPE(persist_type), (id), (sync_cb))

// RecRawStatRange* RecAllocateRawStatRange (int num_buckets);

// int RecRegisterRawStatRange (RecRawStatRange *rsr,

//                           RecT rec_type,

//                           char *name,

//                           RecPersistT persist_type,

//                           int id,

//                           RecInt min,

//                           RecInt max);

//-------------------------------------------------------------------------

// Predefined RawStat Callbacks

//-------------------------------------------------------------------------

int RecRawStatSyncSum(const char *name, RecDataT data_type, RecData *data, RecRawStatBlock *rsb, int id);

int RecRawStatSyncCount(const char *name, RecDataT data_type, RecData *data, RecRawStatBlock *rsb, int id);

int RecRawStatSyncAvg(const char *name, RecDataT data_type, RecData *data, RecRawStatBlock *rsb, int id);

int RecRawStatSyncHrTimeAvg(const char *name, RecDataT data_type, RecData *data, RecRawStatBlock *rsb, int id);

int RecRawStatSyncIntMsecsToFloatSeconds(const char *name, RecDataT data_type, RecData *data, RecRawStatBlock *rsb, int id);

int RecRegisterRawStatSyncCb(const char *name, RecRawStatSyncCb sync_cb, RecRawStatBlock *rsb, int id);

int RecRawStatUpdateSum(RecRawStatBlock *rsb, int id);

int RecExecRawStatSyncCbs();

using RecCallbackFunction = std::function<void()>;

void RecRegNewSyncStatSync(RecCallbackFunction callback);

//-------------------------------------------------------------------------

// RawStat Setting/Getting

//-------------------------------------------------------------------------

// Note: The following RecIncrRawStatXXX calls are fast and don't

// require any ink_atomic_xxx64()'s to be executed.  Use these RawStat

// functions over other RawStat functions whenever possible.

inline int RecIncrRawStat(RecRawStatBlock *rsb, EThread *ethread, int id, int64_t incr = 1);

inline int RecIncrRawStatSum(RecRawStatBlock *rsb, EThread *ethread, int id, int64_t incr = 1);

inline int RecIncrRawStatCount(RecRawStatBlock *rsb, EThread *ethread, int id, int64_t incr = 1);

int RecSetRawStatSum(RecRawStatBlock *rsb, int id, int64_t data);

int RecSetRawStatCount(RecRawStatBlock *rsb, int id, int64_t data);

int RecGetRawStatSum(RecRawStatBlock *rsb, int id, int64_t *data);

int RecGetRawStatCount(RecRawStatBlock *rsb, int id, int64_t *data);

//-------------------------------------------------------------------------

// Global RawStat Items (e.g. same as above, but no thread-local behavior)

//-------------------------------------------------------------------------

int RecIncrGlobalRawStat(RecRawStatBlock *rsb, int id, int64_t incr = 1);

int RecIncrGlobalRawStatSum(RecRawStatBlock *rsb, int id, int64_t incr = 1);

int RecIncrGlobalRawStatCount(RecRawStatBlock *rsb, int id, int64_t incr = 1);

int RecSetGlobalRawStatSum(RecRawStatBlock *rsb, int id, int64_t data);

int RecSetGlobalRawStatCount(RecRawStatBlock *rsb, int id, int64_t data);

int RecGetGlobalRawStatSum(RecRawStatBlock *rsb, int id, int64_t *data);

int RecGetGlobalRawStatCount(RecRawStatBlock *rsb, int id, int64_t *data);

RecRawStat *RecGetGlobalRawStatPtr(RecRawStatBlock *rsb, int id);

int64_t    *RecGetGlobalRawStatSumPtr(RecRawStatBlock *rsb, int id);

int64_t    *RecGetGlobalRawStatCountPtr(RecRawStatBlock *rsb, int id);

//-------------------------------------------------------------------------

// RecIncrRawStatXXX

//-------------------------------------------------------------------------

// inlined functions that are used very frequently.

// FIXME: move it to Inline.cc

inline RecRawStat *

raw_stat_get_tlp(RecRawStatBlock *rsb, int id, EThread *ethread)

{

  ink_assert((id >= 0) && (id < rsb->max_stats));

  if (ethread == nullptr) {

    ethread = this_ethread();

  }

  return ((reinterpret_cast<RecRawStat *>(reinterpret_cast<char *>(ethread) + rsb->ethr_stat_offset)) + id);

}

inline int

RecIncrRawStat(RecRawStatBlock *rsb, EThread *ethread, int id, int64_t incr)

{

  RecRawStat *tlp  = raw_stat_get_tlp(rsb, id, ethread);

  tlp->sum        += incr;

  tlp->count      += 1;

  return REC_ERR_OKAY;

}

inline int

RecDecrRawStat(RecRawStatBlock *rsb, EThread *ethread, int id, int64_t decr)

{

  RecRawStat *tlp  = raw_stat_get_tlp(rsb, id, ethread);

  tlp->sum        -= decr;

  tlp->count      += 1;

  return REC_ERR_OKAY;

}

inline int

RecIncrRawStatSum(RecRawStatBlock *rsb, EThread *ethread, int id, int64_t incr)

{

  RecRawStat *tlp  = raw_stat_get_tlp(rsb, id, ethread);

  tlp->sum        += incr;

  return REC_ERR_OKAY;

}

inline int

RecIncrRawStatCount(RecRawStatBlock *rsb, EThread *ethread, int id, int64_t incr)

{

  RecRawStat *tlp  = raw_stat_get_tlp(rsb, id, ethread);

  tlp->count      += incr;

  return REC_ERR_OKAY;

}