Java Code Examples for org.eclipse.rdf4j.sail.Sail#getConnection()

public HalyardStatsBasedStatementPatternCardinalityCalculator(Sail sail) {
	this.valueFactory = sail.getValueFactory();
    this.statsConnection = sail.getConnection();
}
 

@Test
public void singleRecord() {
    // Create the Statements that will be put by the task.
    final ValueFactory vf = SimpleValueFactory.getInstance();
    final Set<Statement> statements = Sets.newHashSet(
            vf.createStatement(
                    vf.createIRI("urn:Alice"),
                    vf.createIRI("urn:WorksAt"),
                    vf.createIRI("urn:Taco Shop"),
                    vf.createIRI("urn:graph1")),
            vf.createStatement(
                    vf.createIRI("urn:Bob"),
                    vf.createIRI("urn:TalksTo"),
                    vf.createIRI("urn:Charlie"),
                    vf.createIRI("urn:graph2")),
            vf.createStatement(
                    vf.createIRI("urn:Eve"),
                    vf.createIRI("urn:ListensTo"),
                    vf.createIRI("urn:Alice"),
                    vf.createIRI("urn:graph1")));

    // Create the task that will be tested.
    final InMemoryRyaSinkTask task = new InMemoryRyaSinkTask();

    // Setup the properties that will be used to configure the task. We don't actually need to set anything
    // here since we're always returning true for ryaInstanceExists(...) and use an in memory RDF store.
    final Map<String, String> props = new HashMap<>();

    try {
        // Start the task.
        task.start(props);

        // Put the statements as a SinkRecord.
        task.put( Collections.singleton(new SinkRecord("topic", 1, null, "key", null, statements, 0)) );

        // Flush the statements.
        task.flush(new HashMap<>());

        // Fetch the stored Statements to show they match the original set.
        final Set<Statement> fetched = new HashSet<>();

        final Sail sail = task.makeSail(props);
        try(SailConnection conn = sail.getConnection();
                CloseableIteration<? extends Statement, SailException> it = conn.getStatements(null, null, null, false)) {
            while(it.hasNext()) {
                fetched.add( it.next() );
            }
        }

        assertEquals(statements, fetched);

    } finally {
        // Stop the task.
        task.stop();
    }
}
 

@Test
public void multipleRecords() {
    // Create the Statements that will be put by the task.
    final ValueFactory vf = SimpleValueFactory.getInstance();
    final Set<Statement> batch1 = Sets.newHashSet(
            vf.createStatement(
                    vf.createIRI("urn:Alice"),
                    vf.createIRI("urn:WorksAt"),
                    vf.createIRI("urn:Taco Shop"),
                    vf.createIRI("urn:graph1")),
            vf.createStatement(
                    vf.createIRI("urn:Bob"),
                    vf.createIRI("urn:TalksTo"),
                    vf.createIRI("urn:Charlie"),
                    vf.createIRI("urn:graph2")));

    final Set<Statement> batch2 = Sets.newHashSet(
            vf.createStatement(
                    vf.createIRI("urn:Eve"),
                    vf.createIRI("urn:ListensTo"),
                    vf.createIRI("urn:Alice"),
                    vf.createIRI("urn:graph1")));

    // Create the task that will be tested.
    final InMemoryRyaSinkTask task = new InMemoryRyaSinkTask();

    // Setup the properties that will be used to configure the task. We don't actually need to set anything
    // here since we're always returning true for ryaInstanceExists(...) and use an in memory RDF store.
    final Map<String, String> props = new HashMap<>();

    try {
        // Start the task.
        task.start(props);

        // Put the statements as SinkRecords.
        final Collection<SinkRecord> records = Sets.newHashSet(
                new SinkRecord("topic", 1, null, "key", null, batch1, 0),
                new SinkRecord("topic", 1, null, "key", null, batch2, 1));
        task.put( records );

        // Flush the statements.
        task.flush(new HashMap<>());

        // Fetch the stored Statements to show they match the original set.
        final Set<Statement> fetched = new HashSet<>();

        final Sail sail = task.makeSail(props);
        try(SailConnection conn = sail.getConnection();
                CloseableIteration<? extends Statement, SailException> it = conn.getStatements(null, null, null, false)) {
            while(it.hasNext()) {
                fetched.add( it.next() );
            }
        }

        assertEquals(Sets.union(batch1, batch2), fetched);

    } finally {
        // Stop the task.
        task.stop();
    }
}
 

