private void run() throws Exception { stream = new SpannerImpl.GrpcStreamIterator(10); stream.setCall( new SpannerRpc.StreamingCall() { @Override public void cancel(@Nullable String message) {} @Override public void request(int numMessages) {} }); consumer = stream.consumer(); resultSet = new SpannerImpl.GrpcResultSet(stream, new NoOpListener(), QueryMode.NORMAL); JSONArray chunks = testCase.getJSONArray("chunks"); JSONObject expectedResult = testCase.getJSONObject("result"); for (int i = 0; i < chunks.length(); i++) { PartialResultSet.Builder builder = PartialResultSet.newBuilder(); JsonFormat.parser().merge(chunks.getString(i), builder); consumer.onPartialResultSet(builder.build()); } consumer.onCompleted(); assertResultSet(resultSet, expectedResult.getJSONArray("value")); }
try { Lumongo.CosineSimRequest.Builder consineSimRequest = Lumongo.CosineSimRequest.newBuilder(); JsonFormat.parser().merge(cosineSimJson, consineSimRequest); qrBuilder.addCosineSimRequest(consineSimRequest); try { Lumongo.Query.Builder filterQueryBuilder = Lumongo.Query.newBuilder(); JsonFormat.parser().merge(filterJsonQuery, filterQueryBuilder); qrBuilder.addFilterQuery(filterQueryBuilder); try { Lumongo.HighlightRequest.Builder hlBuilder = Lumongo.HighlightRequest.newBuilder(); JsonFormat.parser().merge(hlJson, hlBuilder); qrBuilder.addHighlightRequest(hlBuilder); try { Lumongo.AnalysisRequest.Builder analyzeRequestBuilder = Lumongo.AnalysisRequest.newBuilder(); JsonFormat.parser().merge(alJson, analyzeRequestBuilder); qrBuilder.addAnalysisRequest(analyzeRequestBuilder);
try { Lumongo.FuzzyTerm.Builder fuzzyTermBuilder = Lumongo.FuzzyTerm.newBuilder(); JsonFormat.parser().merge(fuzzyTermJson, fuzzyTermBuilder); termsBuilder.setFuzzyTerm(fuzzyTermBuilder);
functionDetailsJsonString = functionDetailsJsonString.substring(0, functionDetailsJsonString.length() - 1); JsonFormat.parser().merge(functionDetailsJsonString, functionDetailsBuilder); FunctionDetails functionDetails = functionDetailsBuilder.build(); instanceConfig.setFunctionDetails(functionDetails);
functionDetailsJsonString = functionDetailsJsonString.substring(0, functionDetailsJsonString.length() - 1); JsonFormat.parser().merge(functionDetailsJsonString, functionDetailsBuilder); functionDetails = functionDetailsBuilder.build();
JsonHandler.parser().merge(body, toMerge);
private ServerInfos fetchServerInfos() { return SonarLintWsClient.processTimed( () -> wsClient.rawGet("api/system/status"), response -> { if (!response.isSuccessful()) { if (response.code() == HttpURLConnection.HTTP_NOT_FOUND) { return tryFromDeprecatedApi(response); } else { throw SonarLintWsClient.handleError(response); } } else { String responseStr = response.content(); try { ServerInfos.Builder builder = ServerInfos.newBuilder(); JsonFormat.parser().merge(responseStr, builder); return builder.build(); } catch (InvalidProtocolBufferException e) { throw new IllegalStateException("Unable to parse server infos from: " + StringUtils.abbreviate(responseStr, 100), e); } } }, duration -> LOG.debug("Downloaded server infos in {}ms", duration)); }
@SuppressWarnings("unchecked") @Override public T parse(InputStream stream) { Builder builder = defaultInstance.newBuilderForType(); Reader reader = new InputStreamReader(stream, charset); T proto; try { parser.merge(reader, builder); proto = (T) builder.build(); reader.close(); } catch (InvalidProtocolBufferException e) { throw Status.INTERNAL.withDescription("Invalid protobuf byte sequence") .withCause(e).asRuntimeException(); } catch (IOException e) { // Same for now, might be unavailable throw Status.INTERNAL.withDescription("Invalid protobuf byte sequence") .withCause(e).asRuntimeException(); } return proto; } };
@SuppressWarnings("unchecked") public <T extends Message> T convertJsonToProto(T prototype, String json, String extensionName) { try { Builder builder = prototype.newBuilderForType(); JsonFormat.parser().merge(json, builder); return (T) builder.build(); } catch (InvalidProtocolBufferException ex) { diagCollector.addDiag( Diag.error( new SimpleLocation(extensionName), "Extension %s cannot be converted into proto type %s. Details: %s", extensionName, prototype.getDescriptorForType().getFullName(), ex.getMessage())); return prototype; } }
@ProcessElement public void processElement(ProcessContext c) { try { KV<String, String> kv = c.element(); String filePath = GcsUtil.joinPath(importDirectory.get(), kv.getValue()); MatchResult match = FileSystems.match(filePath, EmptyMatchTreatment.DISALLOW); ResourceId resourceId = match.metadata().get(0).resourceId(); TableManifest.Builder builder = TableManifest.newBuilder(); try (InputStream stream = Channels.newInputStream(FileSystems.open(resourceId))) { Reader reader = new InputStreamReader(stream); JsonFormat.parser().merge(reader, builder); } c.output(KV.of(kv.getKey(), builder.build())); } catch (IOException e) { throw new RuntimeException(e); } } }));
@SuppressWarnings("unchecked") public <T extends Message> T convertJsonToProto(T prototype, String json, String extensionName) { try { Builder builder = prototype.newBuilderForType(); JsonFormat.parser().merge(json, builder); return (T) builder.build(); } catch (InvalidProtocolBufferException ex) { diagCollector.addDiag( Diag.error( new SimpleLocation(extensionName), "Extension %s cannot be converted into proto type %s. Details: %s", extensionName, prototype.getDescriptorForType().getFullName(), ex.getMessage())); return prototype; } }
public ImmutableList<DynamicMessage> read() { ImmutableList.Builder<DynamicMessage> resultBuilder = ImmutableList.builder(); try { payloadList.forEach(value -> { DynamicMessage.Builder nextMessage = DynamicMessage.newBuilder(descriptor); try { jsonParser.merge(new Gson().toJson(value), nextMessage); } catch (InvalidProtocolBufferException e) { logger.warning(e.getMessage()); } resultBuilder.add(nextMessage.build()); }); } catch (Exception e) { throw new IllegalArgumentException("Can't read messages"); } return resultBuilder.build(); }
@Test public void jsonSaverSavesMetadata() throws IOException { KeanuSavedBayesNet.Metadata.Builder metadataBuilder = KeanuSavedBayesNet.Metadata.newBuilder(); for (Map.Entry<String, String> entry : someMetadata.entrySet()) { metadataBuilder.putMetadataInfo(entry.getKey(), entry.getValue()); } KeanuSavedBayesNet.Model.Builder modelBuilder = KeanuSavedBayesNet.Model.newBuilder(); JsonFormat.parser().merge(outputStream.toString(), modelBuilder); KeanuSavedBayesNet.Model parsedModel = modelBuilder.build(); assertTrue(parsedModel.hasMetadata()); assertEquals(parsedModel.getMetadata().getMetadataInfoMap().size(), (metadataBuilder.getMetadataInfoMap().size())); assertThat(parsedModel.getMetadata().getMetadataInfoMap().entrySet(), everyItem(isIn(metadataBuilder.getMetadataInfoMap().entrySet()))); }
public static void mergeJson(String json, Builder builder) throws IOException { JsonFormat.parser().merge(json, builder); }
public static void mergeJson(String json, Builder builder) throws IOException { JsonFormat.parser().merge(json, builder); }
@Override public void merge(InputStream input, Charset charset, MediaType contentType, ExtensionRegistry extensionRegistry, Message.Builder builder) throws IOException, HttpMessageConversionException { if (contentType.isCompatibleWith(APPLICATION_JSON)) { InputStreamReader reader = new InputStreamReader(input, charset); this.parser.merge(reader, builder); } else { throw new HttpMessageConversionException( "protobuf-java-util does not support parsing " + contentType); } }
@Override public void merge(InputStream input, Charset charset, MediaType contentType, ExtensionRegistry extensionRegistry, Message.Builder builder) throws IOException { if (contentType.isCompatibleWith(APPLICATION_JSON)) { InputStreamReader reader = new InputStreamReader(input, charset); this.parser.merge(reader, builder); } else { throw new IOException("protobuf-java-util does not support " + contentType + " format"); } }
private static Struct messageAsStruct(MessageOrBuilder message) { try { String json = JsonFormat.printer() .preservingProtoFieldNames() .print(message); Struct.Builder structBuilder = Struct.newBuilder(); JsonFormat.parser().merge(json, structBuilder); return structBuilder.build(); } catch (InvalidProtocolBufferException e) { throw new RuntimeException("Failed to convert protobuf message to struct", e); } }
private static void structAsMessage(Struct struct, Message.Builder messageBuilder) throws InvalidProtocolBufferException { String json = JsonFormat.printer() .preservingProtoFieldNames() .print(struct); JsonFormat.parser().merge(json, messageBuilder); }
public EntityJsonParser() { TypeRegistry typeRegistry = TypeRegistry.newBuilder() .add(Entity.getDescriptor()) .build(); jsonParser = JsonFormat.parser() .usingTypeRegistry(typeRegistry); }