Java Code Examples for org.apache.tinkerpop.gremlin.structure.io.graphson.GraphSONMapper#Builder

@Test
@SuppressWarnings("deprecation")
public void shouldFailOnMessageSerializerWithMapperIfNoGremlinServerModule() {
    org.apache.log4j.Logger logger = org.apache.log4j.Logger.getLogger(AbstractGraphSONMessageSerializerV2d0.class);
    Level previousLevel = logger.getLevel();

    // Disable temporarily logging for this test
    logger.setLevel(Level.OFF);

    GraphSONMapper.Builder builder = GraphSONMapper.build().addCustomModule(GraphSONXModuleV2d0.build().create(false));
    GraphSONMessageSerializerV2d0 graphSONMessageSerializerV2d0 = new GraphSONMessageSerializerV2d0(builder.create());

    try {
        convert("hello", graphSONMessageSerializerV2d0);
        fail("Serialization should have failed since no GremlinServerModule registered.");
    } catch (SerializationException e) {
        assertTrue(e.getMessage().contains("Could not find a type identifier for the class"));
        assertTrue(e.getCause() instanceof JsonMappingException);
        assertTrue(e.getCause().getCause() instanceof IllegalArgumentException);
    }

    // Put logger level back to its original value
    logger.setLevel(previousLevel);
}
 

@Test
public void shouldRegisterGremlinServerModuleAutomaticallyWithMapper() throws SerializationException {
    GraphSONMapper.Builder builder = GraphSONMapper.build().addCustomModule(GraphSONXModuleV2d0.build().create(false));
    GraphSONMessageSerializerV2d0 graphSONMessageSerializerV2d0 = new GraphSONMessageSerializerV2d0(builder);

    ResponseMessage rm = convert("hello", graphSONMessageSerializerV2d0);
    assertEquals(rm.getRequestId(), requestId);
    assertEquals(rm.getResult().getData(), "hello");
}
 

@Test
public void shouldRegisterGremlinServerModuleAutomaticallyWithMapper() throws SerializationException {
    GraphSONMapper.Builder builder = GraphSONMapper.build().addCustomModule(GraphSONXModuleV3d0.build().create(false));
    GraphSONMessageSerializerV3d0 graphSONMessageSerializerV3d0 = new GraphSONMessageSerializerV3d0(builder);

    ResponseMessage rm = convert("hello", graphSONMessageSerializerV3d0);
    assertEquals(rm.getRequestId(), requestId);
    assertEquals(rm.getResult().getData(), "hello");
}
 

private JsonNode getEntityNode(String json) throws IOException {
    GraphSONMapper.Builder builder = GraphSONMapper.build();
    final ObjectMapper mapper  = builder.typeInfo(TypeInfo.NO_TYPES).create().createMapper();
    return mapper.readTree(json);
}
 

@Override
GraphSONMapper.Builder configureBuilder(final GraphSONMapper.Builder builder) {
    // already set to 1.0 in AbstractGraphSONMessageSerializerV1d0
    return builder.addCustomModule(new GraphSONMessageSerializerV1d0.GremlinServerModule())
            .typeInfo(TypeInfo.PARTIAL_TYPES);
}
 

private GraphSONMapper.Builder initBuilder(final GraphSONMapper.Builder builder) {
    final GraphSONMapper.Builder b = null == builder ? GraphSONMapper.build() : builder;
    return b.addCustomModule(GraphSONXModuleV2d0.build().create(false))
            .version(GraphSONVersion.V2_0);
}
 

@Override
public void configure(final Map<String, Object> config, final Map<String, Graph> graphs) {
    final GraphSONMapper.Builder initialBuilder = initBuilder(null);
    addIoRegistries(config, initialBuilder);
    mapper = configureBuilder(initialBuilder).create().createMapper();
}
 

public AbstractGraphSONMessageSerializerV2d0(final GraphSONMapper.Builder mapperBuilder) {
    this.mapper = configureBuilder(mapperBuilder).create().createMapper();
}
 

public AbstractGraphSONMessageSerializerV2d0() {
    final GraphSONMapper.Builder builder = configureBuilder(initBuilder(null));
    mapper = builder.create().createMapper();
}
 

private GraphSONMapper.Builder initBuilder(final GraphSONMapper.Builder builder) {
    final GraphSONMapper.Builder b = null == builder ? GraphSONMapper.build() : builder;
    return b.addCustomModule(new AbstractGraphSONMessageSerializerV1d0.GremlinServerModule())
            .version(GraphSONVersion.V1_0);
}
 

@Override
public void configure(final Map<String, Object> config, final Map<String, Graph> graphs) {
    final GraphSONMapper.Builder initialBuilder = initBuilder(null);
    addIoRegistries(config, initialBuilder);
    mapper = configureBuilder(initialBuilder).create().createMapper();
}
 

public AbstractGraphSONMessageSerializerV1d0() {
    final GraphSONMapper.Builder builder = configureBuilder(initBuilder(null));
    mapper = builder.create().createMapper();
}
 

@Override
GraphSONMapper.Builder configureBuilder(final GraphSONMapper.Builder builder) {
    // already set to 2.0 in AbstractGraphSONMessageSerializerV2d0
    return builder.typeInfo(TypeInfo.PARTIAL_TYPES).addCustomModule(new GremlinServerModule());
}
 

@Override
GraphSONMapper.Builder configureBuilder(final GraphSONMapper.Builder builder) {
    // already set to 1.0 in AbstractGraphSONMessageSerializerV1d0
    return builder.addCustomModule(new GremlinServerModule())
            .typeInfo(TypeInfo.NO_TYPES);
}
 

@Override
GraphSONMapper.Builder configureBuilder(final GraphSONMapper.Builder builder) {
    // override the 2.0 in AbstractGraphSONMessageSerializerV2d0
    return builder.version(GraphSONVersion.V3_0).addCustomModule(new GremlinServerModule());
}
 

@Override
GraphSONMapper.Builder configureBuilder(final GraphSONMapper.Builder builder) {
    // already set to 2.0 in AbstractGraphSONMessageSerializerV2d0
    return builder.typeInfo(TypeInfo.PARTIAL_TYPES).addCustomModule(new GremlinServerModule());
}
 

/**
 * Create a GraphSONMessageSerializer with a provided {@link GraphSONMapper.Builder}.
 *
 * Note that to make this mapper usable in the context of request messages and responses,
 * this method will automatically register a {@link GremlinServerModule} to the provided
 * mapper.
 */
public GraphSONMessageSerializerV3d0(final GraphSONMapper.Builder mapperBuilder) {
    super(mapperBuilder);
}
 

/**
 * Create a GraphSONMessageSerializer with a provided {@link GraphSONMapper.Builder}.
 *
 * Note that to make this mapper usable in the context of request messages and responses,
 * this method will automatically register a {@link GremlinServerModule} to the provided
 * mapper.
 */
public GraphSONMessageSerializerV2d0(final GraphSONMapper.Builder mapperBuilder) {
    super(mapperBuilder);
}
 

abstract GraphSONMapper.Builder configureBuilder(final GraphSONMapper.Builder builder); 

abstract GraphSONMapper.Builder configureBuilder(final GraphSONMapper.Builder builder);