org.elasticsearch.common.xcontent.json.JsonXContent#contentBuilder

Source File: RestActionReceiversTest.java From crate with Apache License 2.0

6 votes

@Test
public void testRestRowCountReceiver() throws Exception {
    RestRowCountReceiver receiver = new RestRowCountReceiver(JsonXContent.contentBuilder(), 0L, true);
    receiver.setNextRow(row);
    XContentBuilder actualBuilder = receiver.finishBuilder();

    ResultToXContentBuilder builder = ResultToXContentBuilder.builder(JsonXContent.contentBuilder());
    builder.cols(Collections.<ScopedSymbol>emptyList());
    builder.colTypes(Collections.<ScopedSymbol>emptyList());
    builder.startRows();
    builder.addRow(row, 0);
    builder.finishRows();
    builder.rowCount(1L);

    assertXContentBuilder(actualBuilder, builder.build());
}

Source File: TransportSchemaUpdateAction.java From crate with Apache License 2.0

6 votes

@VisibleForTesting
static ClusterState updateTemplate(NamedXContentRegistry xContentRegistry,
                                   ClusterState currentState,
                                   String templateName,
                                   Map<String, Object> newMapping) throws Exception {
    IndexTemplateMetaData template = currentState.metaData().templates().get(templateName);
    if (template == null) {
        throw new ResourceNotFoundException("Template \"" + templateName + "\" for partitioned table is missing");
    }

    IndexTemplateMetaData.Builder templateBuilder = new IndexTemplateMetaData.Builder(template);
    for (ObjectObjectCursor<String, CompressedXContent> cursor : template.mappings()) {
        Map<String, Object> source = parseMapping(xContentRegistry, cursor.value.toString());
        mergeIntoSource(source, newMapping);
        try (XContentBuilder xContentBuilder = JsonXContent.contentBuilder()) {
            templateBuilder.putMapping(cursor.key, Strings.toString(xContentBuilder.map(source)));
        }
    }
    MetaData.Builder builder = MetaData.builder(currentState.metaData()).put(templateBuilder);
    return ClusterState.builder(currentState).metaData(builder).build();
}

Source File: JsonType.java From crate with Apache License 2.0

6 votes

@Override
protected byte[] encodeAsUTF8Text(@Nonnull Object value) {
    try {
        XContentBuilder builder = JsonXContent.contentBuilder();
        if (value instanceof List) {
            List values = ((List) value);
            builder.startArray();
            for (Object o : values) {
                builder.value(o);
            }
            builder.endArray();
        } else {
            builder.map((Map) value);
        }
        builder.close();
        return BytesReference.toBytes(BytesReference.bytes(builder));
    } catch (IOException e) {
        throw new RuntimeException(e);
    }
}

Source File: XContentTestUtilsTests.java From crate with Apache License 2.0

6 votes

@SuppressWarnings("unchecked")
public void testInsertIntoXContent() throws IOException {
    XContentBuilder builder = JsonXContent.contentBuilder();
    builder.startObject();
    builder.endObject();
    builder = XContentTestUtils.insertIntoXContent(XContentType.JSON.xContent(), BytesReference.bytes(builder),
            Collections.singletonList(""), () -> "inn.er1", () -> new HashMap<>());
    builder = XContentTestUtils.insertIntoXContent(XContentType.JSON.xContent(), BytesReference.bytes(builder),
            Collections.singletonList(""), () -> "field1", () -> "value1");
    builder = XContentTestUtils.insertIntoXContent(XContentType.JSON.xContent(), BytesReference.bytes(builder),
            Collections.singletonList("inn\\.er1"), () -> "inner2", () -> new HashMap<>());
    builder = XContentTestUtils.insertIntoXContent(XContentType.JSON.xContent(), BytesReference.bytes(builder),
            Collections.singletonList("inn\\.er1"), () -> "field2", () -> "value2");
    try (XContentParser parser = XContentHelper.createParser(NamedXContentRegistry.EMPTY,
        DeprecationHandler.THROW_UNSUPPORTED_OPERATION, BytesReference.bytes(builder), builder.contentType())) {
        Map<String, Object> map = parser.map();
        assertEquals(2, map.size());
        assertEquals("value1", map.get("field1"));
        assertThat(map.get("inn.er1"), instanceOf(Map.class));
        Map<String, Object> innerMap = (Map<String, Object>) map.get("inn.er1");
        assertEquals(2, innerMap.size());
        assertEquals("value2", innerMap.get("field2"));
        assertThat(innerMap.get("inner2"), instanceOf(Map.class));
        assertEquals(0, ((Map<String, Object>) innerMap.get("inner2")).size());
    }
}

