Examples with JsonFactory org.apache.flink.shaded.jackson2.com.fasterxml.jackson.core.JsonFactory used on opensource projects

Example 36 with JsonFactory

use of org.apache.flink.shaded.jackson2.com.fasterxml.jackson.core.JsonFactory in project keycloak by keycloak.

the class ImportUtils method getRealmsFromStream.

public static Map<String, RealmRepresentation> getRealmsFromStream(ObjectMapper mapper, InputStream is) throws IOException {
    Map<String, RealmRepresentation> result = new HashMap<String, RealmRepresentation>();
    JsonFactory factory = mapper.getFactory();
    JsonParser parser = factory.createParser(is);
    try {
        parser.nextToken();
        if (parser.getCurrentToken() == JsonToken.START_ARRAY) {
            // Case with more realms in stream
            parser.nextToken();
            List<RealmRepresentation> realmReps = new ArrayList<RealmRepresentation>();
            while (parser.getCurrentToken() == JsonToken.START_OBJECT) {
                RealmRepresentation realmRep = parser.readValueAs(RealmRepresentation.class);
                parser.nextToken();
                // Ensure that master realm is imported first
                if (Config.getAdminRealm().equals(realmRep.getRealm())) {
                    realmReps.add(0, realmRep);
                } else {
                    realmReps.add(realmRep);
                }
            }
            for (RealmRepresentation realmRep : realmReps) {
                result.put(realmRep.getRealm(), realmRep);
            }
        } else if (parser.getCurrentToken() == JsonToken.START_OBJECT) {
            // Case with single realm in stream
            RealmRepresentation realmRep = parser.readValueAs(RealmRepresentation.class);
            result.put(realmRep.getRealm(), realmRep);
        }
    } finally {
        parser.close();
    }
    return result;
}

Example 37 with JsonFactory

use of org.apache.flink.shaded.jackson2.com.fasterxml.jackson.core.JsonFactory in project keycloak by keycloak.

the class ExportUtils method exportFederatedUsersToStream.

public static void exportFederatedUsersToStream(KeycloakSession session, RealmModel realm, List<String> usersToExport, ObjectMapper mapper, OutputStream os, ExportOptions options) throws IOException {
    JsonFactory factory = mapper.getFactory();
    JsonGenerator generator = factory.createGenerator(os, JsonEncoding.UTF8);
    try {
        if (mapper.isEnabled(SerializationFeature.INDENT_OUTPUT)) {
            generator.useDefaultPrettyPrinter();
        }
        generator.writeStartObject();
        generator.writeStringField("realm", realm.getName());
        // generator.writeStringField("strategy", strategy.toString());
        generator.writeFieldName("federatedUsers");
        generator.writeStartArray();
        for (String userId : usersToExport) {
            UserRepresentation userRep = ExportUtils.exportFederatedUser(session, realm, userId, options);
            generator.writeObject(userRep);
        }
        generator.writeEndArray();
        generator.writeEndObject();
    } finally {
        generator.close();
    }
}

Example 38 with JsonFactory

use of org.apache.flink.shaded.jackson2.com.fasterxml.jackson.core.JsonFactory in project BIMserver by opensourceBIM.

the class StreamingModel method processDownload.

private void processDownload(Long topicId) throws UserException, ServerException, PublicInterfaceNotFoundException, IfcModelInterfaceException, IOException {
    InputStream inputStream = bimServerClient.getDownloadData(topicId);
    if (inputStream == null) {
        throw new IfcModelInterfaceException("No InputStream to read from for topicId " + topicId);
    }
    InputStream downloadData = new org.bimserver.utils.CountingInputStream(inputStream) {
    };
    try {
        JsonFactory jsonFactory = new JsonFactory();
        JsonParser jp = jsonFactory.createParser(downloadData);
    // TODO implement
    } catch (Exception e) {
        throw new IfcModelInterfaceException(e);
    } finally {
        if (downloadData != null) {
            downloadData.close();
        }
    }
}

