Ask Questions and Find Answers

public class UpgradeMyEntityLocalization extends UpgradeProcess {
    
    @Override
    protected void doUpgrade() throws Exception {
        
        addTableMyEntityLocalization();
        
        updateMyEntityDefaultLanguageId();
        
        updateMyEntityLocalizationFields();
        
        alter(MyEntityImpl.class, new AlterTableDropColumn("title"));
        alter(MyEntityImpl.class, new AlterTableDropColumn("description"));
    }
    
    private void addTableMyEntityLocalization() throws Exception {
        
        if (!hasTable(MyEntityLocalizationModelImpl.TABLE_NAME)) {
            
            runSQL(new DBTypeToSQLMap(MyEntityLocalizationModelImpl.TABLE_SQL_CREATE));
            
            runSQL("create unique index IX_BDAEBA79 on MyEntityNamespace_MyEntityLocalization (myEntityId, languageId[$COLUMN_LENGTH:75$]);\n");
        }
    }
    
    private void updateMyEntityDefaultLanguageId() throws Exception {
        
        if (!hasColumn(MyEntityModelImpl.TABLE_NAME, "defaultLanguageId")) {
            
            alter(MyEntityImpl.class, new AlterTableAddColumn("defaultLanguageId VARCHAR(75) null"));
        }
        
        try (PreparedStatement selectFromMyEntityPreparedStatement = connection.prepareStatement(StringBundler
                .concat("select myEntityId, groupId from ", MyEntityModelImpl.TABLE_NAME,
                        " where defaultLanguageId is null or defaultLanguageId = ''"));
                PreparedStatement updateMyEntityPreparedStatement = AutoBatchPreparedStatementUtil
                        .concurrentAutoBatch(connection,
                                "update " + MyEntityModelImpl.TABLE_NAME + " set defaultLanguageId = ? where "
                                        + "myEntityId = ?");
                ResultSet resultSet = selectFromMyEntityPreparedStatement.executeQuery()) {
            
            while (resultSet.next()) {
                
                updateMyEntityPreparedStatement
                        .setString(1, LocaleUtil.toLanguageId(PortalUtil.getSiteDefaultLocale(resultSet.getLong(2))));
                updateMyEntityPreparedStatement
                        .setLong(2, resultSet.getLong(1));
                
                updateMyEntityPreparedStatement.addBatch();
            }
            
            updateMyEntityPreparedStatement.executeBatch();
        }
    }
    
    private void updateMyEntityLocalizationFields() throws Exception {
        
        StringBundler insertIntoMyEntityLocalizationSQL = new StringBundler(3);
        
        insertIntoMyEntityLocalizationSQL.append("insert into ");
        insertIntoMyEntityLocalizationSQL.append(MyEntityLocalizationModelImpl.TABLE_NAME);
        insertIntoMyEntityLocalizationSQL
                .append(" (myEntityLocalizationId, companyId, myEntityId, ");
        insertIntoMyEntityLocalizationSQL.append("languageId, title, description) ");
        insertIntoMyEntityLocalizationSQL.append("values(?, ?, ?, ?, ?, ?)");
        
        try (PreparedStatement selectFromMyEntityPreparedStatement = connection.prepareStatement(
                "select myEntityId, companyId, title, description, defaultLanguageId from "
                        + MyEntityModelImpl.TABLE_NAME);
                PreparedStatement insertIntoMyEntityLocalizationPreparedStatement = AutoBatchPreparedStatementUtil
                        .concurrentAutoBatch(connection, insertIntoMyEntityLocalizationSQL.toString());
                ResultSet selectFromMyEntityResultSet = selectFromMyEntityPreparedStatement
                        .executeQuery()) {
            
            while (selectFromMyEntityResultSet.next()) {
                
                insertIntoMyEntityLocalizationPreparedStatement
                        .setLong(1, CounterLocalServiceUtil.increment());
                insertIntoMyEntityLocalizationPreparedStatement
                        .setLong(2, selectFromMyEntityResultSet.getLong(2));
                insertIntoMyEntityLocalizationPreparedStatement
                        .setLong(3, selectFromMyEntityResultSet.getLong(1));
                insertIntoMyEntityLocalizationPreparedStatement
                        .setString(4, selectFromMyEntityResultSet.getString(5));
                insertIntoMyEntityLocalizationPreparedStatement
                        .setString(5, selectFromMyEntityResultSet.getString(3));
                insertIntoMyEntityLocalizationPreparedStatement
                        .setString(6, selectFromMyEntityResultSet.getString(4));
                
                insertIntoMyEntityLocalizationPreparedStatement.addBatch();
            }
            
            insertIntoMyEntityLocalizationPreparedStatement.executeBatch();
        }
    }
    
}