Source File: AbstractClient.java From elasticshell with Apache License 2.0

6 votes

public JsonOutput availableNodes() throws Exception {
    ClusterStateResponse response = this.client.admin().cluster().state(new ClusterStateRequest()
            .filterBlocks(true).filterNodes(false).filterMetaData(true)
            .filterRoutingTable(true)).actionGet();

    XContentBuilder builder = JsonXContent.contentBuilder();
    builder.startObject();
    for (DiscoveryNode discoveryNode : response.getState().nodes()) {
        builder.startObject(discoveryNode.id());
        builder.field("name", discoveryNode.name());
        builder.endObject();
    }
    builder.endObject();

    return stringToJson.stringToJson(builder.string());
}

Source File: ResponseUtil.java From elasticsearch-auth with Apache License 2.0

6 votes

public static void send(final RestRequest request,
        final RestChannel channel, final RestStatus status,
        final String... args) {
    try {
        final XContentBuilder builder = JsonXContent.contentBuilder();
        builder.startObject();
        builder.field("status", status.getStatus());
        for (int i = 0; i < args.length; i += 2) {
            builder.field(args[i], args[i + 1]);
        }
        builder.endObject();
        channel.sendResponse(new BytesRestResponse(status, builder));
    } catch (final IOException e) {
        logger.error("Failed to send a response.", e);
        try {
            channel.sendResponse(new BytesRestResponse(channel, e));
        } catch (final IOException e1) {
            logger.error("Failed to send a failure response.", e1);
        }
    }
}

Source File: RestDataAction.java From elasticsearch-dataformat with Apache License 2.0

6 votes

private void sendResponse(final RestRequest request, final RestChannel channel, final String file) {
    try {
        final XContentBuilder builder = JsonXContent.contentBuilder();
        final String pretty = request.param("pretty");
        if (pretty != null && !"false".equalsIgnoreCase(pretty)) {
            builder.prettyPrint().lfAtEnd();
        }
        builder.startObject();
        builder.field("acknowledged", true);
        builder.field("file", file);
        builder.endObject();
        channel.sendResponse(new BytesRestResponse(OK, builder));
    } catch (final IOException e) {
        throw new ElasticsearchException("Failed to create a resposne.", e);
    }
}

Source File: TasteActionRestAction.java From elasticsearch-taste with Apache License 2.0

6 votes

private void sendResponse(final RestRequest request,
        final RestChannel channel, final Map<String, Object> params,
        final boolean acknowledged) {
    try {
        final XContentBuilder builder = JsonXContent.contentBuilder();
        if (request.hasParam("pretty")) {
            builder.prettyPrint().lfAtEnd();
        }
        builder.startObject();
        builder.field("acknowledged", acknowledged);
        if (params != null) {
            for (final Map.Entry<String, Object> entry : params.entrySet()) {
                builder.field(entry.getKey(), entry.getValue());
            }
        }
        builder.endObject();
        channel.sendResponse(new BytesRestResponse(OK, builder));
    } catch (final Exception e) {
        sendErrorResponse(channel, e);
    }
}

Source File: ReindexRestAction.java From elasticsearch-reindexing with Apache License 2.0

6 votes

private void sendResponse(final RestRequest request,
        final RestChannel channel, final Map<String, Object> params) {
    try {
        final XContentBuilder builder = JsonXContent.contentBuilder();
        if (request.hasParam("pretty")) {
            builder.prettyPrint().lfAtEnd();
        }
        builder.startObject();
        builder.field("acknowledged", true);
        if (params != null) {
            for (final Map.Entry<String, Object> entry : params.entrySet()) {
                builder.field(entry.getKey(), entry.getValue());
            }
        }
        builder.endObject();
        channel.sendResponse(new BytesRestResponse(OK, builder));
    } catch (final Exception e) {
        sendErrorResponse(channel, e);
    }
}

Source File: LinearRankerParserTests.java From elasticsearch-learning-to-rank with Apache License 2.0

6 votes

