fr.gael.dhus.database.liquibase.CopyProductImages.java Source code

Introduction

Here is the source code for fr.gael.dhus.database.liquibase.CopyProductImages.java

Source

/*
 * 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 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.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import java.sql.Blob;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.Statement;

public class CopyProductImages implements CustomTaskChange {
    private static Log logger = LogFactory.getLog(CopyProductImages.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 {
        PreparedStatement products = null;
        ResultSet products_res = null;
        JdbcConnection db_connection = (JdbcConnection) database.getConnection();
        try {
            products = db_connection.prepareStatement("SELECT ID,QUICKLOOK,THUMBNAIL FROM PRODUCTS");
            products_res = products.executeQuery();
            while (products_res.next()) {
                PreparedStatement copy_blob_stmt = null;
                ResultSet generated_key_res = null;
                try {
                    Blob ql = (Blob) products_res.getObject("QUICKLOOK");
                    Blob th = (Blob) products_res.getObject("THUMBNAIL");
                    Long pid = products_res.getLong("ID");

                    // No images: add false flags
                    if ((ql == null) && (th == null)) {
                        PreparedStatement product_flags_stmt = null;
                        // Add related flags
                        try {
                            product_flags_stmt = db_connection.prepareStatement(
                                    "UPDATE PRODUCTS SET THUMBNAIL_FLAG=?,QUICKLOOK_FLAG=? " + "WHERE ID=?");
                            product_flags_stmt.setBoolean(1, false);
                            product_flags_stmt.setBoolean(2, false);
                            product_flags_stmt.setLong(3, pid);
                            product_flags_stmt.execute();
                        } finally {
                            if (product_flags_stmt != null)
                                try {
                                    product_flags_stmt.close();
                                } catch (Exception e) {
                                    logger.warn("Cannot close Statement !");
                                }
                        }
                        continue;
                    }

                    copy_blob_stmt = db_connection.prepareStatement(
                            "INSERT INTO PRODUCT_IMAGES (QUICKLOOK,THUMBNAIL) " + "VALUES (?,?)",
                            Statement.RETURN_GENERATED_KEYS);

                    copy_blob_stmt.setBlob(1, ql);
                    copy_blob_stmt.setBlob(2, th);
                    copy_blob_stmt.execute();

                    generated_key_res = copy_blob_stmt.getGeneratedKeys();
                    if (generated_key_res.next()) {
                        PreparedStatement set_product_image_id_stmt = null;
                        Long iid = generated_key_res.getLong(1);

                        // Add ProductImages "IMAGES" entry in product
                        try {
                            set_product_image_id_stmt = db_connection
                                    .prepareStatement("UPDATE PRODUCTS SET IMAGES_ID=?, THUMBNAIL_FLAG=?, "
                                            + "QUICKLOOK_FLAG=?  WHERE ID=?");
                            set_product_image_id_stmt.setLong(1, iid);
                            set_product_image_id_stmt.setBoolean(2, th != null);
                            set_product_image_id_stmt.setBoolean(3, ql != null);
                            set_product_image_id_stmt.setLong(4, pid);
                            set_product_image_id_stmt.execute();
                        } finally {
                            if (set_product_image_id_stmt != null)
                                try {
                                    set_product_image_id_stmt.close();
                                } catch (Exception e) {
                                    logger.warn("Cannot close Statement !");
                                }
                        }
                    } else {
                        logger.error("Cannot retrieve Image primary key for " + "product ID #"
                                + products_res.getLong("ID"));
                    }
                } finally {
                    if (generated_key_res != null)
                        try {
                            generated_key_res.close();
                        } catch (Exception e) {
                            logger.warn("Cannot close ResultSet !");
                        }
                    if (copy_blob_stmt != null)
                        try {
                            copy_blob_stmt.close();
                        } catch (Exception e) {
                            logger.warn("Cannot close Statement !");
                        }
                }
            }
        } catch (Exception e) {
            throw new CustomChangeException("Cannot move Blobs from product", e);
        } finally {
            if (products_res != null) {
                try {
                    products_res.close();
                } catch (Exception e) {
                    logger.warn("Cannot close ResultSet !");
                }
            }
            if (products != null) {
                try {
                    products.close();
                } catch (Exception e) {
                    logger.warn("Cannot close Statement !");
                }
            }
            //if (db_connection!=null) try { db_connection.close (); }
            // catch (Exception e) {}
        }
    }
}