/*
* Data Hub Service (DHuS) - For Space data distribution.
* Copyright (C) 2013,2014,2015 GAEL Systems
*
* This file is part of DHuS software sources.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package fr.gael.dhus.database.liquibase;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.util.UUID;
import liquibase.change.custom.CustomTaskChange;
import liquibase.database.Database;
import liquibase.database.jvm.JdbcConnection;
import liquibase.exception.CustomChangeException;
import liquibase.exception.SetupException;
import liquibase.exception.ValidationErrors;
import liquibase.resource.ResourceAccessor;
public class GenerateUserUUIDs implements CustomTaskChange
{
@Override
public String getConfirmationMessage ()
{
return null;
}
@Override
public void setFileOpener (ResourceAccessor resource_accessor)
{
}
@Override
public void setUp () throws SetupException
{
}
@Override
public ValidationErrors validate (Database arg0)
{
return null;
}
@Override
public void execute (Database database) throws CustomChangeException
{
JdbcConnection databaseConnection =
(JdbcConnection) database.getConnection ();
try
{
PreparedStatement getUsers =
databaseConnection.prepareStatement ("SELECT ID FROM USERS");
ResultSet res = getUsers.executeQuery ();
while (res.next ())
{
String uuid = UUID.randomUUID ().toString ();
PreparedStatement updateUsers =
databaseConnection
.prepareStatement ("UPDATE USERS SET UUID = '" + uuid +
"' WHERE ID = " + res.getObject ("ID"));
updateUsers.execute ();
updateUsers.close ();
PreparedStatement updateNetworkUsage =
databaseConnection
.prepareStatement ("UPDATE NETWORK_USAGE SET USER_UUID = '" + uuid +
"' WHERE USER_ID = " + res.getObject ("ID"));
updateNetworkUsage.execute ();
updateNetworkUsage.close ();
PreparedStatement updateProducts =
databaseConnection
.prepareStatement ("UPDATE PRODUCTS SET OWNER_UUID = '" +
uuid + "' WHERE OWNER_ID = " + res.getObject ("ID"));
updateProducts.execute ();
updateProducts.close ();
PreparedStatement updateCollectionuser =
databaseConnection
.prepareStatement ("UPDATE COLLECTION_USER_AUTH SET USERS_UUID = '" +
uuid + "' WHERE USERS_ID = " + res.getObject ("ID"));
updateCollectionuser.execute ();
updateCollectionuser.close ();
PreparedStatement updateProductCarts =
databaseConnection
.prepareStatement ("UPDATE PRODUCTCARTS SET USER_UUID = '" +
uuid + "' WHERE USER_ID = " + res.getObject ("ID"));
updateProductCarts.execute ();
updateProductCarts.close ();
PreparedStatement updateProductUser =
databaseConnection
.prepareStatement ("UPDATE PRODUCT_USER_AUTH SET USERS_UUID = '" +
uuid + "' WHERE USERS_ID = " + res.getObject ("ID"));
updateProductUser.execute ();
updateProductUser.close ();
PreparedStatement updateRestrictions =
databaseConnection
.prepareStatement ("UPDATE USER_RESTRICTIONS SET USER_UUID = '" +
uuid + "' WHERE USER_ID = " + res.getObject ("ID"));
updateRestrictions.execute ();
updateRestrictions.close ();
PreparedStatement updateRoles =
databaseConnection
.prepareStatement ("UPDATE USER_ROLES SET USER_UUID = '" +
uuid + "' WHERE USER_ID = " + res.getObject ("ID"));
updateRoles.execute ();
updateRoles.close ();
}
getUsers.close ();
}
catch (Exception e)
{
e.printStackTrace ();
}
}
}