mirror of
https://github.com/DarkflameUniverse/DarkflameServer.git
synced 2026-10-18 02:32:50 +00:00
2277cbd7e2
Stopping waited for every worker to finish its model, and with the path traced hidden-face removal a big model takes minutes, so a restart hung. UgcThrottle::Cancel(true) (set by UgcProcessor::Stop) makes Checkpoint, which the long loops already call, throw UgcThrottle::Cancelled; the worker abandons the job without writing or recording anything, so its row stays waiting and is made again after the restart. The throttle's sleeps wake to check it. Previews answer 503. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
44 lines
1.8 KiB
C++
44 lines
1.8 KiB
C++
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <string>
|
|
|
|
/**
|
|
* Keeps the UGC workers' CPU use under a budget. The long loops (renders, ambient occlusion, icons) call Checkpoint
|
|
* every few milliseconds of work; it adds the calling thread's CPU time since its last call to a shared account that
|
|
* the budget fills at `budget` CPUs per second, and sleeps while the account is overdrawn. So the workers together
|
|
* average at most the budget however many there are, long jobs included, and without a budget it costs nothing.
|
|
*/
|
|
namespace UgcThrottle {
|
|
// CPUs the workers may use together on average (1.5: one and a half cores); 0 or less: no limit
|
|
void SetBudget(double cpus);
|
|
double GetBudget();
|
|
|
|
// Account the calling thread's CPU time and sleep when over the budget. Throws Cancelled once Cancel(true) was
|
|
// called, so a job stops in moments when the server shuts down.
|
|
void Checkpoint();
|
|
|
|
// Thrown by Checkpoint while cancelled: the job is abandoned, not failed (its row stays waiting)
|
|
struct Cancelled {};
|
|
// Stop (true) or allow (false) every job's work: set when the server stops, cleared when it starts
|
|
void Cancel(bool cancel);
|
|
bool IsCancelled();
|
|
|
|
// Start accounting on this thread from now (a worker starting a job), so time spent idle isn't counted
|
|
void Begin();
|
|
|
|
struct Stats {
|
|
uint64_t sleptMs{}; // since start
|
|
int64_t lastSleepUnixMs{}; // when it last slept, 0 never
|
|
};
|
|
Stats GetStats();
|
|
|
|
// This thread's CPU time in seconds
|
|
double ThreadCpuSeconds();
|
|
|
|
// Parses "18-23" (from 18:00 to 23:59; may wrap past midnight, "22-6") into from and to; false when it isn't that
|
|
bool ParseHours(const std::string& text, int& from, int& to);
|
|
// Whether `hour` (0-23) is inside the range; never when from or to is negative
|
|
bool InHours(int hour, int from, int to);
|
|
}
|