MrBounty
6fa8d6b7c6
Bring test back Removed useless arena, making like 40% perf improv Removed shared Loc
51 lines
1.6 KiB
Zig
51 lines
1.6 KiB
Zig
const std = @import("std");
|
|
const U64 = std.atomic.Value(u64);
|
|
const Pool = std.Thread.Pool;
|
|
|
|
const ZipponError = @import("error").ZipponError;
|
|
const CPU_CORE = @import("config").CPU_CORE;
|
|
const log = std.log.scoped(.thread);
|
|
|
|
pub const ThreadEngine = @This();
|
|
|
|
thread_arena: *std.heap.ThreadSafeAllocator,
|
|
thread_pool: *Pool,
|
|
|
|
pub fn init(allocator: std.mem.Allocator) !ThreadEngine {
|
|
const thread_arena = try allocator.create(std.heap.ThreadSafeAllocator);
|
|
thread_arena.* = std.heap.ThreadSafeAllocator{
|
|
.child_allocator = allocator,
|
|
};
|
|
|
|
const cpu_core = if (CPU_CORE == 0) std.Thread.getCpuCount() catch 1 else CPU_CORE;
|
|
log.info("Using {d} cpu core", .{cpu_core});
|
|
log.info("Using {d}Mb stack size", .{std.Thread.SpawnConfig.default_stack_size / 1024 / 1024});
|
|
|
|
const thread_pool = try allocator.create(std.Thread.Pool);
|
|
try thread_pool.init(std.Thread.Pool.Options{
|
|
.allocator = thread_arena.allocator(),
|
|
.n_jobs = cpu_core,
|
|
});
|
|
|
|
return ThreadEngine{
|
|
.thread_pool = thread_pool,
|
|
.thread_arena = thread_arena,
|
|
};
|
|
}
|
|
|
|
pub fn deinit(self: *ThreadEngine) void {
|
|
const parent_allocator = self.thread_arena.allocator();
|
|
self.thread_pool.deinit();
|
|
parent_allocator.destroy(self.thread_arena);
|
|
parent_allocator.destroy(self.thread_pool);
|
|
}
|
|
|
|
// Not tested, for later when config is runtime
|
|
pub fn setCpuCore(self: *ThreadEngine, cpu_core: usize) void {
|
|
self.thread_pool.deinit();
|
|
self.thread_pool.init(std.Thread.Pool.Options{
|
|
.allocator = self.thread_arena.allocator(),
|
|
.n_jobs = cpu_core,
|
|
});
|
|
}
|