private List<Map<String, Object>> parseTableResult(final OQLQueryResponse response) { final Table table = response.getTableResult(); final ProtocolStringList fieldNames = table.getFieldNameList(); List<Map<String, Object>> results = new ArrayList<>(); for (BasicTypes.EncodedValueList row : table.getRowList()) { final List<Object> decodedRow = row.getElementList().stream().map(valueEncoder::decodeValue) .collect(Collectors.toList()); Map<String, Object> rowMap = new LinkedHashMap<>(decodedRow.size()); for (int i = 0; i < decodedRow.size(); i++) { rowMap.put(fieldNames.get(i), decodedRow.get(i)); } results.add(rowMap); } return results; }
@Test @SuppressWarnings("all") public void listCollectionIdsTest() { String nextPageToken = ""; String collectionIdsElement = "collectionIdsElement1368994900"; List<String> collectionIds = Arrays.asList(collectionIdsElement); ListCollectionIdsResponse expectedResponse = ListCollectionIdsResponse.newBuilder() .setNextPageToken(nextPageToken) .addAllCollectionIds(collectionIds) .build(); mockFirestore.addResponse(expectedResponse); String formattedParent = FirestoreClient.formatAnyPathName("[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]"); ListCollectionIdsPagedResponse pagedListResponse = client.listCollectionIds(formattedParent); List<String> resources = Lists.newArrayList(pagedListResponse.iterateAll()); Assert.assertEquals(1, resources.size()); Assert.assertEquals(expectedResponse.getCollectionIdsList().get(0), resources.get(0)); List<GeneratedMessageV3> actualRequests = mockFirestore.getRequests(); Assert.assertEquals(1, actualRequests.size()); ListCollectionIdsRequest actualRequest = (ListCollectionIdsRequest) actualRequests.get(0); Assert.assertEquals(formattedParent, actualRequest.getParent()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
Assert.assertEquals(expectedResponse.getSubscriptionsList().get(0), resources.get(0)); List<ProjectSubscriptionName> resourceNames = Lists.newArrayList(pagedListResponse.iterateAllAsProjectSubscriptionName()); Assert.assertEquals(1, resourceNames.size()); Assert.assertEquals( ProjectSubscriptionName.parse(expectedResponse.getSubscriptionsList().get(0)), resourceNames.get(0));
@Test @SuppressWarnings("all") public void listCollectionIdsTest() { String nextPageToken = ""; String collectionIdsElement = "collectionIdsElement1368994900"; List<String> collectionIds = Arrays.asList(collectionIdsElement); ListCollectionIdsResponse expectedResponse = ListCollectionIdsResponse.newBuilder() .setNextPageToken(nextPageToken) .addAllCollectionIds(collectionIds) .build(); mockFirestore.addResponse(expectedResponse); String formattedParent = FirestoreClient.formatAnyPathName("[PROJECT]", "[DATABASE]", "[DOCUMENT]", "[ANY_PATH]"); ListCollectionIdsPagedResponse pagedListResponse = client.listCollectionIds(formattedParent); List<String> resources = Lists.newArrayList(pagedListResponse.iterateAll()); Assert.assertEquals(1, resources.size()); Assert.assertEquals(expectedResponse.getCollectionIdsList().get(0), resources.get(0)); List<GeneratedMessageV3> actualRequests = mockFirestore.getRequests(); Assert.assertEquals(1, actualRequests.size()); ListCollectionIdsRequest actualRequest = (ListCollectionIdsRequest) actualRequests.get(0); Assert.assertEquals(formattedParent, actualRequest.getParent()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
@Test @SuppressWarnings("all") public void listLogsTest() { String nextPageToken = ""; String logNamesElement = "logNamesElement-1079688374"; List<String> logNames = Arrays.asList(logNamesElement); ListLogsResponse expectedResponse = ListLogsResponse.newBuilder() .setNextPageToken(nextPageToken) .addAllLogNames(logNames) .build(); mockLoggingServiceV2.addResponse(expectedResponse); ParentName parent = ProjectName.of("[PROJECT]"); ListLogsPagedResponse pagedListResponse = client.listLogs(parent); List<String> resources = Lists.newArrayList(pagedListResponse.iterateAll()); Assert.assertEquals(1, resources.size()); Assert.assertEquals(expectedResponse.getLogNamesList().get(0), resources.get(0)); List<GeneratedMessageV3> actualRequests = mockLoggingServiceV2.getRequests(); Assert.assertEquals(1, actualRequests.size()); ListLogsRequest actualRequest = (ListLogsRequest) actualRequests.get(0); Assert.assertEquals(parent, ParentNames.parse(actualRequest.getParent())); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); }
HashMap<String,Double> nerLabelProbs = new HashMap<>(); if (!proto.getNerLabelProbsList().get(0).equals("empty")) { for (String nerLabelProb : proto.getNerLabelProbsList()) { String[] labelAndProb = nerLabelProb.split("=");
/** * Return a list of protocol strings in JSON form * @param node ArrayNode. The node we will add to. * @param list List. A list of protocol strings. * @return ArrayNode. The node we passed in. */ protected static ArrayNode getAsStringList(ArrayNode node, ProtocolStringList list) { for (int i=0; i<list.size();i++) { node.add(list.get(i)); } return node; }
private List<String> encodeInfoValues(Map<String, String> attr, List<Integer> infoKeys) { // infoKeys.stream().map(x -> attr.get(x)).collect(Collectors.toList()); // not sure if order is protected List<String> values = new ArrayList<>(infoKeys.size()); for (Integer key : infoKeys) { String value = attr.get(fields.getInfoKeysList().get(key)); //Null values are not accepted in proto values.add(value == null ? "" : value); } return values; }
public void addPoint(Point dp, Measurement m) throws IOException { writeLock.lock(); try { int timeBucket = getOrCreateTimeBucket(dp.getTimestamp(), m.getTimeBucketSize()); Field timeField = getOrCreateSeries(timeBucket, TS, false, m); timeField.addDataPoint(m, dp.getTimestamp()); for (int i = 0; i < dp.getFpList().size(); i++) { Field field = getOrCreateSeries(timeBucket, dp.getValueFieldNameList().get(i), dp.getFpList().get(i), m); field.addDataPoint(m, dp.getValueList().get(i)); } } finally { writeLock.unlock(); } }
@Test public void testGetMethodSignatures() { Method getShelfMethod = libraryService.lookupMethod("GetShelf"); List<MethodSignature> getShelfFlattenings = protoParser.getMethodSignatures(getShelfMethod); assertThat(getShelfFlattenings.size()).isEqualTo(3); MethodSignature firstSignature = getShelfFlattenings.get(0); assertThat(firstSignature.getFieldsList().size()).isEqualTo(1); assertThat(firstSignature.getFieldsList().get(0)).isEqualTo("name"); MethodSignature additionalSignature = getShelfFlattenings.get(1); assertThat(additionalSignature.getFieldsList().size()).isEqualTo(2); assertThat(additionalSignature.getFieldsList().get(0)).isEqualTo("name"); assertThat(additionalSignature.getFieldsList().get(1)).isEqualTo("message"); MethodSignature additionalSignature2 = getShelfFlattenings.get(2); assertThat(additionalSignature2.getFieldsList()) .containsExactly("name", "message", "string_builder"); }
formatPositions.put(studyEntry.getFormatList().get(i), i);
static void mergeVariants(Builder destVariant, Variant srcVariant) { // Merge alternates preserving the order of pre-existing alts in the destination. for (String srcAlt : srcVariant.getAlternateBasesList()) { if (!destVariant.getAlternateBasesList().contains(srcAlt)) { destVariant.addAlternateBases(srcAlt); } } // Re-number genotypes and merge calls. for (VariantCall call : srcVariant.getCallsList()) { VariantCall.Builder updatedCall = VariantCall.newBuilder(call); for (int i = 0; i < call.getGenotypeCount(); i++) { if (0 < call.getGenotype(i)) { int updatedGenotype = destVariant.getAlternateBasesList().indexOf( srcVariant.getAlternateBasesList().get(call.getGenotype(i) - 1)) + 1; updatedCall.setGenotype(i, updatedGenotype); } } destVariant.addCalls(updatedCall); } } }
HashMap<String,Double> nerLabelProbs = new HashMap<>(); if (!proto.getNerLabelProbsList().get(0).equals("empty")) { for (String nerLabelProb : proto.getNerLabelProbsList()) { String[] labelAndProb = nerLabelProb.split("=");