source/mircoders/entity/EntityImages.java

   1 /*
   2  * Copyright (C) 2001, 2002  The Mir-coders group
   3  *
   4  * This file is part of Mir.
   5  *
   6  * Mir is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * Mir is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with Mir; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  * In addition, as a special exception, The Mir-coders gives permission to link
  21  * the code of this program with the com.oreilly.servlet library, any library
  22  * licensed under the Apache Software License, The Sun (tm) Java Advanced
  23  * Imaging library (JAI), The Sun JIMI library (or with modified versions of
  24  * the above that use the same license as the above), and distribute linked
  25  * combinations including the two.  You must obey the GNU General Public
  26  * License in all respects for all of the code used other than the above
  27  * mentioned libraries.  If you modify this file, you may extend this exception
  28  * to your version of the file, but you are not obligated to do so.  If you do
  29  * not wish to do so, delete this exception statement from your version.
  30  */
  31
  32 package mircoders.entity;
  33
  34 import java.lang.*;
  35 import java.io.*;
  36 import java.util.*;
  37 import java.sql.*;
  38
  39 /*
  40  * kind of hack for postgres non-standard LargeObjects that Poolman
  41  * doesn't know about. see all the casting, LargeObj stuff in getIcon, getImage
  42  * at some point when postgres has normal BLOB support, this should go.
  43  */
  44 import org.postgresql.Connection;
  45 import org.postgresql.largeobject.LargeObject;
  46 import org.postgresql.largeobject.LargeObjectManager;
  47 import org.postgresql.largeobject.BlobInputStream;
  48
  49 import mir.entity.*;
  50 import mir.misc.*;
  51 import mir.storage.*;
  52
  53 /**
  54  * Diese Klasse enthält die Daten eines MetaObjekts
  55  *
  56  * @author RK, mh, mir-coders
  57  * @version $Id: EntityImages.java,v 1.12 2002/12/06 08:08:14 mh Exp $
  58  */
  59
  60
  61 public class EntityImages extends EntityUploadedMedia
  62 {
  63
  64   public EntityImages()
  65   {
  66     super();
  67   }
  68
  69   public EntityImages(StorageObject theStorage) {
  70     this();
  71     setStorage(theStorage);
  72   }
  73
  74   //
  75   // methods
  76
  77
  78   public InputStream getImage() throws StorageObjectException
  79   {
  80     theLog.printDebugInfo("--getimage started");
  81     java.sql.Connection con=null;Statement stmt=null;
  82     BlobInputStream in; InputStream img_in = null;
  83
  84     try {
  85       con = theStorageObject.getPooledCon();
  86       con.setAutoCommit(false);
  87       LargeObjectManager lom;
  88       java.sql.Connection jCon;
  89       stmt = con.createStatement();
  90       ResultSet rs = theStorageObject.executeSql(stmt,
  91           "select image_data from images where id="+getId());
  92       jCon = ((com.codestudio.sql.PoolManConnectionHandle)con)
  93            .getNativeConnection();
  94       lom = ((org.postgresql.Connection)jCon).getLargeObjectAPI();
  95       if(rs!=null) {
  96         if (rs.next()) {
  97           LargeObject lob = lom.open(rs.getInt(1));
  98           in = (BlobInputStream)lob.getInputStream();
  99           img_in = new ImageInputStream(in, con, stmt);
 100         }
 101         rs.close();
 102       }
 103     }
 104     catch (Exception e) {
 105       e.printStackTrace();
 106       theLog.printError("EntityImages -- getImage failed"+e.toString());
 107       try {
 108         con.setAutoCommit(true);
 109       } catch (Exception e2) {
 110         e.printStackTrace();
 111         theLog.printError(
 112             "EntityImages -- getImage reseting transaction mode failed"
 113             +e2.toString());
 114       }
 115       theStorageObject.freeConnection(con,stmt);
 116       throwStorageObjectException(e, "EntityImages -- getImage failed: ");
 117     }
 118     //}
 119     return img_in;
 120   }
 121
 122   public void setImage(InputStream in, String type)
 123       throws StorageObjectException {
 124
 125     if (in!=null) {
 126       java.sql.Connection con=null;PreparedStatement pstmt=null;
 127       File f = null;
 128       try {
 129
 130         theLog.printDebugInfo("settimage :: making internal representation of image");
 131
 132         File tempDir = new File(MirConfig.getProp("TempDir"));
 133         f = File.createTempFile("mir", ".tmp", tempDir);
 134         FileUtil.write(f, in);
 135         WebdbImage webdbImage= new WebdbImage(f, type);
 136         theLog.printDebugInfo("settimage :: made internal representation of image");
 137
 138         con = theStorageObject.getPooledCon();
 139         con.setAutoCommit(false);
 140         theLog.printDebugInfo("settimage :: trying to insert image");
 141
 142         // setting values
 143         LargeObjectManager lom;
 144         java.sql.Connection jCon;
 145         jCon = ((com.codestudio.sql.PoolManConnectionHandle)con)
 146              .getNativeConnection();
 147         lom = ((org.postgresql.Connection)jCon).getLargeObjectAPI();
 148         int oidImage = lom.create();
 149         int oidIcon = lom.create();
 150         LargeObject lobImage = lom.open(oidImage);
 151         LargeObject lobIcon = lom.open(oidIcon);
 152         webdbImage.setImage(lobImage.getOutputStream());
 153         webdbImage.setIcon(lobIcon.getOutputStream());
 154         lobImage.close();
 155         lobIcon.close();
 156
 157         setValueForProperty("img_height",
 158                           new Integer(webdbImage.getImageHeight()).toString());
 159         setValueForProperty("img_width",
 160                           new Integer(webdbImage.getImageWidth()).toString());
 161         setValueForProperty("icon_height",
 162                           new Integer(webdbImage.getIconHeight()).toString());
 163         setValueForProperty("icon_width",
 164                           new Integer(webdbImage.getIconWidth()).toString());
 165         setValueForProperty("image_data", new Integer(oidImage).toString());
 166         setValueForProperty("icon_data", new Integer(oidIcon).toString());
 167         update();
 168       }
 169       catch (Exception e) {
 170         throwStorageObjectException(e, "settimage :: setImage gescheitert: ");
 171       }
 172       finally {
 173         try {
 174           if (con!=null)
 175             con.setAutoCommit(true);
 176           // get rid of the temp. file
 177           f.delete();
 178         }
 179         catch (Exception e) {
 180         }
 181
 182         if (con!=null)
 183           theStorageObject.freeConnection(con,pstmt);
 184       }
 185     }
 186   }
 187
 188   /**
 189    * Takes an OutputStream as an argument and reads in the data
 190    * from the DB and writes it to the OutputStream.
 191    *
 192    * It will also take care of closing the OutputStream.
 193    */
 194   public InputStream getIcon() throws StorageObjectException
 195   {
 196     java.sql.Connection con=null;Statement stmt=null;
 197     BlobInputStream in=null;ImageInputStream img_in=null;
 198
 199     try {
 200       con = theStorageObject.getPooledCon();
 201       con.setAutoCommit(false);
 202       LargeObjectManager lom;
 203       java.sql.Connection jCon;
 204       stmt = con.createStatement();
 205       ResultSet rs = theStorageObject.executeSql(stmt,
 206           "select icon_data from images where id="+getId());
 207       jCon = ((com.codestudio.sql.PoolManConnectionHandle)con)
 208            .getNativeConnection();
 209       lom = ((org.postgresql.Connection)jCon).getLargeObjectAPI();
 210       if(rs!=null) {
 211         if (rs.next()) {
 212           LargeObject lob = lom.open(rs.getInt(1));
 213           in = (BlobInputStream)lob.getInputStream();
 214           img_in = new ImageInputStream( in, con ,stmt);
 215           //img_data = rs.getBytes(1);
 216         }
 217         rs.close();
 218       }
 219     }
 220     catch (Exception e) {
 221       e.printStackTrace();
 222       theLog.printError("EntityImages -- getIcon failed"+e.toString());
 223       try {
 224         con.setAutoCommit(true);
 225       }
 226       catch (Exception e2) {
 227         e.printStackTrace();
 228         theLog.printError(
 229             "EntityImages -- getIcon reseting transaction mode failed"
 230             +e2.toString());
 231       }
 232       theStorageObject.freeConnection(con,stmt);
 233       throwStorageObjectException(e, "EntityImages -- getIcon failed:");
 234     }
 235
 236     return img_in;
 237   }
 238
 239   /**
 240    * a small wrapper class that allows us to store the DB connection resources
 241    * that the BlobInputStream is using and free them upon closing of the stream
 242    */
 243   private class ImageInputStream extends InputStream {
 244
 245     InputStream _in;
 246     java.sql.Connection _con;
 247     Statement _stmt;
 248
 249     public ImageInputStream(BlobInputStream in, java.sql.Connection con,
 250                             Statement stmt )
 251     {
 252       _in = in;
 253       _con = con;
 254       _stmt = stmt;
 255     }
 256
 257     public void close () throws IOException {
 258       _in.close();
 259
 260       try {
 261         _con.setAutoCommit(true);
 262         theStorageObject.freeConnection(_con,_stmt);
 263       }
 264       catch (Exception e) {
 265         throw new IOException("close(): "+e.toString());
 266       }
 267     }
 268
 269     public int read() throws IOException {
 270       return _in.read();
 271     }
 272
 273   }
 274 }