1 /*
2 * Copyright 2012 The Netty Project
3 *
4 * The Netty Project licenses this file to you under the Apache License,
5 * version 2.0 (the "License"); you may not use this file except in compliance
6 * with the License. You may obtain a copy of the License at:
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
12 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
13 * License for the specific language governing permissions and limitations
14 * under the License.
15 */
16 package org.jboss.netty.handler.codec.http.multipart;
17
18 import org.jboss.netty.handler.codec.http.HttpHeaders;
19
20 import java.nio.charset.Charset;
21
22 /**
23 * Default FileUpload implementation that stores file into memory.<br><br>
24 *
25 * Warning: be aware of the memory limitation.
26 */
27 public class MemoryFileUpload extends AbstractMemoryHttpData implements FileUpload {
28
29 private String filename;
30
31 private String contentType;
32
33 private String contentTransferEncoding;
34
35 public MemoryFileUpload(String name, String filename, String contentType,
36 String contentTransferEncoding, Charset charset, long size) {
37 super(name, charset, size);
38 setFilename(filename);
39 setContentType(contentType);
40 setContentTransferEncoding(contentTransferEncoding);
41 }
42
43 public HttpDataType getHttpDataType() {
44 return HttpDataType.FileUpload;
45 }
46
47 public String getFilename() {
48 return filename;
49 }
50
51 public void setFilename(String filename) {
52 if (filename == null) {
53 throw new NullPointerException("filename");
54 }
55 this.filename = filename;
56 }
57
58 @Override
59 public int hashCode() {
60 return getName().hashCode();
61 }
62
63 @Override
64 public boolean equals(Object o) {
65 if (!(o instanceof Attribute)) {
66 return false;
67 }
68 Attribute attribute = (Attribute) o;
69 return getName().equalsIgnoreCase(attribute.getName());
70 }
71
72 public int compareTo(InterfaceHttpData o) {
73 if (!(o instanceof FileUpload)) {
74 throw new ClassCastException("Cannot compare " + getHttpDataType() +
75 " with " + o.getHttpDataType());
76 }
77 return compareTo((FileUpload) o);
78 }
79
80 public int compareTo(FileUpload o) {
81 int v;
82 v = getName().compareToIgnoreCase(o.getName());
83 if (v != 0) {
84 return v;
85 }
86 // TODO should we compare size for instance ?
87 return v;
88 }
89
90 public void setContentType(String contentType) {
91 if (contentType == null) {
92 throw new NullPointerException("contentType");
93 }
94 this.contentType = contentType;
95 }
96
97 public String getContentType() {
98 return contentType;
99 }
100
101 public String getContentTransferEncoding() {
102 return contentTransferEncoding;
103 }
104
105 public void setContentTransferEncoding(String contentTransferEncoding) {
106 this.contentTransferEncoding = contentTransferEncoding;
107 }
108
109 @Override
110 public String toString() {
111 return HttpPostBodyUtil.CONTENT_DISPOSITION + ": " +
112 HttpPostBodyUtil.FORM_DATA + "; " + HttpPostBodyUtil.NAME + "=\"" + getName() +
113 "\"; " + HttpPostBodyUtil.FILENAME + "=\"" + filename + "\"\r\n" +
114 HttpHeaders.Names.CONTENT_TYPE + ": " + contentType +
115 (charset != null? "; " + HttpHeaders.Values.CHARSET + '=' + charset.name() + "\r\n" : "\r\n") +
116 HttpHeaders.Names.CONTENT_LENGTH + ": " + length() + "\r\n" +
117 "Completed: " + isCompleted() +
118 "\r\nIsInMemory: " + isInMemory();
119 }
120 }
以上源码来自： 即时通讯网 - 即时通讯开发者社区！