001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common Development
008: * and Distribution License("CDDL") (collectively, the "License"). You
009: * may not use this file except in compliance with the License. You can obtain
010: * a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html
011: * or glassfish/bootstrap/legal/LICENSE.txt. See the License for the specific
012: * language governing permissions and limitations under the License.
013: *
014: * When distributing the software, include this License Header Notice in each
015: * file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.
016: * Sun designates this particular file as subject to the "Classpath" exception
017: * as provided by Sun in the GPL Version 2 section of the License file that
018: * accompanied this code. If applicable, add the following below the License
019: * Header, with the fields enclosed by brackets [] replaced by your own
020: * identifying information: "Portions Copyrighted [year]
021: * [name of copyright owner]"
022: *
023: * Contributor(s):
024: *
025: * If you wish your version of this file to be governed by only the CDDL or
026: * only the GPL Version 2, indicate your decision by adding "[Contributor]
027: * elects to include this software in this distribution under the [CDDL or GPL
028: * Version 2] license." If you don't indicate a single choice of license, a
029: * recipient has the option to distribute your version of this file under
030: * either the CDDL, the GPL Version 2 or to extend the choice of license to
031: * its licensees as provided above. However, if you add GPL Version 2 code
032: * and therefore, elected the GPL Version 2 license, then the option applies
033: * only if the new code is made subject to such option by the copyright
034: * holder.
035: */
036: package org.jvnet.mimepull;
037:
038: import java.io.File;
039: import java.io.IOException;
040:
041: /**
042: * Configuration for MIME message parsing and storing.
043: *
044: * @author Jitendra Kotamraju
045: */
046: public class MIMEConfig {
047:
048: private static final int DEFAULT_CHUNK_SIZE = 8192;
049: private static final long DEFAULT_MEMORY_THRESHOLD = 1048576L;
050: private static final String DEFAULT_FILE_PREFIX = "MIME";
051:
052: // Parses the entire message eagerly
053: boolean parseEagerly;
054:
055: // Approximate Chunk size
056: int chunkSize;
057:
058: // Maximum in-memory data per attachment
059: long memoryThreshold;
060:
061: // Do not store to disk
062: boolean onlyMemory;
063:
064: // temp Dir to store large files
065: File tempDir;
066: String prefix;
067: String suffix;
068:
069: private MIMEConfig(boolean parseEagerly, int chunkSize,
070: long inMemoryThreshold, String dir, String prefix,
071: String suffix) {
072: this .parseEagerly = parseEagerly;
073: this .chunkSize = chunkSize;
074: this .memoryThreshold = inMemoryThreshold;
075: this .prefix = prefix;
076: this .suffix = suffix;
077: setDir(dir);
078: }
079:
080: public MIMEConfig() {
081: this (false, DEFAULT_CHUNK_SIZE, DEFAULT_MEMORY_THRESHOLD, null,
082: DEFAULT_FILE_PREFIX, null);
083: }
084:
085: boolean isParseEagerly() {
086: return parseEagerly;
087: }
088:
089: public void setParseEagerly(boolean parseEagerly) {
090: this .parseEagerly = parseEagerly;
091: }
092:
093: int getChunkSize() {
094: return chunkSize;
095: }
096:
097: void setChunkSize(int chunkSize) {
098: this .chunkSize = chunkSize;
099: }
100:
101: long getMemoryThreshold() {
102: return memoryThreshold;
103: }
104:
105: /**
106: * If the attachment is greater than the threshold, it is
107: * written to the disk.
108: *
109: * @param memoryThreshold no of bytes per attachment
110: * if -1, then the whole attachment is kept in memory
111: */
112: public void setMemoryThreshold(long memoryThreshold) {
113: this .memoryThreshold = memoryThreshold;
114: }
115:
116: boolean isOnlyMemory() {
117: return memoryThreshold == -1L;
118: }
119:
120: File getTempDir() {
121: return tempDir;
122: }
123:
124: String getTempFilePrefix() {
125: return prefix;
126: }
127:
128: String getTempFileSuffix() {
129: return suffix;
130: }
131:
132: /**
133: * @param dir
134: */
135: public void setDir(String dir) {
136: if (tempDir == null && dir != null && !dir.equals("")) {
137: tempDir = new File(dir);
138: }
139: }
140:
141: /**
142: * Validates if it can create temporary files. Otherwise, it stores
143: * attachment contents in memory.
144: */
145: public void validate() {
146: if (!isOnlyMemory()) {
147: try {
148: File tempFile = (tempDir == null) ? File
149: .createTempFile(prefix, suffix) : File
150: .createTempFile(prefix, suffix, tempDir);
151: tempFile.delete();
152: } catch (Exception ioe) {
153: memoryThreshold = -1L; // whole attachment will be in-memory
154: }
155: }
156: }
157:
158: }
|