authorgravatar for 14938807+xackus@users.noreply.github.comxackus <14938807+xackus@users.noreply.github.com> 2020-04-21 10:25:09+02:00
committergravatar for 14938807+xackus@users.noreply.github.comxackus <14938807+xackus@users.noreply.github.com> 2020-04-23 22:14:23+02:00
logebbd137a0e90a8e78655d2a5f74cf87120b5a5a1
tree9c5db94755d862da4ad7cbf0c1d3b1c990932d9b
parenta9eb4a6740e0bb00e1984de3768a7de6001c25dd

fix json.WriteStream.emitJson

1 files changed, 32 insertions(+), 23 deletions(-)

lib/std/json/write_stream.zig+32-23
......@@ -148,6 +148,7 @@ pub fn WriteStream(comptime OutStream: type, comptime max_depth: usize) type {
148148 self.popState();
149149 }
150150
151 // TODO better handling of ComptimeInt and ComptimeFloat
151152 pub fn emitNumber(
152153 self: *Self,
153154 /// An integer, float, or `std.math.BigInt`. Emitted as a bare number if it fits losslessly
......@@ -180,6 +181,7 @@ pub fn WriteStream(comptime OutStream: type, comptime max_depth: usize) type {
180181 }
181182
182183 pub fn emitString(self: *Self, string: []const u8) !void {
184 assert(self.state[self.state_index] == State.Value);
183185 try self.writeEscapedString(string);
184186 self.popState();
185187 }
......@@ -191,7 +193,9 @@ pub fn WriteStream(comptime OutStream: type, comptime max_depth: usize) type {
191193
192194 /// Writes the complete json into the output stream
193195 pub fn emitJson(self: *Self, json: std.json.Value) Stream.Error!void {
196 assert(self.state[self.state_index] == State.Value);
194197 try self.stringify(json);
198 self.popState();
195199 }
196200
197201 fn indent(self: *Self) !void {
......@@ -233,7 +237,32 @@ test "json write stream" {
233237 defer arena_allocator.deinit();
234238
235239 var w = std.json.writeStream(out, 10);
236 try w.emitJson(try getJson(&arena_allocator.allocator));
240
241 try w.beginObject();
242
243 try w.objectField("object");
244 try w.emitJson(try getJsonObject(&arena_allocator.allocator));
245
246 try w.objectField("string");
247 try w.emitString("This is a string");
248
249 try w.objectField("array");
250 try w.beginArray();
251 try w.arrayElem();
252 try w.emitString("Another string");
253 try w.arrayElem();
254 try w.emitNumber(@as(i32, 1));
255 try w.arrayElem();
256 try w.emitNumber(@as(f32, 3.5));
257 try w.endArray();
258
259 try w.objectField("int");
260 try w.emitNumber(@as(i32, 10));
261
262 try w.objectField("float");
263 try w.emitNumber(@as(f32, 3.5));
264
265 try w.endObject();
237266
238267 const result = slice_stream.getWritten();
239268 const expected =
......@@ -246,38 +275,18 @@ test "json write stream" {
246275 \\ "array": [
247276 \\ "Another string",
248277 \\ 1,
249 \\ 3.14e+00
278 \\ 3.5e+00
250279 \\ ],
251280 \\ "int": 10,
252 \\ "float": 3.14e+00
281 \\ "float": 3.5e+00
253282 \\}
254283 ;
255284 std.testing.expect(std.mem.eql(u8, expected, result));
256285}
257286
258fn getJson(allocator: *std.mem.Allocator) !std.json.Value {
259 var value = std.json.Value{ .Object = std.json.ObjectMap.init(allocator) };
260 _ = try value.Object.put("string", std.json.Value{ .String = "This is a string" });
261 _ = try value.Object.put("int", std.json.Value{ .Integer = @intCast(i64, 10) });
262 _ = try value.Object.put("float", std.json.Value{ .Float = 3.14 });
263 _ = try value.Object.put("array", try getJsonArray(allocator));
264 _ = try value.Object.put("object", try getJsonObject(allocator));
265 return value;
266}
267
268287fn getJsonObject(allocator: *std.mem.Allocator) !std.json.Value {
269288 var value = std.json.Value{ .Object = std.json.ObjectMap.init(allocator) };
270289 _ = try value.Object.put("one", std.json.Value{ .Integer = @intCast(i64, 1) });
271290 _ = try value.Object.put("two", std.json.Value{ .Float = 2.0 });
272291 return value;
273292}
274
275fn getJsonArray(allocator: *std.mem.Allocator) !std.json.Value {
276 var value = std.json.Value{ .Array = std.json.Array.init(allocator) };
277 var array = &value.Array;
278 _ = try array.append(std.json.Value{ .String = "Another string" });
279 _ = try array.append(std.json.Value{ .Integer = @intCast(i64, 1) });
280 _ = try array.append(std.json.Value{ .Float = 3.14 });
281
282 return value;
283}