HsqlDBServiceImpl.java

package org.objectweb.jonas.db.hsqldb;

import java.io.File;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.Statement;
import java.util.Iterator;
import java.util.List;

import org.hsqldb.Server;
import org.hsqldb.ServerConstants;
import org.objectweb.jonas.common.JProp;
import org.objectweb.jonas.db.AbsDBServiceImpl;
import org.objectweb.jonas.db.AbsDBServiceImplMBean;
import org.objectweb.jonas.db.DBService;
import org.objectweb.jonas.db.User;
import org.objectweb.jonas.service.ServiceException;
import org.objectweb.util.monolog.api.BasicLevel;

public class HsqlDBServiceImpl extends AbsDBServiceImpl implements DBService, AbsDBServiceImplMBean {

    private List users = null;

    private String databaseName = null;

    private static final String DEFAULT_PORT = "9001";

    private static final int SLEEP_VALUE = 100;

    private static final int MAX_RETRY_NB = 20;

    private String portNumber = null;

    private Server server = null;

    protected void initServer(List users, String databaseName, String portNumber) {
        this.users = users;
        if (portNumber != null) {
            this.portNumber = portNumber;
        } else {
            this.portNumber = DEFAULT_PORT;
        }
        this.databaseName = databaseName;
        String jBase = JProp.getJonasBase();

        server = new Server();
        // Remove all traces if level != DEBUG
        if (!getLogger().isLoggable(BasicLevel.DEBUG)) {
            server.setLogWriter(null);
            server.setErrWriter(null);
            server.setSilent(true);
            server.setTrace(false);
            server.setLogWriter(null);
        }

        String baseDir = jBase + File.separator + "work" + File.separator + "hsqldb" + File.separator + databaseName;
        String pString = "";
        if (portNumber != null) {
            pString = ";port=" + portNumber;
        }
        String serverProps = "database.0=" + baseDir + ";dbname.0=" + databaseName + pString;
        server.putPropertiesFromString(serverProps);

    }

    protected void doStart() throws ServiceException {
        super.doStart();
        getLogger().log(BasicLevel.INFO, "Starting " + server.getProductName() + " " + server.getProductVersion() + " on port " + portNumber);
        getLogger().log(BasicLevel.DEBUG, "serverState=" + server.getState());
        server.start();

        // Wait the start
        int retryNb = 0;
        while (server.getState() != ServerConstants.SERVER_STATE_ONLINE) {
            try {
                Thread.sleep(SLEEP_VALUE);
            } catch (InterruptedException ie) {
                getLogger().log(BasicLevel.ERROR, "Can't wait that the service is online :" + ie.getMessage());
            }
            // Error if server state is "SHUTDOWN" during a long period
            // Maybe strange but 'SHUTDOWN' state seems to be an intermediate state during startup
            retryNb++;
            if (server.getState() == ServerConstants.SERVER_STATE_SHUTDOWN && retryNb >= MAX_RETRY_NB) {
                    throw new ServiceException("The server was shutdown. Enable the traces in trace.properties file to see why the service has failed.");
            }
            getLogger().log(BasicLevel.DEBUG, "retry=" + retryNb + ", serverState=" + server.getState());
        }

        getLogger().log(BasicLevel.INFO, server.getProductName() + " started.");
        Connection conn = null;
        Statement st = null;
        try {

            Class.forName("org.hsqldb.jdbcDriver");
            conn = DriverManager.getConnection("jdbc:hsqldb:hsql://localhost:" + portNumber + "/" + databaseName, "sa",
                    "");
            st = conn.createStatement();
        } catch (Exception e) {
            getLogger().log(BasicLevel.ERROR, e.getMessage());
        }

        // Drop users before recreating it
        User user = null;
        String userName = null;
        String password = null;
        ResultSet rs = null;
        for (Iterator it = users.iterator(); it.hasNext();) {
            user = (User) it.next();
            try {
                password = user.getPassword();
                userName = user.getUserName();
                getLogger().log(BasicLevel.INFO,
                        "Dropping and adding user '" + userName + "' with password '" + password + "'.");
                try {
                    rs = st.executeQuery("DROP USER " + userName);
                } catch (Exception ee) {
                    getLogger().log(BasicLevel.DEBUG, "User doesn't exists");
                }
                rs = st.executeQuery("Create USER " + userName + " PASSWORD " + password + " ADMIN");
                rs.close();
            } catch (Exception e) {
                getLogger().log(BasicLevel.ERROR, "Error while creating/adding user : '" + e.getMessage() + "'.");
            }

        }

        try {
            st.close();
        } catch (Exception e) {
            getLogger().log(BasicLevel.DEBUG, "Error while closing statement object : '" + e.getMessage() + "'.");
        }

    }

    protected void doStop() throws ServiceException {
        super.doStop();

        server.stop();
    }

}

---