mirror of
https://github.com/Motorhead1991/qemu.git
synced 2025-08-03 07:43:54 -06:00
e1fde0e038
These way we can make them atomic and use this functions from any place. I also moved all functions that use rate_limit to migration-stats. Functions got renamed, they are not qemu_file anymore. qemu_file_rate_limit -> migration_rate_exceeded qemu_file_set_rate_limit -> migration_rate_set qemu_file_get_rate_limit -> migration_rate_get qemu_file_reset_rate_limit -> migration_rate_reset qemu_file_acct_rate_limit -> migration_rate_account. Reviewed-by: Harsh Prateek Bora <harshpb@linux.ibm.com> Signed-off-by: Juan Quintela <quintela@redhat.com> Message-Id: <20230515195709.63843-6-quintela@redhat.com> Signed-off-by: Juan Quintela <quintela@redhat.com>
61 lines
1.3 KiB
C
61 lines
1.3 KiB
C
/*
|
|
* Migration stats
|
|
*
|
|
* Copyright (c) 2012-2023 Red Hat Inc
|
|
*
|
|
* Authors:
|
|
* Juan Quintela <quintela@redhat.com>
|
|
*
|
|
* This work is licensed under the terms of the GNU GPL, version 2 or later.
|
|
* See the COPYING file in the top-level directory.
|
|
*/
|
|
|
|
#include "qemu/osdep.h"
|
|
#include "qemu/stats64.h"
|
|
#include "qemu-file.h"
|
|
#include "migration-stats.h"
|
|
|
|
MigrationAtomicStats mig_stats;
|
|
|
|
bool migration_rate_exceeded(QEMUFile *f)
|
|
{
|
|
if (qemu_file_get_error(f)) {
|
|
return true;
|
|
}
|
|
|
|
uint64_t rate_limit_used = stat64_get(&mig_stats.rate_limit_used);
|
|
uint64_t rate_limit_max = stat64_get(&mig_stats.rate_limit_max);
|
|
|
|
if (rate_limit_max == RATE_LIMIT_DISABLED) {
|
|
return false;
|
|
}
|
|
if (rate_limit_max > 0 && rate_limit_used > rate_limit_max) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
uint64_t migration_rate_get(void)
|
|
{
|
|
return stat64_get(&mig_stats.rate_limit_max);
|
|
}
|
|
|
|
#define XFER_LIMIT_RATIO (1000 / BUFFER_DELAY)
|
|
|
|
void migration_rate_set(uint64_t limit)
|
|
{
|
|
/*
|
|
* 'limit' is per second. But we check it each BUFER_DELAY miliseconds.
|
|
*/
|
|
stat64_set(&mig_stats.rate_limit_max, limit / XFER_LIMIT_RATIO);
|
|
}
|
|
|
|
void migration_rate_reset(void)
|
|
{
|
|
stat64_set(&mig_stats.rate_limit_used, 0);
|
|
}
|
|
|
|
void migration_rate_account(uint64_t len)
|
|
{
|
|
stat64_add(&mig_stats.rate_limit_used, len);
|
|
}
|