Quellcode-Bibliothek ratelimit.h   Sprache: C

/* Ratelimit */
/* SPDX-FileCopyrightText: Copyright © 2023 Wim Taymans */
/* SPDX-License-Identifier: MIT */

#ifndef SPA_RATELIMIT_H
#define SPA_RATELIMIT_H

#ifdef __cplusplus
extern "C" {
#endif

#include <inttypes.h>
#include <stddef.h>

#include <spa/utils/defs.h>

#ifndef SPA_API_RATELIMIT
 #ifdef SPA_API_IMPL
  #define SPA_API_RATELIMIT SPA_API_IMPL
 #else
  #define SPA_API_RATELIMIT static inline
 #endif
#endif

struct spa_ratelimit {
 uint64_t interval;
 uint64_t begin;
 unsigned burst;
 unsigned n_printed;
 unsigned n_suppressed;
};

SPA_API_RATELIMIT int spa_ratelimit_test(struct spa_ratelimit *r, uint64_t now)
{
 unsigned suppressed = 0;
 if (r->begin + r->interval < now) {
  suppressed = r->n_suppressed;
  r->begin = now;
  r->n_printed = 0;
  r->n_suppressed = 0;
 } else if (r->n_printed >= r->burst) {
  r->n_suppressed++;
  return -1;
 }
 r->n_printed++;
 return suppressed;
}

#ifdef __cplusplus
} /* extern "C" */
#endif

#endif /* SPA_RATELIMIT_H */