Quadcap Image Database

com/quadcap/app/dbimage/DbImageLoader.java

Go to the documentation of this file.

package com.quadcap.app.dbimage;

/* Copyright 2000 - 2003 Quadcap Software.  All rights reserved.
 *
 * This software is distributed under the Quadcap Free Software License.
 * This software may be used or modified for any purpose, personal or
 * commercial.  Open Source redistributions are permitted.  Commercial
 * redistribution of larger works derived from, or works which bundle
 * this software requires a "Commercial Redistribution License"; see
 * http://www.quadcap.com/purchase.
 *
 * Redistributions qualify as "Open Source" under  one of the following terms:
 *   
 *    Redistributions are made at no charge beyond the reasonable cost of
 *    materials and delivery.
 *
 *    Redistributions are accompanied by a copy of the Source Code or by an
 *    irrevocable offer to provide a copy of the Source Code for up to three
 *    years at the cost of materials and delivery.  Such redistributions
 *    must allow further use, modification, and redistribution of the Source
 *    Code under substantially the same terms as this license.
 *
 * Redistributions of source code must retain the copyright notices as they
 * appear in each source code file, these license terms, and the
 * disclaimer/limitation of liability set forth as paragraph 6 below.
 *
 * Redistributions in binary form must reproduce this Copyright Notice,
 * these license terms, and the disclaimer/limitation of liability set
 * forth as paragraph 6 below, in the documentation and/or other materials
 * provided with the distribution.
 *
 * The Software is provided on an "AS IS" basis.  No warranty is
 * provided that the Software is free of defects, or fit for a
 * particular purpose.  
 *
 * Limitation of Liability. Quadcap Software shall not be liable
 * for any damages suffered by the Licensee or any third party resulting
 * from use of the Software.
 */

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import java.util.Enumeration;
import java.util.Properties;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.Statement;
import java.sql.SQLException;

/**
 * This class loads images from the file system into a database.
 *
 * @author Stan Bailes
 */
public class DbImageLoader {
    
    public DbImageLoader() {
    }

    public void loadImages(Connection conn, String root) throws Exception {
        PreparedStatement pstmt = conn.prepareStatement(
            "insert into images values(?,-1,?,?)");
        File f = new File(root);
        try {
            loadImages(pstmt, "", f);
        } finally {
            pstmt.close();
        }
    }

    final void loadImages(PreparedStatement pstmt, String path,
                          File dir)
        throws Exception
    {
        File[] files = dir.listFiles();
        if (files != null) for (int i = 0; i < files.length; i++) {
            File f = files[i];
            String name = f.getName();
            String thispath = path + "/" + name;
            if (!f.getCanonicalPath().equals(f.getAbsolutePath())) {
                // hack: skip anything that appears to be a symbolic link...
            } else if (f.isDirectory()) {
                loadImages(pstmt, thispath, f);
            } else if (name.endsWith(".gif") ||
                       name.endsWith(".jpg") ||
                       name.endsWith(".GIF") ||
                       name.endsWith(".JPG")) {
                pstmt.clearParameters();
                pstmt.setString(1, thispath);
                FileInputStream in = new FileInputStream(f);
                
                int len = (int)f.length();
                pstmt.setInt(3, len);

                pstmt.setBinaryStream(2, in, len);
                try {
                    pstmt.execute();
                } catch (SQLException e) {
                    String state = e.getSQLState();
                    if (state != null && state.startsWith("23")) {
                        // image already there.  We could check timestamps
                        // or something...
                    } else {
                        throw e;
                    }
                }

                in.close();
            }
        }
    }
}