Class: com.technoetic.xplanner.upgrade.RoleDataUpgrader   ©

package com.technoetic.xplanner.upgrade;

import com.technoetic.xplanner.db.hibernate.GlobalSessionFactory;
import com.technoetic.xplanner.db.hibernate.HibernateHelper;
import com.technoetic.xplanner.domain.Project;
import com.technoetic.xplanner.domain.Role;
import com.technoetic.xplanner.domain.RoleAssociation;
import com.technoetic.xplanner.security.auth.Permission;
import net.sf.hibernate.Hibernate;
import net.sf.hibernate.HibernateException;
import net.sf.hibernate.Session;
import org.apache.log4j.Logger;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Iterator;
import java.util.List;

public class RoleDataUpgrader {
    private Logger log = Logger.getLogger(getClass());
    private static final String tableCheckError =
            "There was an unexpected error while checking for the personrole " +
            "table. This could be due to using a non-MySQL database or a JDBC driver version issue.";

    public static void main(String[] args) {
        new RoleDataUpgrader().run(args);
    }

    public void run(String[] args) {
        try {
            HibernateHelper.initializeHibernate();
            Session session = GlobalSessionFactory.get().openSession();
            List projects = session.find("from project in "+Project.class+" where project.hidden = false");
            try {
                if (isOldRoleTableExisting()) {
                    upgradeRoles(session, projects);
                    dropTable(session, "personrole");
                }
            } finally {
                session.flush();
Rate                session.connection().commit();
                session.close();
            }
        } catch (Exception e) {
            log.error("error", e);
        }

    }

    private void upgradeRoles(Session session, List projects) throws HibernateException, SQLException {
        Statement statement = session.connection().createStatement();
        ResultSet rs = null;
        try {
            rs = statement.executeQuery("select person_id,role from personrole");
            while (rs.next()) {
                for (int i = 0; i < projects.size(); i++) {
                    Project project = (Project)projects.get(i);
                    Role role = getRole(session, rs.getString("role"));
                    int personId = rs.getInt("person_id");
                    session.save(new RoleAssociation(project.getId(), personId, role.getId()));
                    // Person should have permission to read/edit their own profile
                    session.save(new Permission("system.person", 0, personId, "read%"));
                    session.save(new Permission("system.person", personId, personId, "edit%"));
                }
            }
        } catch (SQLException e) {
            throw e;
        } finally {
            statement.close();
            if (rs != null) {
                rs.close();
            }
        }
    }

    // TODO move to a utility class
    public static Role getRole(Session session, String rolename) throws HibernateException {
        List roles = session.find("from role in class " +
                Role.class.getName() + " where role.name = ?",
                rolename, Hibernate.STRING);
        Role role = null;
        Iterator roleIterator = roles.iterator();
        if (roleIterator.hasNext()) {
            role = (Role)roleIterator.next();
        }
        return role;
    }

    private void dropTable(Session session, String tableName) throws HibernateException, SQLException {
        Statement statement = session.connection().createStatement();
        try {
            statement.execute("drop table "+tableName);
        } catch (SQLException e) {
                log.error("error dropping table", e);
        } finally {
            statement.close();
        }
    }

    public boolean isOldRoleTableExisting() throws SQLException, HibernateException {
        boolean oldRoleTableExists = false;
        // Some databases (e.g. PostgreSQL) invalidate the transaction when
        // an exception is thrown. Creating a separate session isolates the
        // main test set up code from this behavior.
        Session session = GlobalSessionFactory.get().openSession();
        Statement statement = session.connection().createStatement();
        try {
            statement.execute("select * from personrole");
            oldRoleTableExists = true;
        } catch (SQLException e) {
            if (e.getMessage().indexOf("doesn't exist") == -1 &&
                    e.getMessage().indexOf("does not exist") == -1) {
                log.error(tableCheckError, e);
            }
        } finally {
            statement.close();
            session.close();
        }
        return oldRoleTableExists;
    }
}