/*
* 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.io.File;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
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;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
public class StoreProductQLAndThumbSizes implements CustomTaskChange
{
private static final Logger LOGGER = LogManager.getLogger(StoreProductQLAndThumbSizes.class);
@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
{
// contentStart contentEnd ingestionDate download.size
JdbcConnection databaseConnection =
(JdbcConnection) database.getConnection ();
try
{
PreparedStatement getProducts =
databaseConnection
.prepareStatement ("SELECT ID, QUICKLOOK_PATH, " +
"THUMBNAIL_PATH FROM Products");
ResultSet res = getProducts.executeQuery ();
while (res.next ())
{
Long productId = (Long) res.getObject("ID");
String thumbnail = (String) res.getObject ("THUMBNAIL_PATH");
String quicklook = (String) res.getObject ("QUICKLOOK_PATH");
Long tbSize = null;
Long qlSize = null;
if (thumbnail != null)
{
File tbFile = new File (thumbnail);
if (tbFile.exists ())
{
tbSize = tbFile.length ();
}
}
if (quicklook != null)
{
File qlFile = new File (quicklook);
if (qlFile.exists ())
{
qlSize = qlFile.length ();
}
}
PreparedStatement updateProduct =
databaseConnection.prepareStatement ("UPDATE PRODUCTS SET" +
" QUICKLOOK_SIZE = " + qlSize +
", THUMBNAIL_SIZE = "+tbSize+" WHERE ID = "+productId);
updateProduct.execute ();
updateProduct.close();
}
getProducts.close ();
}
catch (Exception e)
{
LOGGER.error("Error during liquibase update " +
"'ExtractProductDatesAndDownloadSize'", e);
}
}
}