1   /**
    2    * Licensed to the Apache Software Foundation (ASF) under one or more
    3    * contributor license agreements.  See the NOTICE file distributed with
    4    * this work for additional information regarding copyright ownership.
    5    * The ASF licenses this file to You under the Apache License, Version 2.0
    6    * (the "License"); you may not use this file except in compliance with
    7    * the License.  You may obtain a copy of the License at
    8    *
    9    *      http://www.apache.org/licenses/LICENSE-2.0
   10    *
   11    * Unless required by applicable law or agreed to in writing, software
   12    * distributed under the License is distributed on an "AS IS" BASIS,
   13    * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   14    * See the License for the specific language governing permissions and
   15    * limitations under the License.
   16    */
   17   package org.apache.activemq.blob;
   18   
   19   import java.io.File;
   20   import java.io.IOException;
   21   import java.io.InputStream;
   22   import java.net.URL;
   23   
   24   import javax.jms.JMSException;
   25   
   26   import org.apache.activemq.command.ActiveMQBlobMessage;
   27   
   28   /**
   29    * A helper class to represent a required upload of a BLOB to some remote URL
   30    * 
   31    * @version $Revision: $
   32    */
   33   public class BlobUploader {
   34   
   35       private BlobTransferPolicy blobTransferPolicy;
   36       private File file;
   37       private InputStream in;
   38   
   39       public BlobUploader(BlobTransferPolicy blobTransferPolicy, InputStream in) {
   40           this.blobTransferPolicy = blobTransferPolicy;
   41           this.in = in;
   42       }
   43   
   44       public BlobUploader(BlobTransferPolicy blobTransferPolicy, File file) {
   45           this.blobTransferPolicy = blobTransferPolicy;
   46           this.file = file;
   47       }
   48   
   49       public URL upload(ActiveMQBlobMessage message) throws JMSException, IOException {
   50           if (file != null) {
   51               return getStrategy().uploadFile(message, file);
   52           } else {
   53               return getStrategy().uploadStream(message, in);
   54           }
   55       }
   56   
   57       public BlobTransferPolicy getBlobTransferPolicy() {
   58           return blobTransferPolicy;
   59       }
   60   
   61       public BlobUploadStrategy getStrategy() {
   62           return getBlobTransferPolicy().getUploadStrategy();
   63       }
   64   }