@Test
public void flushBetweenPuts() {
    // Create the Statements that will be put by the task.
    final ValueFactory vf = SimpleValueFactory.getInstance();
    final Set<Statement> batch1 = Sets.newHashSet(
            vf.createStatement(
                    vf.createIRI("urn:Alice"),
                    vf.createIRI("urn:WorksAt"),
                    vf.createIRI("urn:Taco Shop"),
                    vf.createIRI("urn:graph1")),
            vf.createStatement(
                    vf.createIRI("urn:Bob"),
                    vf.createIRI("urn:TalksTo"),
                    vf.createIRI("urn:Charlie"),
                    vf.createIRI("urn:graph2")));

    final Set<Statement> batch2 = Sets.newHashSet(
            vf.createStatement(
                    vf.createIRI("urn:Eve"),
                    vf.createIRI("urn:ListensTo"),
                    vf.createIRI("urn:Alice"),
                    vf.createIRI("urn:graph1")));

    // Create the task that will be tested.
    final InMemoryRyaSinkTask task = new InMemoryRyaSinkTask();

    // Setup the properties that will be used to configure the task. We don't actually need to set anything
    // here since we're always returning true for ryaInstanceExists(...) and use an in memory RDF store.
    final Map<String, String> props = new HashMap<>();

    try {
        // Start the task.
        task.start(props);

        // Put the statements with flushes between them.
        task.put( Collections.singleton(new SinkRecord("topic", 1, null, "key", null, batch1, 0)) );
        task.flush(new HashMap<>());
        task.put( Collections.singleton(new SinkRecord("topic", 1, null, "key", null, batch2, 1)) );
        task.flush(new HashMap<>());

        // Fetch the stored Statements to show they match the original set.
        final Set<Statement> fetched = new HashSet<>();

        final Sail sail = task.makeSail(props);
        try(SailConnection conn = sail.getConnection();
                CloseableIteration<? extends Statement, SailException> it = conn.getStatements(null, null, null, false)) {
            while(it.hasNext()) {
                fetched.add( it.next() );
            }
        }

        assertEquals(Sets.union(batch1, batch2), fetched);

    } finally {
        // Stop the task.
        task.stop();
    }
}
 

/**
 * Ensure a user that has not been added to the Rya instance can not interact with it.
 */
@Test
public void userNotAddedCanNotInsert() throws Exception {
    final String user = testInstance.createUniqueUser();
    final SecurityOperations secOps = super.getConnector().securityOperations();

    final RyaClient userAClient = AccumuloRyaClientFactory.build(
            new AccumuloConnectionDetails(ADMIN_USER, ADMIN_USER.toCharArray(), getInstanceName(), getZookeepers()),
            super.getClusterInstance().getCluster().getConnector(ADMIN_USER, ADMIN_USER));

    // Install the instance of Rya.
    userAClient.getInstall().install(getRyaInstanceName(), InstallConfiguration.builder().build());

    // Create the user that will not be added to the instance of Rya, but will try to scan it.
    secOps.createLocalUser(user, new PasswordToken(user));

    //Try to add a statement the Rya instance with the unauthorized user. This should fail.
    boolean securityExceptionThrown = false;

    Sail sail = null;
    SailConnection sailConn = null;
    try {
        final AccumuloRdfConfiguration userCConf = makeRyaConfig(getRyaInstanceName(), user, user, getInstanceName(), getZookeepers());
        sail = RyaSailFactory.getInstance(userCConf);
        sailConn = sail.getConnection();

        final ValueFactory vf = sail.getValueFactory();
        sailConn.addStatement(vf.createIRI("urn:Alice"), vf.createIRI("urn:talksTo"), vf.createIRI("urn:Bob"));

    } catch(final RuntimeException e) {
        final Throwable cause = e.getCause();
        if(cause instanceof AccumuloSecurityException) {
            securityExceptionThrown = true;
        }
    } finally {
        if(sailConn != null) {
            sailConn.close();
        }
        if(sail != null) {
            sail.shutDown();
        }
    }

    assertTrue(securityExceptionThrown);
}
 

