org.umlg.sqlg.test.batch.TestBatchTemporaryVertex Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sqlg-test Show documentation

There is a newer version: 3.1.1

package org.umlg.sqlg.test.batch;

import org.apache.commons.lang3.time.StopWatch;
import org.apache.tinkerpop.gremlin.structure.T;
import org.junit.Assert;
import org.junit.Test;
import org.umlg.sqlg.test.BaseTest;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * @author Pieter Martin (https://github.com/pietermartin)
 * Date: 2017/07/27
 */
public class TestBatchTemporaryVertex extends BaseTest {

    @Test
    public void testBatchTempVertex() throws SQLException {
        int INSERT_COUNT = 1_000_000;
        StopWatch stopWatch = new StopWatch();
        stopWatch.start();
        this.sqlgGraph.tx().normalBatchModeOn();
        for (int i = 0; i < INSERT_COUNT; i++) {
            this.sqlgGraph.addTemporaryVertex(T.label, "A", "name", "halo");
        }
        this.sqlgGraph.tx().flush();
        stopWatch.stop();
        System.out.println(stopWatch.toString());
        int count = 0;
        Connection conn = this.sqlgGraph.tx().getConnection();
        String sql = "select * from ";
        if (this.sqlgGraph.getSqlDialect().needsTemporaryTableSchema()) {
            sql += this.sqlgGraph.getSqlDialect().maybeWrapInQoutes(this.sqlgGraph.getSqlDialect().getPublicSchema()) +
                    ".";
        }
        if (!this.sqlgGraph.getSqlDialect().needsTemporaryTablePrefix()) {
            sql += this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("V_A");
        } else {
            sql += this.sqlgGraph.getSqlDialect().maybeWrapInQoutes(
                    this.sqlgGraph.getSqlDialect().temporaryTablePrefix() +
                    "V_A");
        }
        try (PreparedStatement s = conn.prepareStatement(sql)) {
            ResultSet resultSet = s.executeQuery();
            while (resultSet.next()) {
                count++;
                Assert.assertEquals("halo", resultSet.getString(2));
            }
        }
        Assert.assertEquals(INSERT_COUNT, count);
    }

}