cz.cvut.kbss.ontodriver.owlapi.OwlapiDriver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ontodriver-owlapi Show documentation
OWLAPI-based implementation of the OntoDriver API
The newest version!
/*
 * JOPA
 * Copyright (C) 2024 Czech Technical University in Prague
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3.0 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library.
 */
package cz.cvut.kbss.ontodriver.owlapi;

import cz.cvut.kbss.ontodriver.Closeable;
import cz.cvut.kbss.ontodriver.Connection;
import cz.cvut.kbss.ontodriver.OntologyStorageProperties;
import cz.cvut.kbss.ontodriver.config.DriverConfigParam;
import cz.cvut.kbss.ontodriver.config.DriverConfiguration;
import cz.cvut.kbss.ontodriver.config.ConfigurationParameter;
import cz.cvut.kbss.ontodriver.exception.OntoDriverException;
import cz.cvut.kbss.ontodriver.owlapi.config.OwlapiConfigParam;
import cz.cvut.kbss.ontodriver.owlapi.connector.ConnectorFactory;
import cz.cvut.kbss.ontodriver.owlapi.exception.OwlapiDriverException;
import cz.cvut.kbss.ontodriver.owlapi.list.OwlapiLists;

import java.util.*;

class OwlapiDriver implements Closeable, ConnectionListener {

    private static final List CONFIGS = Arrays
            .asList(DriverConfigParam.AUTO_COMMIT, DriverConfigParam.MODULE_EXTRACTION_SIGNATURE,
                    DriverConfigParam.REASONER_FACTORY_CLASS,
                    OwlapiConfigParam.IRI_MAPPING_DELIMITER, OwlapiConfigParam.MAPPING_FILE_LOCATION,
                    OwlapiConfigParam.WRITE_ON_COMMIT);

    private final DriverConfiguration configuration;
    private volatile boolean open = true;

    private final ConnectorFactory connectorFactory;
    private final Set openConnections = new HashSet<>();

    OwlapiDriver(OntologyStorageProperties storageProperties, Map properties) {
        this.configuration = new DriverConfiguration(storageProperties);
        configuration.addConfiguration(properties, CONFIGS);
        this.connectorFactory = ConnectorFactory.createFactory();
    }

    @Override
    public synchronized void close() throws OntoDriverException {
        if (!open) {
            return;
        }
        for (OwlapiConnection c : openConnections) {
            try {
                c.removeListener();
                c.close();
            } catch (Exception e) {
                throw new OwlapiDriverException(e);
            }
        }
        connectorFactory.close();
        this.open = false;
    }

    @Override
    public boolean isOpen() {
        return open;
    }

    Connection acquireConnection() throws OntoDriverException {
        assert open;
        final OwlapiAdapter adapter = new OwlapiAdapter(connectorFactory.getConnector(configuration));
        final OwlapiConnection c = new OwlapiConnection(adapter);
        c.setTypes(new OwlapiTypes(adapter, c::ensureOpen, c::commitIfAuto));
        c.setProperties(new OwlapiProperties(adapter, c::ensureOpen, c::commitIfAuto));
        c.setLists(new OwlapiLists(adapter, c::ensureOpen, c::commitIfAuto));
        openConnections.add(c);
        c.setListener(this);
        return c;
    }

    synchronized void reloadData() throws OwlapiDriverException {
        assert open;
        connectorFactory.reloadData();
    }

    @Override
    public void connectionClosed(Connection connection) {
        openConnections.remove(connection);
    }
}