engine.zig 76 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613
  1. const std = @import("std");
  2. const pkvdb = @import("pkvdb.zig");
  3. const keydir = @import("keydir.zig");
  4. const ordered_index = @import("ordered_index.zig");
  5. pub const RecordRef = keydir.RecordRef;
  6. const KeyReader = keydir.Reader;
  7. pub const pkvdb_max_frame = pkvdb.max_transaction_size;
  8. pub const Operation = struct {
  9. opcode: pkvdb.Opcode,
  10. key: []const u8,
  11. value: []const u8 = "",
  12. };
  13. pub const Value = struct {
  14. bytes: []u8,
  15. lsn: u64,
  16. };
  17. pub const CompareCheck = struct {
  18. key: []const u8,
  19. expected_lsn: u64,
  20. };
  21. pub const CompareBatchResult = struct {
  22. lsn: u64,
  23. committed: bool,
  24. };
  25. pub const ScanEntry = struct {
  26. key: []u8,
  27. value: ?[]u8,
  28. lsn: u64,
  29. };
  30. pub const ScanBatch = struct {
  31. entries: []ScanEntry,
  32. next_cursor: []u8,
  33. done: bool,
  34. pub fn deinit(self: *ScanBatch, allocator: std.mem.Allocator) void {
  35. for (self.entries) |entry| {
  36. allocator.free(entry.key);
  37. if (entry.value) |value| allocator.free(value);
  38. }
  39. allocator.free(self.entries);
  40. allocator.free(self.next_cursor);
  41. self.* = undefined;
  42. }
  43. };
  44. pub const Status = struct {
  45. uuid: [16]u8,
  46. file_bytes: u64,
  47. latest_lsn: u64,
  48. oldest_lsn: u64,
  49. checkpoint_lsn: u64,
  50. journal_bytes_since_checkpoint: u64,
  51. live_keys: u64,
  52. keydir_bytes: u64,
  53. ordered_index_bytes: u64,
  54. bytes_written: u64,
  55. checksum_failures: u64,
  56. partial_tails: u64,
  57. recovery_ns: u64,
  58. checkpoint_ns: u64,
  59. connection_bytes: u64,
  60. active_requests: u64,
  61. commit_groups: u64,
  62. committed_transactions: u64,
  63. largest_commit_group: u64,
  64. };
  65. const Root = struct {
  66. block: pkvdb.Superblock,
  67. manifest: ?pkvdb.Manifest,
  68. };
  69. const PendingWrite = struct {
  70. operations: []const Operation,
  71. metadata: []const u8,
  72. checks: []const CompareCheck = &.{},
  73. next: ?*PendingWrite = null,
  74. completion: *WriteCompletion,
  75. lsn: u64 = 0,
  76. committed: bool = true,
  77. frame_position: usize = 0,
  78. changed: bool = false,
  79. prepared_position: usize = 0,
  80. prepared_count: usize = 0,
  81. bytes: usize,
  82. };
  83. const WriteCompletion = struct {
  84. condition: std.Thread.Condition = .{},
  85. remaining: usize,
  86. failure: ?anyerror = null,
  87. };
  88. const max_group_transactions = 4096;
  89. const max_group_bytes = 64 * 1024 * 1024;
  90. const max_queued_bytes = 128 * 1024 * 1024;
  91. const group_wait_ns = 250 * std.time.ns_per_us;
  92. const map_interval = 64 * 1024 * 1024;
  93. const Mapping = struct {
  94. bytes: []align(std.heap.page_size_min) u8,
  95. file_start: u64,
  96. logical_start: u64,
  97. logical_end: u64,
  98. };
  99. pub const Engine = struct {
  100. allocator: std.mem.Allocator,
  101. file: std.fs.File,
  102. directory: keydir.KeyDir,
  103. ordered: ordered_index.OrderedIndex,
  104. ordered_ready: bool = false,
  105. mappings: std.ArrayListUnmanaged(Mapping) = .{},
  106. mapping_lock: std.Thread.RwLock = .{},
  107. mapped_length: u64 = 0,
  108. lock: std.Thread.RwLock = .{},
  109. io_mutex: std.Thread.Mutex = .{},
  110. checkpoint_mutex: std.Thread.Mutex = .{},
  111. ordered_gate: std.Thread.Mutex = .{},
  112. queue_mutex: std.Thread.Mutex = .{},
  113. queue_condition: std.Thread.Condition = .{},
  114. queue_head: ?*PendingWrite = null,
  115. queue_tail: ?*PendingWrite = null,
  116. queued_bytes: usize = 0,
  117. writer_thread: ?std.Thread = null,
  118. writer_stopping: bool = false,
  119. writer_failed: bool = false,
  120. commit_groups: u64 = 0,
  121. committed_transactions: u64 = 0,
  122. largest_commit_group: u64 = 0,
  123. file_length: u64,
  124. latest_lsn: u64 = 0,
  125. oldest_lsn: u64 = 0,
  126. checkpoint_lsn: u64 = 0,
  127. journal_bytes_since_checkpoint: u64 = 0,
  128. bytes_written: u64 = 0,
  129. checksum_failures: u64 = 0,
  130. partial_tails: u64 = 0,
  131. recovery_ns: u64 = 0,
  132. checkpoint_ns: u64 = 0,
  133. connection_bytes: std.atomic.Value(u64) = std.atomic.Value(u64).init(0),
  134. active_requests: std.atomic.Value(u64) = std.atomic.Value(u64).init(0),
  135. uuid: [16]u8,
  136. generation: u64,
  137. active_superblock: u1,
  138. created_ns: i64,
  139. pub fn open(allocator: std.mem.Allocator, path: []const u8) !Engine {
  140. const started = std.time.nanoTimestamp();
  141. const file = std.fs.cwd().openFile(path, .{ .mode = .read_write }) catch |err| switch (err) {
  142. error.FileNotFound => try std.fs.cwd().createFile(path, .{ .read = true, .truncate = false }),
  143. else => return err,
  144. };
  145. errdefer file.close();
  146. var directory = try keydir.KeyDir.init(allocator);
  147. errdefer directory.deinit();
  148. var engine = Engine{
  149. .allocator = allocator,
  150. .file = file,
  151. .directory = directory,
  152. .ordered = ordered_index.OrderedIndex.init(allocator),
  153. .file_length = try file.getEndPos(),
  154. .uuid = undefined,
  155. .generation = 0,
  156. .active_superblock = 0,
  157. .created_ns = now(),
  158. };
  159. errdefer engine.ordered.deinit();
  160. if (engine.file_length == 0) {
  161. try engine.initialize();
  162. } else {
  163. try engine.recover();
  164. }
  165. const elapsed = std.time.nanoTimestamp() - started;
  166. engine.recovery_ns = if (elapsed > 0) @intCast(elapsed) else 0;
  167. engine.mapTail(true);
  168. return engine;
  169. }
  170. pub fn close(self: *Engine) void {
  171. self.queue_mutex.lock();
  172. self.writer_stopping = true;
  173. self.queue_condition.broadcast();
  174. self.queue_mutex.unlock();
  175. if (self.writer_thread) |thread| thread.join();
  176. for (self.mappings.items) |mapping| std.posix.munmap(mapping.bytes);
  177. self.mappings.deinit(self.allocator);
  178. self.ordered.deinit();
  179. self.directory.deinit();
  180. self.file.close();
  181. self.* = undefined;
  182. }
  183. fn now() i64 {
  184. const value = std.time.nanoTimestamp();
  185. return std.math.cast(i64, value) orelse if (value < 0) std.math.minInt(i64) else std.math.maxInt(i64);
  186. }
  187. fn mapTail(self: *Engine, force: bool) void {
  188. if (self.file_length <= self.mapped_length or !force and self.file_length - self.mapped_length < map_interval) return;
  189. const page_size = std.heap.pageSize();
  190. const file_start = self.mapped_length - self.mapped_length % page_size;
  191. const length = std.math.cast(usize, self.file_length - file_start) orelse return;
  192. const bytes = std.posix.mmap(null, length, std.posix.PROT.READ, .{ .TYPE = .SHARED }, self.file.handle, file_start) catch return;
  193. if (!force) self.mapping_lock.lock();
  194. defer if (!force) self.mapping_lock.unlock();
  195. self.mappings.append(self.allocator, .{ .bytes = bytes, .file_start = file_start, .logical_start = self.mapped_length, .logical_end = self.file_length }) catch {
  196. std.posix.munmap(bytes);
  197. return;
  198. };
  199. self.mapped_length = self.file_length;
  200. }
  201. fn readBytes(self: *Engine, destination: []u8, file_offset: u64) !usize {
  202. const read_end = try std.math.add(u64, file_offset, destination.len);
  203. self.mapping_lock.lockShared();
  204. var index = self.mappings.items.len;
  205. while (index != 0) {
  206. index -= 1;
  207. const mapping = self.mappings.items[index];
  208. if (file_offset >= mapping.logical_start and read_end <= mapping.logical_end) {
  209. const start: usize = @intCast(file_offset - mapping.file_start);
  210. @memcpy(destination, mapping.bytes[start .. start + destination.len]);
  211. self.mapping_lock.unlockShared();
  212. return destination.len;
  213. }
  214. }
  215. self.mapping_lock.unlockShared();
  216. return self.file.preadAll(destination, file_offset);
  217. }
  218. fn readKeyBytes(context: *const anyopaque, destination: []u8, file_offset: u64) anyerror!usize {
  219. const self: *Engine = @ptrCast(@alignCast(@constCast(context)));
  220. return self.readBytes(destination, file_offset);
  221. }
  222. fn keyReader(self: *Engine) KeyReader {
  223. return .{ .context = self, .readFn = readKeyBytes };
  224. }
  225. fn initialize(self: *Engine) !void {
  226. std.crypto.random.bytes(&self.uuid);
  227. self.generation = 1;
  228. self.created_ns = now();
  229. var first: [4096]u8 = undefined;
  230. var second: [4096]u8 = undefined;
  231. pkvdb.encodeSuperblock(.{
  232. .generation = 1,
  233. .uuid = self.uuid,
  234. .manifest_offset = 0,
  235. .checkpoint_lsn = 0,
  236. .known_lsn = 0,
  237. .known_file_length = pkvdb.data_offset,
  238. .created_ns = self.created_ns,
  239. .updated_ns = self.created_ns,
  240. }, &first);
  241. pkvdb.encodeSuperblock(.{
  242. .generation = 0,
  243. .uuid = self.uuid,
  244. .manifest_offset = 0,
  245. .checkpoint_lsn = 0,
  246. .known_lsn = 0,
  247. .known_file_length = pkvdb.data_offset,
  248. .created_ns = self.created_ns,
  249. .updated_ns = self.created_ns,
  250. }, &second);
  251. try self.file.pwriteAll(&first, 0);
  252. try self.file.pwriteAll(&second, pkvdb.superblock_size);
  253. try self.file.sync();
  254. self.file_length = pkvdb.data_offset;
  255. try self.file.seekTo(self.file_length);
  256. self.bytes_written = pkvdb.data_offset;
  257. }
  258. fn readExtentHeader(self: *Engine, offset: u64) !pkvdb.ExtentHeader {
  259. var bytes: [64]u8 = undefined;
  260. if (try self.file.preadAll(&bytes, offset) != bytes.len) return error.Truncated;
  261. return pkvdb.decodeExtentHeader(&bytes) catch |err| {
  262. if (err == error.ChecksumMismatch) self.checksum_failures += 1;
  263. return err;
  264. };
  265. }
  266. fn validatePayload(self: *Engine, offset: u64, length: u64, expected: u32) !void {
  267. var crc: u32 = 0xffffffff;
  268. var buffer: [64 * 1024]u8 = undefined;
  269. var done: u64 = 0;
  270. while (done < length) {
  271. const amount: usize = @intCast(@min(buffer.len, length - done));
  272. if (try self.file.preadAll(buffer[0..amount], try std.math.add(u64, offset, done)) != amount) return error.Truncated;
  273. crc = pkvdb.crc32cUpdate(crc, buffer[0..amount]);
  274. done += amount;
  275. }
  276. if (~crc != expected) {
  277. self.checksum_failures += 1;
  278. return error.ChecksumMismatch;
  279. }
  280. }
  281. fn readExtentPayload(self: *Engine, offset: u64, header: pkvdb.ExtentHeader, max: u64) ![]u8 {
  282. if (header.payload_length > max or header.payload_length > std.math.maxInt(usize)) return error.InvalidLength;
  283. const payload_offset = try std.math.add(u64, offset, pkvdb.extent_header_size);
  284. const payload = try self.allocator.alloc(u8, @intCast(header.payload_length));
  285. errdefer self.allocator.free(payload);
  286. if (try self.file.preadAll(payload, payload_offset) != payload.len) return error.Truncated;
  287. if (pkvdb.crc32c(payload) != header.payload_crc) {
  288. self.checksum_failures += 1;
  289. return error.ChecksumMismatch;
  290. }
  291. return payload;
  292. }
  293. fn extentEnd(offset: u64, payload_length: u64) !u64 {
  294. return pkvdb.align8(try std.math.add(u64, try std.math.add(u64, offset, pkvdb.extent_header_size), payload_length));
  295. }
  296. fn loadManifest(self: *Engine, sb: pkvdb.Superblock) !?pkvdb.Manifest {
  297. if (sb.manifest_offset == 0) return null;
  298. const header = try self.readExtentHeader(sb.manifest_offset);
  299. if (header.extent_type != .manifest or header.version != 1 or header.payload_length != pkvdb.manifest_size) return error.InvalidManifest;
  300. if (try extentEnd(sb.manifest_offset, header.payload_length) > sb.known_file_length) return error.InvalidManifest;
  301. const payload = try self.readExtentPayload(sb.manifest_offset, header, pkvdb.manifest_size);
  302. defer self.allocator.free(payload);
  303. const manifest = try pkvdb.decodeManifest(payload);
  304. if (!std.mem.eql(u8, &manifest.uuid, &sb.uuid) or manifest.generation != sb.generation or manifest.checkpoint_lsn != sb.checkpoint_lsn) return error.InvalidManifest;
  305. if (manifest.known_tail != sb.known_file_length or manifest.known_lsn != sb.known_lsn) return error.InvalidManifest;
  306. if (manifest.entries_offset == 0 or manifest.entries_offset >= sb.manifest_offset or manifest.replay_offset < pkvdb.data_offset or manifest.replay_offset > manifest.entries_offset) return error.InvalidManifest;
  307. try self.validateCheckpointExtent(manifest.entries_offset, manifest.checkpoint_lsn, sb.known_file_length);
  308. return manifest;
  309. }
  310. fn validateCheckpointExtent(self: *Engine, offset: u64, lsn: u64, file_length: u64) !void {
  311. const header = try self.readExtentHeader(offset);
  312. if (header.extent_type != .checkpoint_entries or header.version != 1 or header.first_lsn != lsn or header.last_lsn != lsn) return error.InvalidCheckpoint;
  313. const end = try extentEnd(offset, header.payload_length);
  314. if (end > file_length) return error.InvalidCheckpoint;
  315. try self.validatePayload(offset + pkvdb.extent_header_size, header.payload_length, header.payload_crc);
  316. var bytes: [56]u8 = undefined;
  317. if (header.payload_length < bytes.len or try self.file.preadAll(&bytes, offset + pkvdb.extent_header_size) != bytes.len) return error.InvalidCheckpoint;
  318. const checkpoint_header = try pkvdb.decodeCheckpointHeaderOnly(&bytes, header.payload_length);
  319. if (checkpoint_header.lsn != lsn) return error.InvalidCheckpoint;
  320. }
  321. fn recover(self: *Engine) !void {
  322. if (self.file_length < pkvdb.data_offset) return error.Truncated;
  323. var blocks: [2][4096]u8 = undefined;
  324. _ = try self.file.preadAll(&blocks[0], 0);
  325. _ = try self.file.preadAll(&blocks[1], pkvdb.superblock_size);
  326. var roots: [2]?Root = .{ null, null };
  327. for (0..2) |index| {
  328. const sb = pkvdb.decodeSuperblock(&blocks[index], self.file_length) catch continue;
  329. const manifest = self.loadManifest(sb) catch continue;
  330. roots[index] = .{ .block = sb, .manifest = manifest };
  331. }
  332. var selected_index: usize = 0;
  333. const selected = if (roots[0] != null and roots[1] != null) blk: {
  334. selected_index = if (roots[1].?.block.generation > roots[0].?.block.generation) 1 else 0;
  335. break :blk roots[selected_index].?;
  336. } else if (roots[0]) |root| root else if (roots[1]) |root| blk: {
  337. selected_index = 1;
  338. break :blk root;
  339. } else return error.NoUsableSuperblock;
  340. self.uuid = selected.block.uuid;
  341. self.generation = selected.block.generation;
  342. self.active_superblock = @intCast(selected_index);
  343. self.created_ns = selected.block.created_ns;
  344. self.latest_lsn = selected.block.checkpoint_lsn;
  345. self.checkpoint_lsn = selected.block.checkpoint_lsn;
  346. var replay_offset = pkvdb.data_offset;
  347. if (selected.manifest) |manifest| {
  348. try self.loadCheckpoint(manifest.entries_offset, selected.block.known_file_length);
  349. replay_offset = manifest.replay_offset;
  350. self.oldest_lsn = manifest.history_start_lsn;
  351. }
  352. try self.scanExtents(replay_offset);
  353. try self.file.setEndPos(self.file_length);
  354. try self.file.seekTo(self.file_length);
  355. self.bytes_written = self.file_length;
  356. }
  357. fn loadCheckpoint(self: *Engine, offset: u64, known_length: u64) !void {
  358. const extent = try self.readExtentHeader(offset);
  359. var header_bytes: [56]u8 = undefined;
  360. const payload_offset = offset + pkvdb.extent_header_size;
  361. if (try self.file.preadAll(&header_bytes, payload_offset) != header_bytes.len) return error.Truncated;
  362. const header = try pkvdb.decodeCheckpointHeaderOnly(&header_bytes, extent.payload_length);
  363. if (header.entry_count > std.math.maxInt(usize)) return error.InvalidLength;
  364. try self.directory.ensureAdditional(self.file, @intCast(header.entry_count));
  365. var index: u64 = 0;
  366. while (index < header.entry_count) : (index += 1) {
  367. var bytes: [48]u8 = undefined;
  368. const entry_offset = try std.math.add(u64, payload_offset + pkvdb.checkpoint_header_size, try std.math.mul(u64, index, pkvdb.checkpoint_entry_size));
  369. if (try self.file.preadAll(&bytes, entry_offset) != bytes.len) return error.Truncated;
  370. const entry = try pkvdb.decodeCheckpointEntry(&bytes, known_length);
  371. const record = RecordRef{ .hash = entry.hash, .lsn = entry.lsn, .key_offset = entry.key_offset, .value_offset = entry.value_offset, .key_len = entry.key_len, .value_len = entry.value_len, .flags = entry.flags };
  372. try self.directory.put(self.file, record);
  373. }
  374. }
  375. fn scanExtents(self: *Engine, start: u64) !void {
  376. var offset = start;
  377. var last_journal_lsn: u64 = 0;
  378. while (offset < self.file_length) {
  379. if (self.file_length - offset < pkvdb.extent_header_size) {
  380. self.partial_tails += 1;
  381. self.file_length = offset;
  382. break;
  383. }
  384. const header = self.readExtentHeader(offset) catch |failure| {
  385. if (self.file_length - offset == pkvdb.extent_header_size) {
  386. self.partial_tails += 1;
  387. self.file_length = offset;
  388. break;
  389. }
  390. return failure;
  391. };
  392. const end = extentEnd(offset, header.payload_length) catch return error.InvalidLength;
  393. if (end > self.file_length) {
  394. self.partial_tails += 1;
  395. self.file_length = offset;
  396. break;
  397. }
  398. try self.validatePayload(offset + pkvdb.extent_header_size, header.payload_length, header.payload_crc);
  399. if (header.extent_type == .journal and header.version == 1) {
  400. if (header.payload_length > pkvdb.max_transaction_size + pkvdb.group_header_size) return error.InvalidLength;
  401. const payload = try self.readExtentPayload(offset, header, pkvdb.max_transaction_size + pkvdb.group_header_size);
  402. defer self.allocator.free(payload);
  403. last_journal_lsn = try self.replayJournal(offset, payload, last_journal_lsn);
  404. self.journal_bytes_since_checkpoint += end - offset;
  405. } else if (header.extent_type == .store_metadata and header.version == 1) {
  406. if (header.payload_length > pkvdb.max_transaction_size) return error.InvalidLength;
  407. const payload = try self.readExtentPayload(offset, header, pkvdb.max_transaction_size);
  408. defer self.allocator.free(payload);
  409. try self.replayBaseline(offset, payload);
  410. }
  411. offset = end;
  412. }
  413. }
  414. fn replayBaseline(self: *Engine, extent_offset: u64, payload: []const u8) !void {
  415. if (payload.len < 24 or readInt(u16, payload, 0) != 1 or readInt(u64, payload, 8) != 1) return error.InvalidBaseline;
  416. const count = readInt(u32, payload, 4);
  417. if (count > pkvdb.max_operations) return error.InvalidBaseline;
  418. try self.directory.ensureAdditional(self.file, count);
  419. var position: usize = 24;
  420. for (0..count) |_| {
  421. if (position > payload.len or payload.len - position < 8) return error.InvalidBaseline;
  422. const key_length = readInt(u32, payload, position);
  423. const value_length = readInt(u32, payload, position + 4);
  424. if (key_length > pkvdb.max_key_size or value_length > pkvdb.max_value_size) return error.InvalidBaseline;
  425. const key_start = position + 8;
  426. const key_end = try std.math.add(usize, key_start, key_length);
  427. const value_end = try std.math.add(usize, key_end, value_length);
  428. if (value_end > payload.len) return error.InvalidBaseline;
  429. const key = payload[key_start..key_end];
  430. const key_offset = extent_offset + pkvdb.extent_header_size + key_start;
  431. try self.replaceRecord(key, .{ .hash = keydir.KeyDir.hash(key), .lsn = 1, .key_offset = key_offset, .value_offset = key_offset + key_length, .key_len = key_length, .value_len = value_length });
  432. position = @intCast(try pkvdb.align8(value_end));
  433. }
  434. if (position != payload.len) return error.InvalidBaseline;
  435. self.latest_lsn = 1;
  436. self.oldest_lsn = 1;
  437. }
  438. fn replayJournal(self: *Engine, extent_offset: u64, payload: []const u8, previous_lsn: u64) !u64 {
  439. if (payload.len < pkvdb.group_header_size or readInt(u16, payload, 0) != 1) return error.InvalidJournal;
  440. const count = readInt(u32, payload, 4);
  441. if (count == 0 or count > pkvdb.max_operations) return error.InvalidJournal;
  442. var position: usize = pkvdb.group_header_size;
  443. var last = previous_lsn;
  444. for (0..count) |_| {
  445. if (position > payload.len or payload.len - position < pkvdb.transaction_header_size) return error.InvalidJournal;
  446. const tx = try pkvdb.decodeTransactionHeader(payload[position..]);
  447. if (tx.total_length > payload.len - position) return error.InvalidJournal;
  448. const tx_end = position + tx.total_length;
  449. const body = payload[position + pkvdb.transaction_header_size .. tx_end];
  450. if (pkvdb.crc32c(body) != tx.payload_crc or tx.metadata_length > body.len) return error.ChecksumMismatch;
  451. if (last != 0 and tx.lsn <= last) return error.InvalidLsn;
  452. last = tx.lsn;
  453. if (tx.lsn > self.latest_lsn) try self.applyTransaction(extent_offset + pkvdb.extent_header_size, position, tx, payload[position..tx_end]);
  454. position = tx_end;
  455. }
  456. if (position != payload.len) return error.InvalidJournal;
  457. return last;
  458. }
  459. fn applyTransaction(self: *Engine, extent_payload_offset: u64, tx_position: usize, tx: pkvdb.TransactionHeader, frame: []const u8) !void {
  460. var position: usize = pkvdb.transaction_header_size + tx.metadata_length;
  461. try self.directory.ensureAdditional(self.file, tx.operation_count);
  462. for (0..tx.operation_count) |_| {
  463. if (position > frame.len or frame.len - position < pkvdb.operation_header_size) return error.InvalidJournal;
  464. const operation = try pkvdb.decodeOperationHeader(frame[position..]);
  465. const data_start = try std.math.add(usize, position, pkvdb.operation_header_size);
  466. const key_end = try std.math.add(usize, data_start, operation.key_length);
  467. const value_end = try std.math.add(usize, key_end, operation.value_length);
  468. const extension_end = try std.math.add(usize, value_end, operation.extension_length);
  469. if (extension_end > frame.len) return error.InvalidJournal;
  470. const key = frame[data_start..key_end];
  471. switch (operation.opcode) {
  472. .put => {
  473. const key_offset = try std.math.add(u64, extent_payload_offset, tx_position + data_start);
  474. const record = RecordRef{ .hash = keydir.KeyDir.hash(key), .lsn = tx.lsn, .key_offset = key_offset, .value_offset = key_offset + operation.key_length, .key_len = operation.key_length, .value_len = operation.value_length };
  475. try self.replaceRecord(key, record);
  476. },
  477. .delete => _ = try self.removeRecord(key),
  478. }
  479. position = @intCast(try pkvdb.align8(extension_end));
  480. }
  481. if (position != frame.len) return error.InvalidJournal;
  482. self.latest_lsn = tx.lsn;
  483. if (self.oldest_lsn == 0) self.oldest_lsn = tx.lsn;
  484. }
  485. fn replaceRecord(self: *Engine, key: []const u8, record: RecordRef) !void {
  486. if (self.ordered_ready) try self.ordered.put(key, record);
  487. try self.directory.putWithKey(self.file, key, record);
  488. }
  489. fn removeRecord(self: *Engine, key: []const u8) !bool {
  490. const removed = try self.directory.remove(self.file, key);
  491. if (removed and self.ordered_ready and !self.ordered.remove(key)) return error.IndexInconsistent;
  492. return removed;
  493. }
  494. fn appendExtent(self: *Engine, extent_type: pkvdb.ExtentType, first_lsn: u64, last_lsn: u64, payload: []const u8) !u64 {
  495. const offset = try pkvdb.align8(self.file_length);
  496. if (offset != self.file_length) {
  497. const padding = [_]u8{0} ** 8;
  498. try self.file.pwriteAll(padding[0 .. offset - self.file_length], self.file_length);
  499. }
  500. var header: [64]u8 = undefined;
  501. pkvdb.encodeExtentHeader(.{ .extent_type = extent_type, .payload_length = payload.len, .first_lsn = first_lsn, .last_lsn = last_lsn, .payload_crc = pkvdb.crc32c(payload) }, &header);
  502. const old_length = self.file_length;
  503. errdefer self.file.setEndPos(old_length) catch {};
  504. try self.file.pwriteAll(&header, offset);
  505. try self.file.pwriteAll(payload, offset + header.len);
  506. const end = try extentEnd(offset, payload.len);
  507. if (end > offset + header.len + payload.len) {
  508. const padding = [_]u8{0} ** 8;
  509. try self.file.pwriteAll(padding[0 .. end - (offset + header.len + payload.len)], offset + header.len + payload.len);
  510. }
  511. self.file_length = end;
  512. self.bytes_written += end - offset;
  513. return offset;
  514. }
  515. fn transactionLength(operations: []const Operation, metadata: []const u8) !usize {
  516. var frame_length: u64 = pkvdb.transaction_header_size + metadata.len;
  517. for (operations) |operation| {
  518. const raw = try std.math.add(u64, pkvdb.operation_header_size + operation.key.len + operation.value.len, 0);
  519. frame_length = try pkvdb.align8(try std.math.add(u64, frame_length, raw));
  520. }
  521. if (frame_length > pkvdb.max_transaction_size) return error.TransactionTooLarge;
  522. return @intCast(frame_length);
  523. }
  524. fn encodeTransaction(frame: []u8, operations: []const Operation, metadata: []const u8, lsn: u64, transaction_id: u64, timestamp_ns: i64) !void {
  525. @memset(frame, 0);
  526. var position: usize = pkvdb.transaction_header_size;
  527. @memcpy(frame[position .. position + metadata.len], metadata);
  528. position += metadata.len;
  529. for (operations) |operation| {
  530. var header: [16]u8 = undefined;
  531. pkvdb.encodeOperationHeader(.{ .opcode = operation.opcode, .key_length = @intCast(operation.key.len), .value_length = @intCast(operation.value.len) }, &header);
  532. @memcpy(frame[position .. position + header.len], &header);
  533. position += header.len;
  534. @memcpy(frame[position .. position + operation.key.len], operation.key);
  535. position += operation.key.len;
  536. @memcpy(frame[position .. position + operation.value.len], operation.value);
  537. position += operation.value.len;
  538. position = @intCast(try pkvdb.align8(position));
  539. }
  540. if (position != frame.len) return error.InvalidLength;
  541. const body = frame[pkvdb.transaction_header_size..];
  542. var tx_header: [56]u8 = undefined;
  543. pkvdb.encodeTransactionHeader(.{ .total_length = @intCast(frame.len), .lsn = lsn, .transaction_id = transaction_id, .timestamp_ns = timestamp_ns, .operation_count = @intCast(operations.len), .metadata_length = @intCast(metadata.len), .payload_crc = pkvdb.crc32c(body) }, &tx_header);
  544. @memcpy(frame[0..tx_header.len], &tx_header);
  545. }
  546. pub fn batchWrite(self: *Engine, operations: []const Operation, metadata: []const u8) !u64 {
  547. if (operations.len == 0 or operations.len > pkvdb.max_operations or metadata.len > pkvdb.max_transaction_size) return error.InvalidLength;
  548. for (operations) |operation| {
  549. if (operation.key.len > pkvdb.max_key_size or operation.value.len > pkvdb.max_value_size) return error.InvalidLength;
  550. if (operation.opcode == .delete and operation.value.len != 0) return error.InvalidLength;
  551. }
  552. const frame_length = try transactionLength(operations, metadata);
  553. var completion = WriteCompletion{ .remaining = 1 };
  554. var pending = PendingWrite{ .operations = operations, .metadata = metadata, .bytes = frame_length, .completion = &completion };
  555. try self.enqueueAndWait(&.{&pending}, &completion);
  556. if (completion.failure) |failure| return failure;
  557. return pending.lsn;
  558. }
  559. pub fn compareBatchWrite(self: *Engine, checks: []const CompareCheck, operations: []const Operation, metadata: []const u8) !CompareBatchResult {
  560. if (operations.len == 0 or operations.len > pkvdb.max_operations or metadata.len > pkvdb.max_transaction_size) return error.InvalidLength;
  561. if (checks.len > pkvdb.max_operations) return error.InvalidLength;
  562. for (operations) |operation| {
  563. if (operation.key.len > pkvdb.max_key_size or operation.value.len > pkvdb.max_value_size) return error.InvalidLength;
  564. if (operation.opcode == .delete and operation.value.len != 0) return error.InvalidLength;
  565. }
  566. for (checks) |check| {
  567. if (check.key.len > pkvdb.max_key_size) return error.InvalidLength;
  568. }
  569. const frame_length = try transactionLength(operations, metadata);
  570. var completion = WriteCompletion{ .remaining = 1 };
  571. var pending = PendingWrite{ .operations = operations, .metadata = metadata, .checks = checks, .bytes = frame_length, .completion = &completion };
  572. try self.enqueueAndWait(&.{&pending}, &completion);
  573. if (completion.failure) |failure| return failure;
  574. return .{ .lsn = pending.lsn, .committed = pending.committed };
  575. }
  576. pub fn putMany(self: *Engine, operations: []const Operation, lsns: []u64) !void {
  577. if (operations.len == 0 or operations.len != lsns.len or operations.len > max_group_transactions) return error.InvalidLength;
  578. const pending = try self.allocator.alloc(PendingWrite, operations.len);
  579. defer self.allocator.free(pending);
  580. const pointers = try self.allocator.alloc(*PendingWrite, operations.len);
  581. defer self.allocator.free(pointers);
  582. var completion = WriteCompletion{ .remaining = operations.len };
  583. for (operations, 0..) |operation, index| {
  584. if (operation.opcode != .put or operation.key.len > pkvdb.max_key_size or operation.value.len > pkvdb.max_value_size) return error.InvalidLength;
  585. const operation_slice = operations[index .. index + 1];
  586. pending[index] = .{ .operations = operation_slice, .metadata = "", .bytes = try transactionLength(operation_slice, ""), .completion = &completion };
  587. pointers[index] = &pending[index];
  588. }
  589. try self.enqueueAndWait(pointers, &completion);
  590. for (pending, 0..) |result, index| {
  591. if (completion.failure) |failure| return failure;
  592. lsns[index] = result.lsn;
  593. }
  594. }
  595. fn enqueueAndWait(self: *Engine, pending: []const *PendingWrite, completion: *WriteCompletion) !void {
  596. self.queue_mutex.lock();
  597. defer self.queue_mutex.unlock();
  598. if (self.writer_failed or self.writer_stopping) return error.StorageUnavailable;
  599. if (self.writer_thread == null) self.writer_thread = try std.Thread.spawn(.{}, writerMain, .{self});
  600. for (pending) |write| {
  601. while (self.queued_bytes > max_queued_bytes - write.bytes and !self.writer_failed and !self.writer_stopping) self.queue_condition.wait(&self.queue_mutex);
  602. if (self.writer_failed or self.writer_stopping) return error.StorageUnavailable;
  603. if (self.queue_tail) |tail| tail.next = write else self.queue_head = write;
  604. self.queue_tail = write;
  605. self.queued_bytes += write.bytes;
  606. }
  607. self.queue_condition.signal();
  608. while (completion.remaining != 0) completion.condition.wait(&self.queue_mutex);
  609. }
  610. fn writerMain(self: *Engine) void {
  611. var group: [max_group_transactions]*PendingWrite = undefined;
  612. while (true) {
  613. self.queue_mutex.lock();
  614. while (self.queue_head == null and !self.writer_stopping) self.queue_condition.wait(&self.queue_mutex);
  615. if (self.queue_head == null and self.writer_stopping) {
  616. self.queue_mutex.unlock();
  617. return;
  618. }
  619. // Conditional transactions cannot share this commit group, so
  620. // waiting for more queue entries cannot amortize their fsync.
  621. if (canCoalesce(self.queue_head.?)) {
  622. _ = self.queue_condition.timedWait(&self.queue_mutex, group_wait_ns) catch {};
  623. }
  624. var count: usize = 0;
  625. var bytes: usize = pkvdb.group_header_size;
  626. var conditional: bool = false;
  627. while (self.queue_head) |pending| {
  628. if (count != 0 and (count == group.len or bytes + pending.bytes > max_group_bytes or conditional or pending.checks.len != 0)) break;
  629. self.queue_head = pending.next;
  630. if (self.queue_head == null) self.queue_tail = null;
  631. pending.next = null;
  632. group[count] = pending;
  633. count += 1;
  634. bytes += pending.bytes;
  635. self.queued_bytes -= pending.bytes;
  636. conditional = pending.checks.len != 0;
  637. }
  638. self.queue_condition.broadcast();
  639. self.queue_mutex.unlock();
  640. self.processGroup(group[0..count], bytes) catch |failure| {
  641. self.queue_mutex.lock();
  642. self.writer_failed = true;
  643. for (group[0..count]) |pending| {
  644. pending.completion.failure = failure;
  645. pending.completion.remaining -= 1;
  646. if (pending.completion.remaining == 0) pending.completion.condition.signal();
  647. }
  648. while (self.queue_head) |pending| {
  649. self.queue_head = pending.next;
  650. pending.completion.failure = error.StorageUnavailable;
  651. pending.completion.remaining -= 1;
  652. if (pending.completion.remaining == 0) pending.completion.condition.signal();
  653. }
  654. self.queue_tail = null;
  655. self.queued_bytes = 0;
  656. self.queue_condition.broadcast();
  657. self.queue_mutex.unlock();
  658. continue;
  659. };
  660. self.queue_mutex.lock();
  661. for (group[0..count]) |pending| {
  662. pending.completion.remaining -= 1;
  663. if (pending.completion.remaining == 0) pending.completion.condition.signal();
  664. }
  665. self.queue_mutex.unlock();
  666. }
  667. }
  668. fn canCoalesce(pending: *const PendingWrite) bool {
  669. return pending.checks.len == 0;
  670. }
  671. fn checkConditions(self: *Engine, checks: []const CompareCheck) !bool {
  672. for (checks) |check| {
  673. const record = try self.directory.getWithReader(self.keyReader(), check.key);
  674. if (check.expected_lsn == 0) {
  675. if (record != null) return false;
  676. } else if (record == null or record.?.lsn != check.expected_lsn) {
  677. return false;
  678. }
  679. }
  680. return true;
  681. }
  682. fn processGroup(self: *Engine, group: []*PendingWrite, payload_length: usize) !void {
  683. var operation_count: usize = 0;
  684. for (group) |pending| operation_count = try std.math.add(usize, operation_count, pending.operations.len);
  685. const payload = try self.allocator.alloc(u8, payload_length);
  686. defer self.allocator.free(payload);
  687. self.ordered_gate.lock();
  688. defer self.ordered_gate.unlock();
  689. self.lock.lock();
  690. if (group[0].checks.len != 0) {
  691. if (group.len != 1) {
  692. self.lock.unlock();
  693. return error.InvalidConditionalGroup;
  694. }
  695. const conditions_match = self.checkConditions(group[0].checks) catch |failure| {
  696. self.lock.unlock();
  697. return failure;
  698. };
  699. if (!conditions_match) {
  700. group[0].committed = false;
  701. group[0].lsn = 0;
  702. self.lock.unlock();
  703. return;
  704. }
  705. }
  706. self.directory.ensureAdditional(self.file, operation_count) catch |failure| {
  707. self.lock.unlock();
  708. return failure;
  709. };
  710. const first_lsn = std.math.add(u64, self.latest_lsn, 1) catch |failure| {
  711. self.lock.unlock();
  712. return failure;
  713. };
  714. const prepare_ordered = self.ordered_ready;
  715. self.lock.unlock();
  716. var prepared_storage: []ordered_index.Prepared = &.{};
  717. if (prepare_ordered) prepared_storage = try self.allocator.alloc(ordered_index.Prepared, operation_count);
  718. var prepared_count: usize = 0;
  719. defer {
  720. for (prepared_storage[0..prepared_count]) |*entry| self.ordered.discard(entry);
  721. if (prepared_storage.len != 0) self.allocator.free(prepared_storage);
  722. }
  723. if (prepare_ordered) for (group) |pending| {
  724. pending.prepared_position = prepared_count;
  725. for (pending.operations) |operation| if (operation.opcode == .put) {
  726. prepared_storage[prepared_count] = try self.ordered.prepare(operation.key);
  727. prepared_count += 1;
  728. };
  729. pending.prepared_count = prepared_count - pending.prepared_position;
  730. };
  731. @memset(payload, 0);
  732. const group_timestamp = now();
  733. writeInt(u16, payload, 0, 1);
  734. writeInt(u32, payload, 4, @intCast(group.len));
  735. writeInt(i64, payload, 8, group_timestamp);
  736. var position: usize = pkvdb.group_header_size;
  737. var lsn = first_lsn;
  738. for (group, 0..) |pending, index| {
  739. pending.frame_position = position;
  740. const timestamp = std.math.add(i64, group_timestamp, @intCast(index)) catch std.math.maxInt(i64);
  741. try encodeTransaction(payload[position .. position + pending.bytes], pending.operations, pending.metadata, lsn, lsn, timestamp);
  742. pending.lsn = lsn;
  743. position += pending.bytes;
  744. lsn = try std.math.add(u64, lsn, 1);
  745. }
  746. const last_lsn = lsn - 1;
  747. self.io_mutex.lock();
  748. defer self.io_mutex.unlock();
  749. const offset = try self.appendExtent(.journal, first_lsn, last_lsn, payload);
  750. self.file.sync() catch |failure| {
  751. self.file.setEndPos(offset) catch {};
  752. self.file_length = offset;
  753. return failure;
  754. };
  755. self.mapTail(false);
  756. self.lock.lock();
  757. defer self.lock.unlock();
  758. for (group) |pending| {
  759. try self.publishPending(offset + pkvdb.extent_header_size, pending, prepared_storage[0..prepared_count]);
  760. }
  761. self.journal_bytes_since_checkpoint += self.file_length - offset;
  762. self.commit_groups += 1;
  763. self.committed_transactions += group.len;
  764. self.largest_commit_group = @max(self.largest_commit_group, group.len);
  765. }
  766. fn publishPending(self: *Engine, extent_payload_offset: u64, pending: *PendingWrite, prepared: []ordered_index.Prepared) !void {
  767. var position: usize = pkvdb.transaction_header_size + pending.metadata.len;
  768. var prepared_position = pending.prepared_position;
  769. for (pending.operations) |operation| {
  770. const key_offset = try std.math.add(u64, extent_payload_offset, pending.frame_position + position + pkvdb.operation_header_size);
  771. switch (operation.opcode) {
  772. .put => {
  773. const record = RecordRef{ .hash = keydir.KeyDir.hash(operation.key), .lsn = pending.lsn, .key_offset = key_offset, .value_offset = key_offset + operation.key.len, .key_len = @intCast(operation.key.len), .value_len = @intCast(operation.value.len) };
  774. if (self.ordered_ready) {
  775. self.ordered.putPrepared(&prepared[prepared_position], record);
  776. prepared_position += 1;
  777. }
  778. try self.directory.putWithKey(self.file, operation.key, record);
  779. },
  780. .delete => pending.changed = (try self.removeRecord(operation.key)) or pending.changed,
  781. }
  782. position = @intCast(try pkvdb.align8(position + pkvdb.operation_header_size + operation.key.len + operation.value.len));
  783. }
  784. if (prepared_position != pending.prepared_position + pending.prepared_count) return error.IndexInconsistent;
  785. if (position != pending.bytes) return error.InvalidLength;
  786. self.latest_lsn = pending.lsn;
  787. if (self.oldest_lsn == 0) self.oldest_lsn = pending.lsn;
  788. }
  789. pub fn importBaseline(self: *Engine, operations: []const Operation) !void {
  790. if (operations.len > pkvdb.max_operations) return error.InvalidLength;
  791. self.io_mutex.lock();
  792. defer self.io_mutex.unlock();
  793. self.lock.lock();
  794. defer self.lock.unlock();
  795. if (self.checkpoint_lsn != 0 or self.latest_lsn > 1 or self.directory.count != 0 and self.latest_lsn == 0) return error.InvalidState;
  796. var length: u64 = 24;
  797. for (operations) |operation| {
  798. if (operation.opcode != .put or operation.key.len > pkvdb.max_key_size or operation.value.len > pkvdb.max_value_size) return error.InvalidLength;
  799. length = try pkvdb.align8(try std.math.add(u64, length, 8 + operation.key.len + operation.value.len));
  800. }
  801. if (length > pkvdb.max_transaction_size) return error.TransactionTooLarge;
  802. const payload = try self.allocator.alloc(u8, @intCast(length));
  803. defer self.allocator.free(payload);
  804. @memset(payload, 0);
  805. writeInt(u16, payload, 0, 1);
  806. writeInt(u32, payload, 4, @intCast(operations.len));
  807. writeInt(u64, payload, 8, 1);
  808. writeInt(i64, payload, 16, now());
  809. var position: usize = 24;
  810. for (operations) |operation| {
  811. writeInt(u32, payload, position, @intCast(operation.key.len));
  812. writeInt(u32, payload, position + 4, @intCast(operation.value.len));
  813. position += 8;
  814. @memcpy(payload[position .. position + operation.key.len], operation.key);
  815. position += operation.key.len;
  816. @memcpy(payload[position .. position + operation.value.len], operation.value);
  817. position += operation.value.len;
  818. const aligned: usize = @intCast(try pkvdb.align8(position));
  819. @memset(payload[position..aligned], 0);
  820. position = aligned;
  821. }
  822. try self.directory.ensureAdditional(self.file, operations.len);
  823. const offset = try self.appendExtent(.store_metadata, 1, 1, payload);
  824. try self.file.sync();
  825. try self.replayBaseline(offset, payload);
  826. }
  827. pub fn put(self: *Engine, key: []const u8, value: []const u8) !u64 {
  828. return self.batchWrite(&.{.{ .opcode = .put, .key = key, .value = value }}, "");
  829. }
  830. pub fn delete(self: *Engine, key: []const u8) !bool {
  831. if (key.len > pkvdb.max_key_size) return error.InvalidLength;
  832. const operations = [_]Operation{.{ .opcode = .delete, .key = key }};
  833. var completion = WriteCompletion{ .remaining = 1 };
  834. var pending = PendingWrite{ .operations = &operations, .metadata = "", .bytes = try transactionLength(&operations, ""), .completion = &completion };
  835. try self.enqueueAndWait(&.{&pending}, &completion);
  836. if (completion.failure) |failure| return failure;
  837. return pending.changed;
  838. }
  839. pub fn get(self: *Engine, allocator: std.mem.Allocator, key: []const u8) !?Value {
  840. const record = try self.getRef(key) orelse return null;
  841. const bytes = try allocator.alloc(u8, record.value_len);
  842. errdefer allocator.free(bytes);
  843. _ = try self.readValue(record, bytes, 0);
  844. return .{ .bytes = bytes, .lsn = record.lsn };
  845. }
  846. pub fn getRef(self: *Engine, key: []const u8) !?RecordRef {
  847. if (key.len > pkvdb.max_key_size) return error.InvalidLength;
  848. self.lock.lockShared();
  849. defer self.lock.unlockShared();
  850. return try self.directory.getWithReader(self.keyReader(), key);
  851. }
  852. pub fn readValue(self: *Engine, record: RecordRef, destination: []u8, value_position: u32) !usize {
  853. if (value_position > record.value_len) return error.InvalidOffset;
  854. const amount = @min(destination.len, record.value_len - value_position);
  855. const file_offset = record.value_offset + value_position;
  856. const got = try self.readBytes(destination[0..amount], file_offset);
  857. if (got != amount) return error.Truncated;
  858. return amount;
  859. }
  860. pub fn exists(self: *Engine, key: []const u8) !bool {
  861. self.lock.lockShared();
  862. defer self.lock.unlockShared();
  863. return (try self.directory.getWithReader(self.keyReader(), key)) != null;
  864. }
  865. pub fn multiGet(self: *Engine, allocator: std.mem.Allocator, keys: []const []const u8) ![]?Value {
  866. if (keys.len > pkvdb.max_operations) return error.InvalidLength;
  867. const refs = try allocator.alloc(?RecordRef, keys.len);
  868. defer allocator.free(refs);
  869. @memset(refs, null);
  870. self.lock.lockShared();
  871. for (keys, 0..) |key, index| {
  872. if (key.len > pkvdb.max_key_size) {
  873. self.lock.unlockShared();
  874. return error.InvalidLength;
  875. }
  876. refs[index] = self.directory.getWithReader(self.keyReader(), key) catch |failure| {
  877. self.lock.unlockShared();
  878. return failure;
  879. };
  880. }
  881. self.lock.unlockShared();
  882. const values = try allocator.alloc(?Value, keys.len);
  883. errdefer allocator.free(values);
  884. @memset(values, null);
  885. errdefer for (values) |value| if (value) |present| allocator.free(present.bytes);
  886. for (refs, 0..) |entry, index| {
  887. const record = entry orelse continue;
  888. const bytes = try allocator.alloc(u8, record.value_len);
  889. errdefer allocator.free(bytes);
  890. _ = try self.readValue(record, bytes, 0);
  891. values[index] = .{ .bytes = bytes, .lsn = record.lsn };
  892. }
  893. return values;
  894. }
  895. pub fn scan(self: *Engine, allocator: std.mem.Allocator, prefix: []const u8, cursor: []const u8, limit: u32, include_values: bool, max_bytes: u32) !ScanBatch {
  896. if (prefix.len > pkvdb.max_key_size or cursor.len > pkvdb.max_key_size or limit == 0 or limit > 4096 or max_bytes == 0 or max_bytes > pkvdb.max_key_size + pkvdb.max_value_size + 1024) return error.InvalidLength;
  897. try self.ensureOrdered();
  898. self.lock.lockShared();
  899. defer self.lock.unlockShared();
  900. var node = self.ordered.lowerBound(if (cursor.len == 0) prefix else cursor);
  901. if (cursor.len != 0 and node != null and std.mem.eql(u8, node.?.key, cursor)) node = ordered_index.OrderedIndex.next(node.?);
  902. var entries = std.ArrayListUnmanaged(ScanEntry){};
  903. errdefer {
  904. for (entries.items) |entry| {
  905. allocator.free(entry.key);
  906. if (entry.value) |value| allocator.free(value);
  907. }
  908. entries.deinit(allocator);
  909. }
  910. var bytes_used: usize = 0;
  911. while (node) |current| {
  912. if (entries.items.len >= limit or !std.mem.startsWith(u8, current.key, prefix)) break;
  913. const record = current.record;
  914. const next_size = 16 + current.key.len + if (include_values) record.value_len else 0;
  915. if (next_size > max_bytes) return error.ScanEntryTooLarge;
  916. if (bytes_used + next_size > max_bytes) break;
  917. const key = try allocator.dupe(u8, current.key);
  918. errdefer allocator.free(key);
  919. var value: ?[]u8 = null;
  920. if (include_values) {
  921. value = try allocator.alloc(u8, record.value_len);
  922. errdefer allocator.free(value.?);
  923. _ = try self.readValue(record, value.?, 0);
  924. }
  925. try entries.append(allocator, .{ .key = key, .value = value, .lsn = record.lsn });
  926. bytes_used += next_size;
  927. node = ordered_index.OrderedIndex.next(current);
  928. }
  929. const next_cursor = if (entries.items.len == 0) try allocator.alloc(u8, 0) else try allocator.dupe(u8, entries.items[entries.items.len - 1].key);
  930. errdefer allocator.free(next_cursor);
  931. const done = node == null or !std.mem.startsWith(u8, node.?.key, prefix);
  932. return .{ .entries = try entries.toOwnedSlice(allocator), .next_cursor = next_cursor, .done = done };
  933. }
  934. fn ensureOrdered(self: *Engine) !void {
  935. self.lock.lockShared();
  936. const ready = self.ordered_ready;
  937. self.lock.unlockShared();
  938. if (ready) return;
  939. self.ordered_gate.lock();
  940. defer self.ordered_gate.unlock();
  941. self.lock.lock();
  942. defer self.lock.unlock();
  943. if (self.ordered_ready) return;
  944. const records = try self.directory.records(self.allocator);
  945. defer self.allocator.free(records);
  946. for (records) |record| {
  947. const key = try self.allocator.alloc(u8, record.key_len);
  948. defer self.allocator.free(key);
  949. if (try self.file.preadAll(key, record.key_offset) != key.len) return error.Truncated;
  950. try self.ordered.put(key, record);
  951. }
  952. self.ordered_ready = true;
  953. }
  954. pub fn checkpoint(self: *Engine) !void {
  955. const started = std.time.nanoTimestamp();
  956. self.checkpoint_mutex.lock();
  957. defer self.checkpoint_mutex.unlock();
  958. self.io_mutex.lock();
  959. self.lock.lockShared();
  960. const records = self.directory.records(self.allocator) catch |err| {
  961. self.lock.unlockShared();
  962. self.io_mutex.unlock();
  963. return err;
  964. };
  965. const lsn = self.latest_lsn;
  966. const replay_offset = self.file_length;
  967. self.lock.unlockShared();
  968. self.io_mutex.unlock();
  969. defer self.allocator.free(records);
  970. const payload_length = try std.math.add(usize, pkvdb.checkpoint_header_size, try std.math.mul(usize, records.len, pkvdb.checkpoint_entry_size));
  971. const payload = try self.allocator.alloc(u8, payload_length);
  972. defer self.allocator.free(payload);
  973. var header: [56]u8 = undefined;
  974. pkvdb.encodeCheckpointHeader(.{ .lsn = lsn, .timestamp_ns = now(), .entry_count = records.len, .source_start = pkvdb.data_offset, .source_end = replay_offset }, &header);
  975. @memcpy(payload[0..header.len], &header);
  976. for (records, 0..) |record, index| {
  977. if (record.flags > std.math.maxInt(u16)) return error.InvalidFlags;
  978. var entry: [48]u8 = undefined;
  979. pkvdb.encodeCheckpointEntry(.{ .hash = record.hash, .lsn = record.lsn, .key_offset = record.key_offset, .value_offset = record.value_offset, .key_len = record.key_len, .value_len = record.value_len, .flags = @intCast(record.flags) }, &entry);
  980. @memcpy(payload[pkvdb.checkpoint_header_size + index * pkvdb.checkpoint_entry_size ..][0..pkvdb.checkpoint_entry_size], &entry);
  981. }
  982. self.io_mutex.lock();
  983. defer self.io_mutex.unlock();
  984. self.lock.lockShared();
  985. const known_lsn = self.latest_lsn;
  986. const next_generation = std.math.add(u64, self.generation, 1) catch |failure| {
  987. self.lock.unlockShared();
  988. return failure;
  989. };
  990. const history_start_lsn = self.oldest_lsn;
  991. self.lock.unlockShared();
  992. const entries_offset = try self.appendExtent(.checkpoint_entries, lsn, lsn, payload);
  993. const manifest_offset = try pkvdb.align8(self.file_length);
  994. const known_tail = try extentEnd(manifest_offset, pkvdb.manifest_size);
  995. var manifest_bytes: [104]u8 = undefined;
  996. pkvdb.encodeManifest(.{ .uuid = self.uuid, .generation = next_generation, .checkpoint_lsn = lsn, .entries_offset = entries_offset, .ordered_offset = 0, .replay_offset = replay_offset, .known_tail = known_tail, .known_lsn = known_lsn, .history_start_lsn = history_start_lsn }, &manifest_bytes);
  997. const actual_manifest = try self.appendExtent(.manifest, lsn, known_lsn, &manifest_bytes);
  998. if (actual_manifest != manifest_offset or self.file_length != known_tail) return error.InvalidManifest;
  999. try self.file.sync();
  1000. const inactive: u1 = self.active_superblock ^ 1;
  1001. var block: [4096]u8 = undefined;
  1002. pkvdb.encodeSuperblock(.{ .generation = next_generation, .uuid = self.uuid, .manifest_offset = manifest_offset, .checkpoint_lsn = lsn, .known_lsn = known_lsn, .known_file_length = known_tail, .created_ns = self.created_ns, .updated_ns = now() }, &block);
  1003. try self.file.pwriteAll(&block, @as(u64, inactive) * pkvdb.superblock_size);
  1004. try self.file.sync();
  1005. self.lock.lock();
  1006. self.active_superblock = inactive;
  1007. self.generation = next_generation;
  1008. self.checkpoint_lsn = lsn;
  1009. self.journal_bytes_since_checkpoint = self.file_length - replay_offset;
  1010. const elapsed = std.time.nanoTimestamp() - started;
  1011. self.checkpoint_ns = if (elapsed > 0) @intCast(elapsed) else 0;
  1012. self.lock.unlock();
  1013. }
  1014. pub fn status(self: *Engine) Status {
  1015. self.io_mutex.lock();
  1016. defer self.io_mutex.unlock();
  1017. self.lock.lockShared();
  1018. defer self.lock.unlockShared();
  1019. return .{
  1020. .uuid = self.uuid,
  1021. .file_bytes = self.file_length,
  1022. .latest_lsn = self.latest_lsn,
  1023. .oldest_lsn = self.oldest_lsn,
  1024. .checkpoint_lsn = self.checkpoint_lsn,
  1025. .journal_bytes_since_checkpoint = self.journal_bytes_since_checkpoint,
  1026. .live_keys = self.directory.count,
  1027. .keydir_bytes = self.directory.bytes(),
  1028. .ordered_index_bytes = self.ordered.allocated_bytes,
  1029. .bytes_written = self.bytes_written,
  1030. .checksum_failures = self.checksum_failures,
  1031. .partial_tails = self.partial_tails,
  1032. .recovery_ns = self.recovery_ns,
  1033. .checkpoint_ns = self.checkpoint_ns,
  1034. .connection_bytes = self.connection_bytes.load(.monotonic),
  1035. .active_requests = self.active_requests.load(.monotonic),
  1036. .commit_groups = self.commit_groups,
  1037. .committed_transactions = self.committed_transactions,
  1038. .largest_commit_group = self.largest_commit_group,
  1039. };
  1040. }
  1041. pub fn addConnectionBytes(self: *Engine, amount: u64) void {
  1042. _ = self.connection_bytes.fetchAdd(amount, .monotonic);
  1043. }
  1044. pub fn removeConnectionBytes(self: *Engine, amount: u64) void {
  1045. _ = self.connection_bytes.fetchSub(amount, .monotonic);
  1046. }
  1047. pub fn beginRequest(self: *Engine) void {
  1048. _ = self.active_requests.fetchAdd(1, .monotonic);
  1049. }
  1050. pub fn endRequest(self: *Engine) void {
  1051. _ = self.active_requests.fetchSub(1, .monotonic);
  1052. }
  1053. };
  1054. fn readInt(comptime T: type, bytes: []const u8, offset: usize) T {
  1055. return std.mem.readInt(T, bytes[offset..][0..@sizeOf(T)], .little);
  1056. }
  1057. fn writeInt(comptime T: type, bytes: []u8, offset: usize, value: T) void {
  1058. std.mem.writeInt(T, bytes[offset..][0..@sizeOf(T)], value, .little);
  1059. }
  1060. test "insert overwrite delete recreate and recovery" {
  1061. var tmp = std.testing.tmpDir(.{});
  1062. defer tmp.cleanup();
  1063. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1064. const directory = try tmp.dir.realpath(".", &path_buffer);
  1065. const path = try std.fmt.allocPrint(std.testing.allocator, "{s}/test.pkvdb", .{directory});
  1066. defer std.testing.allocator.free(path);
  1067. var engine = try Engine.open(std.testing.allocator, path);
  1068. _ = try engine.put("key", "one");
  1069. _ = try engine.put("key", "two");
  1070. try std.testing.expect(try engine.delete("key"));
  1071. _ = try engine.put("key", "three");
  1072. var value = (try engine.get(std.testing.allocator, "key")).?;
  1073. try std.testing.expectEqualStrings("three", value.bytes);
  1074. std.testing.allocator.free(value.bytes);
  1075. engine.close();
  1076. engine = try Engine.open(std.testing.allocator, path);
  1077. defer engine.close();
  1078. value = (try engine.get(std.testing.allocator, "key")).?;
  1079. defer std.testing.allocator.free(value.bytes);
  1080. try std.testing.expectEqualStrings("three", value.bytes);
  1081. try std.testing.expectEqual(@as(u64, 4), value.lsn);
  1082. }
  1083. test "atomic batch checkpoint tail and ordered scan" {
  1084. var tmp = std.testing.tmpDir(.{});
  1085. defer tmp.cleanup();
  1086. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1087. const directory = try tmp.dir.realpath(".", &path_buffer);
  1088. const path = try std.fmt.allocPrint(std.testing.allocator, "{s}/test.pkvdb", .{directory});
  1089. defer std.testing.allocator.free(path);
  1090. var engine = try Engine.open(std.testing.allocator, path);
  1091. _ = try engine.batchWrite(&.{ .{ .opcode = .put, .key = "p/2", .value = "b" }, .{ .opcode = .put, .key = "p/1", .value = "a" } }, "meta");
  1092. try engine.checkpoint();
  1093. _ = try engine.put("p/3", "c");
  1094. engine.close();
  1095. engine = try Engine.open(std.testing.allocator, path);
  1096. defer engine.close();
  1097. var batch = try engine.scan(std.testing.allocator, "p/", "", 2, true, 1024);
  1098. try std.testing.expectEqual(@as(usize, 2), batch.entries.len);
  1099. try std.testing.expectEqualStrings("p/1", batch.entries[0].key);
  1100. const cursor = try std.testing.allocator.dupe(u8, batch.next_cursor);
  1101. batch.deinit(std.testing.allocator);
  1102. defer std.testing.allocator.free(cursor);
  1103. batch = try engine.scan(std.testing.allocator, "p/", cursor, 2, true, 1024);
  1104. defer batch.deinit(std.testing.allocator);
  1105. try std.testing.expectEqual(@as(usize, 1), batch.entries.len);
  1106. try std.testing.expectEqualStrings("p/3", batch.entries[0].key);
  1107. }
  1108. test "scan supports full-size PKBFI pages and accounts for entry framing" {
  1109. var tmp = std.testing.tmpDir(.{});
  1110. defer tmp.cleanup();
  1111. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1112. const path = try testPath(&tmp, "large-scan.pkvdb", &path_buffer);
  1113. defer std.testing.allocator.free(path);
  1114. var engine = try Engine.open(std.testing.allocator, path);
  1115. defer engine.close();
  1116. const value = try std.testing.allocator.alloc(u8, 2 * 1024 * 1024);
  1117. defer std.testing.allocator.free(value);
  1118. @memset(value, 'x');
  1119. _ = try engine.put("large/key", value);
  1120. const entry_size = 16 + "large/key".len + value.len;
  1121. var batch = try engine.scan(std.testing.allocator, "large/", "", 1, true, @intCast(entry_size));
  1122. defer batch.deinit(std.testing.allocator);
  1123. try std.testing.expectEqual(@as(usize, 1), batch.entries.len);
  1124. try std.testing.expectEqual(value.len, batch.entries[0].value.?.len);
  1125. try std.testing.expectError(
  1126. error.ScanEntryTooLarge,
  1127. engine.scan(std.testing.allocator, "large/", "", 1, true, @intCast(entry_size - 1)),
  1128. );
  1129. }
  1130. fn testPath(tmp: *std.testing.TmpDir, name: []const u8, buffer: *[std.fs.max_path_bytes]u8) ![]const u8 {
  1131. const directory = try tmp.dir.realpath(".", buffer);
  1132. return std.fs.path.join(std.testing.allocator, &.{ directory, name });
  1133. }
  1134. test "partial final extent is ignored and removed" {
  1135. var tmp = std.testing.tmpDir(.{});
  1136. defer tmp.cleanup();
  1137. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1138. const path = try testPath(&tmp, "partial.pkvdb", &path_buffer);
  1139. defer std.testing.allocator.free(path);
  1140. var engine = try Engine.open(std.testing.allocator, path);
  1141. _ = try engine.put("durable", "value");
  1142. const valid_length = engine.status().file_bytes;
  1143. engine.close();
  1144. const file = try std.fs.cwd().openFile(path, .{ .mode = .read_write });
  1145. var header: [64]u8 = undefined;
  1146. pkvdb.encodeExtentHeader(.{ .extent_type = .journal, .payload_length = 100, .first_lsn = 2, .last_lsn = 2, .payload_crc = 0 }, &header);
  1147. try file.pwriteAll(&header, valid_length);
  1148. try file.pwriteAll("partial transaction", valid_length + header.len);
  1149. file.close();
  1150. engine = try Engine.open(std.testing.allocator, path);
  1151. defer engine.close();
  1152. const value = (try engine.get(std.testing.allocator, "durable")).?;
  1153. defer std.testing.allocator.free(value.bytes);
  1154. try std.testing.expectEqualStrings("value", value.bytes);
  1155. try std.testing.expectEqual(valid_length, engine.status().file_bytes);
  1156. try std.testing.expectEqual(@as(u64, 1), engine.status().partial_tails);
  1157. }
  1158. test "corruption in committed journal is explicit" {
  1159. var tmp = std.testing.tmpDir(.{});
  1160. defer tmp.cleanup();
  1161. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1162. const path = try testPath(&tmp, "corrupt.pkvdb", &path_buffer);
  1163. defer std.testing.allocator.free(path);
  1164. var engine = try Engine.open(std.testing.allocator, path);
  1165. _ = try engine.put("key", "value");
  1166. engine.close();
  1167. const file = try std.fs.cwd().openFile(path, .{ .mode = .read_write });
  1168. var byte: [1]u8 = undefined;
  1169. const offset = pkvdb.data_offset + pkvdb.extent_header_size + pkvdb.group_header_size + pkvdb.transaction_header_size + pkvdb.operation_header_size + 3;
  1170. _ = try file.preadAll(&byte, offset);
  1171. byte[0] ^= 1;
  1172. try file.pwriteAll(&byte, offset);
  1173. file.close();
  1174. try std.testing.expectError(error.ChecksumMismatch, Engine.open(std.testing.allocator, path));
  1175. }
  1176. test "one corrupted superblock falls back and adopts tail" {
  1177. var tmp = std.testing.tmpDir(.{});
  1178. defer tmp.cleanup();
  1179. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1180. const path = try testPath(&tmp, "root.pkvdb", &path_buffer);
  1181. defer std.testing.allocator.free(path);
  1182. var engine = try Engine.open(std.testing.allocator, path);
  1183. _ = try engine.put("before", "one");
  1184. try engine.checkpoint();
  1185. _ = try engine.put("after", "two");
  1186. engine.close();
  1187. const file = try std.fs.cwd().openFile(path, .{ .mode = .read_write });
  1188. var byte: [1]u8 = undefined;
  1189. _ = try file.preadAll(&byte, pkvdb.superblock_size + 24);
  1190. byte[0] ^= 1;
  1191. try file.pwriteAll(&byte, pkvdb.superblock_size + 24);
  1192. file.close();
  1193. engine = try Engine.open(std.testing.allocator, path);
  1194. defer engine.close();
  1195. const value = (try engine.get(std.testing.allocator, "after")).?;
  1196. defer std.testing.allocator.free(value.bytes);
  1197. try std.testing.expectEqualStrings("two", value.bytes);
  1198. try std.testing.expectEqual(@as(u64, 2), engine.status().latest_lsn);
  1199. }
  1200. test "unrooted checkpoint and manifest do not hide journal" {
  1201. var tmp = std.testing.tmpDir(.{});
  1202. defer tmp.cleanup();
  1203. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1204. const path = try testPath(&tmp, "interrupted-checkpoint.pkvdb", &path_buffer);
  1205. defer std.testing.allocator.free(path);
  1206. var engine = try Engine.open(std.testing.allocator, path);
  1207. _ = try engine.put("key", "value");
  1208. var roots: [8192]u8 = undefined;
  1209. _ = try engine.file.preadAll(&roots, 0);
  1210. try engine.checkpoint();
  1211. engine.close();
  1212. const file = try std.fs.cwd().openFile(path, .{ .mode = .read_write });
  1213. try file.pwriteAll(&roots, 0);
  1214. file.close();
  1215. engine = try Engine.open(std.testing.allocator, path);
  1216. defer engine.close();
  1217. const value = (try engine.get(std.testing.allocator, "key")).?;
  1218. defer std.testing.allocator.free(value.bytes);
  1219. try std.testing.expectEqualStrings("value", value.bytes);
  1220. try std.testing.expectEqual(@as(u64, 1), engine.status().latest_lsn);
  1221. }
  1222. test "unknown compatible extent is skipped by length" {
  1223. var tmp = std.testing.tmpDir(.{});
  1224. defer tmp.cleanup();
  1225. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1226. const path = try testPath(&tmp, "unknown.pkvdb", &path_buffer);
  1227. defer std.testing.allocator.free(path);
  1228. var engine = try Engine.open(std.testing.allocator, path);
  1229. _ = try engine.put("key", "value");
  1230. const offset = engine.status().file_bytes;
  1231. engine.close();
  1232. const file = try std.fs.cwd().openFile(path, .{ .mode = .read_write });
  1233. var header: [64]u8 = undefined;
  1234. pkvdb.encodeExtentHeader(.{ .extent_type = @enumFromInt(99), .version = 9, .payload_length = 3, .first_lsn = 0, .last_lsn = 0, .payload_crc = pkvdb.crc32c("new") }, &header);
  1235. try file.pwriteAll(&header, offset);
  1236. try file.pwriteAll("new", offset + header.len);
  1237. try file.pwriteAll(&([_]u8{0} ** 5), offset + header.len + 3);
  1238. file.close();
  1239. engine = try Engine.open(std.testing.allocator, path);
  1240. defer engine.close();
  1241. try std.testing.expect(try engine.exists("key"));
  1242. }
  1243. test "repeated overwrite and delete keep directory memory bounded" {
  1244. var tmp = std.testing.tmpDir(.{});
  1245. defer tmp.cleanup();
  1246. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1247. const path = try testPath(&tmp, "memory.pkvdb", &path_buffer);
  1248. defer std.testing.allocator.free(path);
  1249. var engine = try Engine.open(std.testing.allocator, path);
  1250. defer engine.close();
  1251. _ = try engine.put("same", "first");
  1252. const initial = engine.status().keydir_bytes;
  1253. for (0..100) |index| {
  1254. var value: [16]u8 = undefined;
  1255. const encoded = try std.fmt.bufPrint(&value, "{d}", .{index});
  1256. _ = try engine.put("same", encoded);
  1257. }
  1258. try std.testing.expect(try engine.delete("same"));
  1259. _ = try engine.put("same", "last");
  1260. try std.testing.expectEqual(initial, engine.status().keydir_bytes);
  1261. try std.testing.expectEqual(@as(u64, 1), engine.status().live_keys);
  1262. }
  1263. test "concurrent reads and writes preserve complete values" {
  1264. var tmp = std.testing.tmpDir(.{});
  1265. defer tmp.cleanup();
  1266. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1267. const path = try testPath(&tmp, "concurrent.pkvdb", &path_buffer);
  1268. defer std.testing.allocator.free(path);
  1269. var engine = try Engine.open(std.testing.allocator, path);
  1270. defer engine.close();
  1271. _ = try engine.put("key", "00000000");
  1272. var failed = std.atomic.Value(bool).init(false);
  1273. const Writer = struct {
  1274. fn run(target: *Engine, failure: *std.atomic.Value(bool)) void {
  1275. for (0..50) |index| {
  1276. var value: [8]u8 = undefined;
  1277. _ = std.fmt.bufPrint(&value, "{d:0>8}", .{index}) catch {
  1278. failure.store(true, .release);
  1279. return;
  1280. };
  1281. _ = target.put("key", &value) catch {
  1282. failure.store(true, .release);
  1283. return;
  1284. };
  1285. }
  1286. }
  1287. };
  1288. const thread = try std.Thread.spawn(.{}, Writer.run, .{ &engine, &failed });
  1289. for (0..50) |_| {
  1290. const value = try engine.get(std.testing.allocator, "key") orelse return error.TestUnexpectedResult;
  1291. try std.testing.expectEqual(@as(usize, 8), value.bytes.len);
  1292. std.testing.allocator.free(value.bytes);
  1293. }
  1294. thread.join();
  1295. try std.testing.expect(!failed.load(.acquire));
  1296. }
  1297. fn copyPrefix(source_path: []const u8, destination_path: []const u8) !void {
  1298. const source = try std.fs.cwd().openFile(source_path, .{ .mode = .read_only });
  1299. defer source.close();
  1300. const length = try source.getEndPos();
  1301. const destination = try std.fs.cwd().createFile(destination_path, .{ .read = true, .truncate = true });
  1302. defer destination.close();
  1303. var buffer: [4096]u8 = undefined;
  1304. var offset: u64 = 0;
  1305. while (offset < length) {
  1306. const amount: usize = @intCast(@min(buffer.len, length - offset));
  1307. const got = try source.preadAll(buffer[0..amount], offset);
  1308. if (got == 0) break;
  1309. try destination.writeAll(buffer[0..got]);
  1310. offset += got;
  1311. }
  1312. }
  1313. test "every live copy recovers while writes continue" {
  1314. var tmp = std.testing.tmpDir(.{});
  1315. defer tmp.cleanup();
  1316. var source_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1317. const source = try testPath(&tmp, "live.pkvdb", &source_buffer);
  1318. defer std.testing.allocator.free(source);
  1319. var engine = try Engine.open(std.testing.allocator, source);
  1320. defer engine.close();
  1321. _ = try engine.put("seed", "value");
  1322. var failed = std.atomic.Value(bool).init(false);
  1323. const Writer = struct {
  1324. fn run(target: *Engine, failure: *std.atomic.Value(bool)) void {
  1325. for (0..30) |index| {
  1326. var key: [16]u8 = undefined;
  1327. const encoded = std.fmt.bufPrint(&key, "key-{d}", .{index}) catch {
  1328. failure.store(true, .release);
  1329. return;
  1330. };
  1331. _ = target.put(encoded, "value") catch {
  1332. failure.store(true, .release);
  1333. return;
  1334. };
  1335. if (index == 15) target.checkpoint() catch {
  1336. failure.store(true, .release);
  1337. return;
  1338. };
  1339. }
  1340. }
  1341. };
  1342. const thread = try std.Thread.spawn(.{}, Writer.run, .{ &engine, &failed });
  1343. for (0..8) |index| {
  1344. var name: [32]u8 = undefined;
  1345. const filename = try std.fmt.bufPrint(&name, "copy-{d}.pkvdb", .{index});
  1346. var copy_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1347. const destination = try testPath(&tmp, filename, &copy_buffer);
  1348. defer std.testing.allocator.free(destination);
  1349. try copyPrefix(source, destination);
  1350. var copy = try Engine.open(std.testing.allocator, destination);
  1351. try std.testing.expect(try copy.exists("seed"));
  1352. copy.close();
  1353. }
  1354. thread.join();
  1355. try std.testing.expect(!failed.load(.acquire));
  1356. }
  1357. test "group commit preserves independent transaction LSNs" {
  1358. var tmp = std.testing.tmpDir(.{});
  1359. defer tmp.cleanup();
  1360. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1361. const path = try testPath(&tmp, "groups.pkvdb", &path_buffer);
  1362. defer std.testing.allocator.free(path);
  1363. var engine = try Engine.open(std.testing.allocator, path);
  1364. var operations: [256]Operation = undefined;
  1365. var keys: [256][8]u8 = undefined;
  1366. var lsns: [256]u64 = undefined;
  1367. for (&operations, 0..) |*operation, index| {
  1368. const key = try std.fmt.bufPrint(&keys[index], "k{d}", .{index});
  1369. operation.* = .{ .opcode = .put, .key = key, .value = "value" };
  1370. }
  1371. try engine.putMany(&operations, &lsns);
  1372. const status = engine.status();
  1373. try std.testing.expectEqual(@as(u64, 1), status.commit_groups);
  1374. try std.testing.expectEqual(@as(u64, 256), status.committed_transactions);
  1375. for (lsns, 0..) |lsn, index| try std.testing.expectEqual(index + 1, lsn);
  1376. engine.close();
  1377. engine = try Engine.open(std.testing.allocator, path);
  1378. defer engine.close();
  1379. try std.testing.expectEqual(@as(u64, 256), engine.status().latest_lsn);
  1380. for (operations) |operation| try std.testing.expect(try engine.exists(operation.key));
  1381. }
  1382. test "ordered index stays current after lazy construction" {
  1383. var tmp = std.testing.tmpDir(.{});
  1384. defer tmp.cleanup();
  1385. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1386. const path = try testPath(&tmp, "lazy-order.pkvdb", &path_buffer);
  1387. defer std.testing.allocator.free(path);
  1388. var engine = try Engine.open(std.testing.allocator, path);
  1389. defer engine.close();
  1390. _ = try engine.put("p/a", "one");
  1391. var batch = try engine.scan(std.testing.allocator, "p/", "", 10, false, 1024);
  1392. batch.deinit(std.testing.allocator);
  1393. _ = try engine.put("p/b", "two");
  1394. _ = try engine.put("p/a", "updated");
  1395. try std.testing.expect(try engine.delete("p/b"));
  1396. _ = try engine.put("p/c", "three");
  1397. batch = try engine.scan(std.testing.allocator, "p/", "", 10, true, 1024);
  1398. defer batch.deinit(std.testing.allocator);
  1399. try std.testing.expectEqual(@as(usize, 2), batch.entries.len);
  1400. try std.testing.expectEqualStrings("p/a", batch.entries[0].key);
  1401. try std.testing.expectEqualStrings("updated", batch.entries[0].value.?);
  1402. try std.testing.expectEqualStrings("p/c", batch.entries[1].key);
  1403. }
  1404. test "concurrent delete reports one removal" {
  1405. var tmp = std.testing.tmpDir(.{});
  1406. defer tmp.cleanup();
  1407. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1408. const path = try testPath(&tmp, "delete-race.pkvdb", &path_buffer);
  1409. defer std.testing.allocator.free(path);
  1410. var engine = try Engine.open(std.testing.allocator, path);
  1411. defer engine.close();
  1412. _ = try engine.put("key", "value");
  1413. var results: [2]bool = undefined;
  1414. var failed = std.atomic.Value(bool).init(false);
  1415. const Deleter = struct {
  1416. fn run(target: *Engine, result: *bool, failure: *std.atomic.Value(bool)) void {
  1417. result.* = target.delete("key") catch {
  1418. failure.store(true, .release);
  1419. return;
  1420. };
  1421. }
  1422. };
  1423. const first = try std.Thread.spawn(.{}, Deleter.run, .{ &engine, &results[0], &failed });
  1424. const second = try std.Thread.spawn(.{}, Deleter.run, .{ &engine, &results[1], &failed });
  1425. first.join();
  1426. second.join();
  1427. try std.testing.expect(!failed.load(.acquire));
  1428. try std.testing.expect(results[0] != results[1]);
  1429. }
  1430. test "compare batch write absent match and stale conflict" {
  1431. var tmp = std.testing.tmpDir(.{});
  1432. defer tmp.cleanup();
  1433. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1434. const path = try testPath(&tmp, "compare.pkvdb", &path_buffer);
  1435. defer std.testing.allocator.free(path);
  1436. var engine = try Engine.open(std.testing.allocator, path);
  1437. const absent_checks = [_]CompareCheck{.{ .key = "k", .expected_lsn = 0 }};
  1438. var result = try engine.compareBatchWrite(&absent_checks, &.{.{ .opcode = .put, .key = "k", .value = "one" }}, "meta");
  1439. try std.testing.expect(result.committed);
  1440. try std.testing.expectEqual(@as(u64, 1), result.lsn);
  1441. const match_checks = [_]CompareCheck{.{ .key = "k", .expected_lsn = result.lsn }};
  1442. result = try engine.compareBatchWrite(&match_checks, &.{.{ .opcode = .put, .key = "k", .value = "two" }}, "");
  1443. try std.testing.expect(result.committed);
  1444. try std.testing.expectEqual(@as(u64, 2), result.lsn);
  1445. const stale_checks = [_]CompareCheck{.{ .key = "k", .expected_lsn = 1 }};
  1446. result = try engine.compareBatchWrite(&stale_checks, &.{.{ .opcode = .put, .key = "k", .value = "three" }}, "");
  1447. try std.testing.expect(!result.committed);
  1448. try std.testing.expectEqual(@as(u64, 0), result.lsn);
  1449. var value = (try engine.get(std.testing.allocator, "k")).?;
  1450. try std.testing.expectEqualStrings("two", value.bytes);
  1451. try std.testing.expectEqual(@as(u64, 2), value.lsn);
  1452. std.testing.allocator.free(value.bytes);
  1453. engine.close();
  1454. engine = try Engine.open(std.testing.allocator, path);
  1455. defer engine.close();
  1456. value = (try engine.get(std.testing.allocator, "k")).?;
  1457. defer std.testing.allocator.free(value.bytes);
  1458. try std.testing.expectEqualStrings("two", value.bytes);
  1459. try std.testing.expectEqual(@as(u64, 2), value.lsn);
  1460. }
  1461. test "conditional queue heads bypass group coalescing" {
  1462. var completion = WriteCompletion{ .remaining = 1 };
  1463. var pending = PendingWrite{ .operations = &.{}, .metadata = "", .completion = &completion, .bytes = 0 };
  1464. try std.testing.expect(Engine.canCoalesce(&pending));
  1465. pending.checks = &.{.{ .key = "k", .expected_lsn = 1 }};
  1466. try std.testing.expect(!Engine.canCoalesce(&pending));
  1467. }
  1468. test "compare absent check rejects existing key" {
  1469. var tmp = std.testing.tmpDir(.{});
  1470. defer tmp.cleanup();
  1471. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1472. const path = try testPath(&tmp, "compare-absent.pkvdb", &path_buffer);
  1473. defer std.testing.allocator.free(path);
  1474. var engine = try Engine.open(std.testing.allocator, path);
  1475. defer engine.close();
  1476. _ = try engine.put("k", "seed");
  1477. const checks = [_]CompareCheck{.{ .key = "k", .expected_lsn = 0 }};
  1478. const result = try engine.compareBatchWrite(&checks, &.{.{ .opcode = .put, .key = "k", .value = "x" }}, "");
  1479. try std.testing.expect(!result.committed);
  1480. try std.testing.expectEqual(@as(u64, 0), result.lsn);
  1481. const value = (try engine.get(std.testing.allocator, "k")).?;
  1482. defer std.testing.allocator.free(value.bytes);
  1483. try std.testing.expectEqualStrings("seed", value.bytes);
  1484. }
  1485. test "concurrent compare batch writes expecting same lsn commit exactly one" {
  1486. var tmp = std.testing.tmpDir(.{});
  1487. defer tmp.cleanup();
  1488. var path_buffer: [std.fs.max_path_bytes]u8 = undefined;
  1489. const path = try testPath(&tmp, "compare-race.pkvdb", &path_buffer);
  1490. defer std.testing.allocator.free(path);
  1491. var engine = try Engine.open(std.testing.allocator, path);
  1492. defer engine.close();
  1493. _ = try engine.put("key", "seed");
  1494. var results: [2]CompareBatchResult = undefined;
  1495. var failed = std.atomic.Value(bool).init(false);
  1496. const Worker = struct {
  1497. fn run(target: *Engine, result: *CompareBatchResult, failure: *std.atomic.Value(bool)) void {
  1498. const checks = [_]CompareCheck{.{ .key = "key", .expected_lsn = 1 }};
  1499. const operations = [_]Operation{.{ .opcode = .put, .key = "key", .value = "winner" }};
  1500. result.* = target.compareBatchWrite(&checks, &operations, "") catch {
  1501. failure.store(true, .release);
  1502. return;
  1503. };
  1504. }
  1505. };
  1506. const first = try std.Thread.spawn(.{}, Worker.run, .{ &engine, &results[0], &failed });
  1507. const second = try std.Thread.spawn(.{}, Worker.run, .{ &engine, &results[1], &failed });
  1508. first.join();
  1509. second.join();
  1510. try std.testing.expect(!failed.load(.acquire));
  1511. const committed = @as(u64, @intFromBool(results[0].committed)) + @as(u64, @intFromBool(results[1].committed));
  1512. try std.testing.expectEqual(@as(u64, 1), committed);
  1513. const winner = if (results[0].committed) results[0] else results[1];
  1514. try std.testing.expectEqual(@as(u64, 2), winner.lsn);
  1515. const value = (try engine.get(std.testing.allocator, "key")).?;
  1516. defer std.testing.allocator.free(value.bytes);
  1517. try std.testing.expectEqualStrings("winner", value.bytes);
  1518. }