public void testParse() throws IOException {
    XContentBuilder builder = JsonXContent.contentBuilder();
    StoredFeatureSet set = LtrTestUtils.randomFeatureSet();
    float[] expectedWeights = new float[set.size()];
    builder.startObject();
    for (int i = 0; i < set.size(); i++) {
        float weight = random().nextFloat();
        expectedWeights[i] = weight;
        builder.field(set.feature(i).name(), weight);
    }
    builder.endObject();
    String json = Strings.toString(builder);
    LinearRankerParser parser = new LinearRankerParser();
    LinearRanker ranker = parser.parse(set, json);
    DenseFeatureVector v = ranker.newFeatureVector(null);
    LinearRankerTests.fillRandomWeights(v.scores);
    LinearRanker expectedRanker = new LinearRanker(expectedWeights);
    Assert.assertEquals(expectedRanker.score(v), ranker.score(v), Math.ulp(expectedRanker.score(v)));
}

Source File: ResponseUtil.java From Elasticsearch with Apache License 2.0

6 votes

public static void send(final RestChannel channel, final RestStatus status, final String message) {
    try {
        final XContentBuilder builder = JsonXContent.contentBuilder();
        builder.startObject()
                .field("status", status.getStatus())
                .field("message", message).endObject();
        BytesRestResponse bytesRestResponse = new BytesRestResponse(status, builder);
        if (status == RestStatus.UNAUTHORIZED) {
            bytesRestResponse.addHeader("WWW-authenticate", "Basic realm=\"Elasticsearch Authentication\"");
        }
        channel.sendResponse(bytesRestResponse);
    } catch (final IOException e) {
        logger.error("Failed to send a response.", e);
        try {
            channel.sendResponse(new BytesRestResponse(channel, e));
        } catch (final IOException e1) {
            logger.error("Failed to send a failure response.", e1);
        }
    }
}

Source File: LtrTestUtils.java From elasticsearch-learning-to-rank with Apache License 2.0

5 votes

public static StoredLtrModel randomLinearModel(String name, StoredFeatureSet set) throws IOException {
    XContentBuilder builder = JsonXContent.contentBuilder();
    builder.startObject();
    for (int i = 0; i < set.size(); i++) {
        builder.field(set.feature(i).name(), random().nextFloat());
    }
    builder.endObject();
    return new StoredLtrModel(name, set, LinearRankerParser.TYPE, Strings.toString(builder), false, randomFtrNorms(set));
}

Source File: LinearRankerParserTests.java From elasticsearch-learning-to-rank with Apache License 2.0

5 votes

public static String generateRandomModelString(FeatureSet set) throws IOException {
    XContentBuilder builder = JsonXContent.contentBuilder();
    builder.startObject();
    for (int i = 0; i < set.size(); i++) {
        builder.field(set.feature(i).name(), random().nextFloat());
    }
    builder.endObject().close();
    return Strings.toString(builder);
}

Source File: StoredLtrModel.java From elasticsearch-learning-to-rank with Apache License 2.0

5 votes

private void parseModel(XContentParser parser) throws IOException {
        if (parser.currentToken() == XContentParser.Token.VALUE_STRING) {
            modelAsString = true;
            definition = parser.text();
        } else {
            try (XContentBuilder builder = JsonXContent.contentBuilder()) {
                builder.copyCurrentStructure(parser);
                modelAsString = false;
                definition = Strings.toString(builder);
            }
        }
}

Source File: Strings.java From Elasticsearch with Apache License 2.0

5 votes

/**
 * Return a {@link String} that is the json representation of the provided
 * {@link ToXContent}.
 */
public static String toString(ToXContent toXContent) {
    try {
        XContentBuilder builder = JsonXContent.contentBuilder();
        toXContent.toXContent(builder, ToXContent.EMPTY_PARAMS);
        return builder.string();
    } catch (IOException e) {
        throw new AssertionError("Cannot happen", e);
    }
}

Source File: XContentFactory.java From crate with Apache License 2.0

5 votes

/**
 * Returns a binary content builder for the provided content type.
 */
public static XContentBuilder contentBuilder(XContentType type) throws IOException {
    if (type == XContentType.JSON) {
        return JsonXContent.contentBuilder();
    } else if (type == XContentType.SMILE) {
        return SmileXContent.contentBuilder();
    } else if (type == XContentType.YAML) {
        return YamlXContent.contentBuilder();
    }
    throw new IllegalArgumentException("No matching content type for " + type);
}

