org.vertexium.sql.SqlGraphConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vertexium-sql Show documentation
There is a newer version: 3.2.3
package org.vertexium.sql;

import com.google.common.base.Preconditions;
import org.apache.commons.dbcp2.BasicDataSource;
import org.vertexium.VertexiumSerializer;
import org.vertexium.inmemory.InMemoryEdge;
import org.vertexium.inmemory.InMemoryGraphConfiguration;
import org.vertexium.inmemory.InMemoryTableElement;
import org.vertexium.inmemory.InMemoryVertex;
import org.vertexium.sql.collections.SqlMap;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

public class SqlGraphConfiguration extends InMemoryGraphConfiguration {
    protected static final String CONFIG_JDBC_DRIVER_CLASS = "sql.jdbc.driverClass";
    protected static final String CONFIG_JDBC_URL = "sql.jdbc.url";
    protected static final String CONFIG_JDBC_USERNAME = "sql.jdbc.username";
    protected static final String CONFIG_JDBC_PASSWORD = "sql.jdbc.password";
    protected static final String KEY_COLUMN_NAME = "id";
    protected static final String VALUE_COLUMN_NAME = "object";
    protected static final String VERTEX_TABLE_NAME = "vertex";
    protected static final String EDGE_TABLE_NAME = "edge";
    protected static final String METADATA_TABLE_NAME = "metadata";
    protected static final String STREAMING_PROPERTIES_TABLE_NAME = "streaming_properties";
    protected static final String IN_VERTEX_ID_COLUMN = "in_vertex_id";
    protected static final String OUT_VERTEX_ID_COLUMN = "out_vertex_id";

    private final BasicDataSource dataSource;
    private final VertexiumSerializer serializer;

    public SqlGraphConfiguration(Map config) {
        super(config);
        String driverClass = getConfigString(config, CONFIG_JDBC_DRIVER_CLASS);
        String url = getConfigString(config, CONFIG_JDBC_URL);
        String username = getConfigString(config, CONFIG_JDBC_USERNAME);
        String password = getConfigString(config, CONFIG_JDBC_PASSWORD);
        dataSource = new BasicDataSource();
        dataSource.setDriverClassName(driverClass);
        dataSource.setUrl(url);
        dataSource.setUsername(username);
        dataSource.setPassword(password);
        serializer = createSerializer();
    }

    protected DataSource getDataSource() {
        return dataSource;
    }

    protected String tableNameWithPrefix(String tableName) {
        return getTableNamePrefix() + "_" + tableName;
    }

    protected SqlMap> newEdgeMap() {
        return new SqlMap>(
                tableNameWithPrefix(EDGE_TABLE_NAME), KEY_COLUMN_NAME, VALUE_COLUMN_NAME, dataSource, serializer) {

            @Override
            protected Map additionalColumns(String key, InMemoryTableElement value) {
                SqlTableEdge edge = (SqlTableEdge) value;
                Map columns = new HashMap<>();
                columns.put(IN_VERTEX_ID_COLUMN, edge.inVertexId());
                columns.put(OUT_VERTEX_ID_COLUMN, edge.outVertexId());
                return columns;
            }
        };
    }

    protected SqlMap> newVertexMap() {
        return new SqlMap<>(
                tableNameWithPrefix(VERTEX_TABLE_NAME), KEY_COLUMN_NAME, VALUE_COLUMN_NAME, dataSource, serializer);
    }

    protected SqlMap