Example 39 with JsonFactory

use of org.apache.flink.shaded.jackson2.com.fasterxml.jackson.core.JsonFactory in project components by Talend.

the class FixedDatasetRuntime method getValues.

public List<IndexedRecord> getValues(int limit) {
    List<IndexedRecord> values = new ArrayList<>();
    switch(properties.format.getValue()) {
        case CSV:
            try {
                CsvRecordToIndexedRecordConverter converter = new CsvRecordToIndexedRecordConverter(getSchema());
                for (CSVRecord r : // 
                CSVFormat.RFC4180.withDelimiter(// 
                properties.getFieldDelimiter().charAt(0)).withRecordSeparator(properties.getRecordDelimiter()).parse(new StringReader(properties.values.getValue()))) values.add(converter.convertToAvro(r));
            } catch (IOException e) {
                throw LocalIOErrorCode.createCannotParseSchema(e, properties.values.getValue());
            }
            break;
        case JSON:
            ObjectMapper mapper = new ObjectMapper();
            JsonSchemaInferrer jsonSchemaInferrer = new JsonSchemaInferrer(mapper);
            JsonGenericRecordConverter converter = null;
            JsonFactory jsonFactory = new JsonFactory();
            try (StringReader r = new StringReader(properties.values.getValue())) {
                Iterator<JsonNode> value = mapper.readValues(jsonFactory.createParser(r), JsonNode.class);
                int count = 0;
                while (value.hasNext() && count++ < limit) {
                    String json = value.next().toString();
                    if (converter == null) {
                        Schema jsonSchema = jsonSchemaInferrer.inferSchema(json);
                        converter = new JsonGenericRecordConverter(jsonSchema);
                    }
                    values.add(converter.convertToAvro(json));
                }
            } catch (IOException e) {
                throw LocalIOErrorCode.createCannotParseJson(e, properties.schema.getValue(), properties.values.getValue());
            }
            break;
        case AVRO:
            Schema schema = getSchema();
            if (isRandom()) {
                GeneratorFunction<IndexedRecord> gf = (GeneratorFunction<IndexedRecord>) GeneratorFunctions.of(getSchema());
                GeneratorFunction.GeneratorContext ctx = GeneratorFunction.GeneratorContext.of(0, 0L);
                for (int i = 0; i < limit; i++) {
                    ctx.setRowId(i);
                    values.add(gf.apply(ctx));
                }
            } else {
                try (ByteArrayInputStream bais = new ByteArrayInputStream(properties.values.getValue().trim().getBytes())) {
                    JsonDecoder decoder = DecoderFactory.get().jsonDecoder(schema, bais);
                    DatumReader<IndexedRecord> reader = new GenericDatumReader<>(schema);
                    int count = 0;
                    while (count++ < limit) {
                        values.add(reader.read(null, decoder));
                    }
                } catch (EOFException e) {
                // Indicates the end of the values.
                } catch (IOException e) {
                    throw LocalIOErrorCode.createCannotParseAvroJson(e, properties.schema.getValue(), properties.values.getValue());
                }
            }
            break;
    }
    return values;
}

Example 40 with JsonFactory

use of org.apache.flink.shaded.jackson2.com.fasterxml.jackson.core.JsonFactory in project atlasdb by palantir.

the class AtlasConsoleServiceTest method fromJson.

private <T> Expectations fromJson(final T input, final Class<T> clazz) throws IOException {
    final JsonFactory factory = context.mock(JsonFactory.class);
    final JsonParser parser = context.mock(JsonParser.class);
    return new Expectations() {

        {
            oneOf(mapper).getFactory();
            will(returnValue(factory));
            oneOf(factory).createParser(QUERY);
            will(returnValue(parser));
            oneOf(parser).readValueAs(clazz);
            will(returnValue(input));
        }
    };
}