Source File: Strings.java From crate with Apache License 2.0

5 votes

private static XContentBuilder createBuilder(boolean pretty, boolean human) throws IOException {
    XContentBuilder builder = JsonXContent.contentBuilder();
    if (pretty) {
        builder.prettyPrint();
    }
    if (human) {
        builder.humanReadable(true);
    }
    return builder;
}

Source File: RepositoryDataTests.java From crate with Apache License 2.0

5 votes

public void testXContent() throws IOException {
    RepositoryData repositoryData = generateRandomRepoData();
    XContentBuilder builder = JsonXContent.contentBuilder();
    repositoryData.snapshotsToXContent(builder, true);
    try (XContentParser parser = createParser(JsonXContent.JSON_XCONTENT, BytesReference.bytes(builder))) {
        long gen = (long) randomIntBetween(0, 500);
        RepositoryData fromXContent = RepositoryData.snapshotsFromXContent(parser, gen);
        assertEquals(repositoryData, fromXContent);
        assertEquals(gen, fromXContent.getGenId());
    }
}

Source File: SqlHttpHandler.java From crate with Apache License 2.0

5 votes

private CompletableFuture<XContentBuilder> executeSimpleRequest(Session session,
                                                                String stmt,
                                                                List<Object> args,
                                                                boolean includeTypes) throws IOException {
    long startTimeInNs = System.nanoTime();
    session.parse(UNNAMED, stmt, emptyList());
    session.bind(UNNAMED, UNNAMED, args == null ? emptyList() : args, null);
    DescribeResult description = session.describe('P', UNNAMED);
    List<Symbol> resultFields = description.getFields();
    ResultReceiver<XContentBuilder> resultReceiver;
    if (resultFields == null) {
        resultReceiver = new RestRowCountReceiver(JsonXContent.contentBuilder(), startTimeInNs, includeTypes);
    } else {
        CircuitBreaker breaker = circuitBreakerProvider.apply(HierarchyCircuitBreakerService.QUERY);
        RamAccounting ramAccounting = new BlockBasedRamAccounting(
            b -> breaker.addEstimateBytesAndMaybeBreak(b, "http-result"),
            MAX_BLOCK_SIZE_IN_BYTES);
        resultReceiver = new RestResultSetReceiver(
            JsonXContent.contentBuilder(),
            resultFields,
            startTimeInNs,
            new RowAccountingWithEstimators(
                Symbols.typeView(resultFields),
                ramAccounting
            ),
            includeTypes
        );
        resultReceiver.completionFuture().whenComplete((result, error) -> ramAccounting.close());
    }
    session.execute(UNNAMED, 0, resultReceiver);
    return session.sync()
        .thenCompose(ignored -> resultReceiver.completionFuture());
}

Source File: XContentTestUtilsTests.java From crate with Apache License 2.0

4 votes

public void testGetInsertPaths() throws IOException {
    XContentBuilder builder = JsonXContent.contentBuilder();
    builder.startObject();
    {
        builder.field("field1", "value");
        builder.startArray("list1");
        {
            builder.value(0);
            builder.value(1);
            builder.startObject();
            builder.endObject();
            builder.value(3);
            builder.startObject();
            builder.endObject();
        }
        builder.endArray();
        builder.startObject("inner1");
        {
            builder.field("inner1field1", "value");
            builder.startObject("inn.er2");
            {
                builder.field("inner2field1", "value");
            }
            builder.endObject();
        }
        builder.endObject();
    }
    builder.endObject();

    try (XContentParser parser = XContentHelper.createParser(NamedXContentRegistry.EMPTY,
        DeprecationHandler.THROW_UNSUPPORTED_OPERATION, BytesReference.bytes(builder), builder.contentType())) {
        parser.nextToken();
        List<String> insertPaths = XContentTestUtils.getInsertPaths(parser, new Stack<>());
        assertEquals(5, insertPaths.size());
        assertThat(insertPaths, hasItem(equalTo("")));
        assertThat(insertPaths, hasItem(equalTo("list1.2")));
        assertThat(insertPaths, hasItem(equalTo("list1.4")));
        assertThat(insertPaths, hasItem(equalTo("inner1")));
        assertThat(insertPaths, hasItem(equalTo("inner1.inn\\.er2")));
    }
}

Java Code Examples for org.elasticsearch.common.xcontent.json.JsonXContent#contentBuilder()