138 lines
3.9 KiB
C
138 lines
3.9 KiB
C
/*
|
|
Copyright (c) 2015 Anton Titov.
|
|
|
|
Copyright (c) 2015 pCloud Ltd. All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions
|
|
are met: Redistributions of source code must retain the above
|
|
copyright notice, this list of conditions and the following
|
|
disclaimer. Redistributions in binary form must reproduce the
|
|
above copyright notice, this list of conditions and the following
|
|
disclaimer in the documentation and/or other materials provided
|
|
with the distribution. Neither the name of pCloud Ltd nor the
|
|
names of its contributors may be used to endorse or promote
|
|
products derived from this software without specific prior written
|
|
permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
|
FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL pCloud
|
|
Ltd BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
|
OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
|
|
USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
|
|
DAMAGE.
|
|
*/
|
|
|
|
#include "plibs.h"
|
|
#include "prunratelimit.h"
|
|
#include "ptimer.h"
|
|
#include "ptree.h"
|
|
|
|
typedef struct {
|
|
psync_tree tree;
|
|
psync_run_ratelimit_callback0 call;
|
|
const char *name;
|
|
unsigned char scheduled;
|
|
} psync_rr_tree_node;
|
|
|
|
static pthread_mutex_t task_mutex=PTHREAD_MUTEX_INITIALIZER;
|
|
static psync_tree *tasks=PSYNC_TREE_EMPTY;
|
|
|
|
static void psync_run_ratelimited_timer(psync_timer_t timer, void *ptr){
|
|
psync_rr_tree_node *node;
|
|
const char *name;
|
|
psync_run_ratelimit_callback0 call;
|
|
int run;
|
|
node=(psync_rr_tree_node *)ptr;
|
|
pthread_mutex_lock(&task_mutex);
|
|
if (node->scheduled){
|
|
run=1;
|
|
node->scheduled=0;
|
|
call=node->call;
|
|
name=node->name;
|
|
}
|
|
else{
|
|
run=0;
|
|
psync_tree_del(&tasks, &node->tree);
|
|
}
|
|
pthread_mutex_unlock(&task_mutex);
|
|
if (run){
|
|
debug(D_NOTICE, "running %s in a thread", name);
|
|
psync_run_thread(name, call);
|
|
}
|
|
else{
|
|
psync_timer_stop(timer);
|
|
psync_free(node);
|
|
}
|
|
}
|
|
|
|
void psync_run_ratelimited(const char *name, psync_run_ratelimit_callback0 call, uint32_t minintervalsec, int runinthread){
|
|
psync_tree *tr, **addto;
|
|
psync_rr_tree_node *node;
|
|
int found;
|
|
found=0;
|
|
pthread_mutex_lock(&task_mutex);
|
|
tr=tasks;
|
|
if (tr){
|
|
while (1){
|
|
node=psync_tree_element(tr, psync_rr_tree_node, tree);
|
|
if (call<node->call){
|
|
if (tr->left)
|
|
tr=tr->left;
|
|
else{
|
|
addto=&tr->left;
|
|
break;
|
|
}
|
|
}
|
|
else if (call>node->call){
|
|
if (tr->right)
|
|
tr=tr->right;
|
|
else{
|
|
addto=&tr->right;
|
|
break;
|
|
}
|
|
}
|
|
else{
|
|
found=1;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
addto=&tasks;
|
|
if (found){
|
|
if (node->scheduled)
|
|
debug(D_NOTICE, "skipping run of %s as it is already scheduled", name);
|
|
else{
|
|
debug(D_NOTICE, "scheduling run of %s on existing timer", name);
|
|
node->scheduled=1;
|
|
}
|
|
}
|
|
else{
|
|
node=psync_new(psync_rr_tree_node);
|
|
node->call=call;
|
|
node->name=name;
|
|
node->scheduled=0;
|
|
*addto=&node->tree;
|
|
psync_tree_added_at(&tasks, tr, &node->tree);
|
|
}
|
|
pthread_mutex_unlock(&task_mutex);
|
|
if (!found){
|
|
if (runinthread){
|
|
debug(D_NOTICE, "running %s in a thread", name);
|
|
psync_run_thread(name, call);
|
|
}
|
|
else{
|
|
debug(D_NOTICE, "running %s on this thread", name);
|
|
call();
|
|
}
|
|
psync_timer_register(psync_run_ratelimited_timer, minintervalsec, node);
|
|
}
|
|
}
|