/**
 * Ensure a user that has been added to the Rya instance can interact with it.
 */
@Test
public void userAddedCanInsert() throws Exception {
    final String user = testInstance.createUniqueUser();
    final SecurityOperations secOps = super.getConnector().securityOperations();

    final RyaClient userAClient = AccumuloRyaClientFactory.build(
            new AccumuloConnectionDetails(ADMIN_USER, ADMIN_USER.toCharArray(), getInstanceName(), getZookeepers()),
            super.getClusterInstance().getCluster().getConnector(ADMIN_USER, ADMIN_USER));

    // Create the user that will not be added to the instance of Rya, but will try to scan it.
    secOps.createLocalUser(user, new PasswordToken(user));

    // Install the instance of Rya.
    userAClient.getInstall().install(getRyaInstanceName(), InstallConfiguration.builder().build());

    // Add the user.
    userAClient.getAddUser().get().addUser(getRyaInstanceName(), user);

    // Try to add a statement to the Rya instance. This should succeed.
    Sail sail = null;
    SailConnection sailConn = null;

    try {
        final AccumuloRdfConfiguration userDConf = makeRyaConfig(getRyaInstanceName(), user, user, getInstanceName(), getZookeepers());
        sail = RyaSailFactory.getInstance(userDConf);
        sailConn = sail.getConnection();

        final ValueFactory vf = sail.getValueFactory();
        sailConn.begin();
        sailConn.addStatement(vf.createIRI("urn:Alice"), vf.createIRI("urn:talksTo"), vf.createIRI("urn:Bob"));
        sailConn.close();

    } finally {
        if(sailConn != null) {
            sailConn.close();
        }
        if(sail != null) {
            sail.shutDown();
        }
    }
}
 

/**
 * Ensure a user that has been removed from the Rya instance can not interact with it.
 */
@Test
public void removedUserCanNotInsert() throws Exception {
    final String adminUser = testInstance.createUniqueUser();
    final String user = testInstance.createUniqueUser();
    final SecurityOperations secOps = super.getConnector().securityOperations();

    // Create the user that will install the instance of Rya.
    secOps.createLocalUser(adminUser, new PasswordToken(adminUser));
    secOps.grantSystemPermission(adminUser, SystemPermission.CREATE_TABLE);

    final RyaClient userAClient = AccumuloRyaClientFactory.build(
            new AccumuloConnectionDetails(adminUser, adminUser.toCharArray(), getInstanceName(), getZookeepers()),
            super.getClusterInstance().getCluster().getConnector(adminUser, adminUser));

    // Create the user that will be added to the instance of Rya.
    secOps.createLocalUser(user, new PasswordToken(user));

    final RyaClient userCClient = AccumuloRyaClientFactory.build(
            new AccumuloConnectionDetails(user, user.toCharArray(), getInstanceName(), getZookeepers()),
            super.getClusterInstance().getCluster().getConnector(user, user));

    // Install the instance of Rya.
    userAClient.getInstall().install(getRyaInstanceName(), InstallConfiguration.builder().build());

    // Add userC.
    userAClient.getAddUser().get().addUser(getRyaInstanceName(), user);

    // Remove userA.
    userCClient.getRemoveUser().get().removeUser(getRyaInstanceName(), adminUser);

    // Show that userA can not insert anything.
    boolean securityExceptionThrown = false;

    Sail sail = null;
    SailConnection sailConn = null;
    try {
        final AccumuloRdfConfiguration userAConf = makeRyaConfig(getRyaInstanceName(), adminUser, adminUser, getInstanceName(), getZookeepers());
        sail = RyaSailFactory.getInstance(userAConf);
        sailConn = sail.getConnection();

        final ValueFactory vf = sail.getValueFactory();
        sailConn.addStatement(vf.createIRI("urn:Alice"), vf.createIRI("urn:talksTo"), vf.createIRI("urn:Bob"));

    } catch(final RuntimeException e) {
        final Throwable cause = e.getCause();
        if(cause instanceof AccumuloSecurityException) {
            securityExceptionThrown = true;
        }
    } finally {
        if(sailConn != null) {
            sailConn.close();
        }
        if(sail != null) {
            sail.shutDown();
        }
    }

    assertTrue(securityExceptionThrown);
}