| ... | ... | @@ -126,6 +126,14 @@ pub fn HashMap(comptime K: type, comptime V: type, comptime hash: fn (key: K) u3 |
| 126 | 126 | }; |
| 127 | 127 | } |
| 128 | 128 | |
| 129 | pub fn getOrPutValue(self: *Self, key: K, value: V) !*KV { |
| 130 | const res = try self.getOrPut(key); |
| 131 | if (!res.found_existing) |
| 132 | res.kv.value = value; |
| 133 | |
| 134 | return res.kv; |
| 135 | } |
| 136 | |
| 129 | 137 | fn ensureCapacity(self: *Self) !void { |
| 130 | 138 | if (self.entries.len == 0) { |
| 131 | 139 | return self.initCapacity(16); |
| ... | ... | @@ -354,6 +362,12 @@ test "basic hash map usage" { |
| 354 | 362 | gop2.kv.value = 42; |
| 355 | 363 | assert(map.get(99).?.value == 42); |
| 356 | 364 | |
| 365 | const gop3 = try map.getOrPutValue(5, 5); |
| 366 | assert(gop3.value == 77); |
| 367 | |
| 368 | const gop4 = try map.getOrPutValue(100, 41); |
| 369 | assert(gop4.value == 41); |
| 370 | |
| 357 | 371 | assert(map.contains(2)); |
| 358 | 372 | assert(map.get(2).?.value == 22); |
| 359 | 373 | _ = map.remove(2); |
Jimmi Holst Christensen <jhc@dismail.de> 2018-11-29 09:10:15+01:00