1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23 package org.apache.struts.webapp.example.memory;
24
25
26 import java.io.BufferedInputStream;
27 import java.io.BufferedOutputStream;
28 import java.io.File;
29 import java.io.FileOutputStream;
30 import java.io.InputStream;
31 import java.util.ArrayList;
32 import javax.servlet.ServletException;
33 import org.apache.commons.logging.Log;
34 import org.apache.commons.logging.LogFactory;
35 import org.apache.struts.action.ActionServlet;
36 import org.apache.struts.action.PlugIn;
37 import org.apache.struts.config.ModuleConfig;
38 import org.apache.struts.util.LabelValueBean;
39 import org.apache.struts.webapp.example.Constants;
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59 public final class MemoryDatabasePlugIn implements PlugIn {
60
61
62
63
64
65
66
67
68 private MemoryUserDatabase database = null;
69
70
71
72
73
74 private Log log = LogFactory.getLog(this.getClass());
75
76
77
78
79
80 private ActionServlet servlet = null;
81
82
83
84
85
86
87
88
89
90 private String pathname = "/WEB-INF/database.xml";
91
92 public String getPathname() {
93 return (this.pathname);
94 }
95
96 public void setPathname(String pathname) {
97 this.pathname = pathname;
98 }
99
100
101
102
103
104
105
106
107
108 public void destroy() {
109
110 log.info("Finalizing memory database plug in");
111
112 if (database != null) {
113 try {
114 database.close();
115 } catch (Exception e) {
116 log.error("Closing memory database", e);
117 }
118 }
119
120 servlet.getServletContext().removeAttribute(Constants.DATABASE_KEY);
121 database = null;
122 servlet = null;
123 database = null;
124
125 }
126
127
128
129
130
131
132
133
134
135
136 public void init(ActionServlet servlet, ModuleConfig config)
137 throws ServletException {
138
139 log.info("Initializing memory database plug in from '" +
140 pathname + "'");
141
142
143 this.servlet = servlet;
144
145
146 database = new MemoryUserDatabase();
147 try {
148 String path = calculatePath();
149 if (log.isDebugEnabled()) {
150 log.debug(" Loading database from '" + path + "'");
151 }
152 database.setPathname(path);
153 database.open();
154 } catch (Exception e) {
155 log.error("Opening memory database", e);
156 throw new ServletException("Cannot load database from '" +
157 pathname + "'", e);
158 }
159
160
161 servlet.getServletContext().setAttribute(Constants.DATABASE_KEY,
162 database);
163
164
165 setupCache(servlet, config);
166
167 }
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183 protected void setupCache(ActionServlet servlet, ModuleConfig config) {
184
185
186 ArrayList serverTypes = new ArrayList();
187 serverTypes.add(new LabelValueBean("IMAP Protocol", "imap"));
188 serverTypes.add(new LabelValueBean("POP3 Protocol", "pop3"));
189 servlet.getServletContext().setAttribute("serverTypes", serverTypes);
190
191 }
192
193
194
195
196
197
198
199
200
201
202
203
204
205 private String calculatePath() throws Exception {
206
207
208 String path = servlet.getServletContext().getRealPath(pathname);
209 if (path != null) {
210 return (path);
211 }
212
213
214 File dir = (File)
215 servlet.getServletContext().getAttribute
216 ("javax.servlet.context.tempdir");
217 File file = new File(dir, "struts-example-database.xml");
218 if (file.exists()) {
219 return (file.getAbsolutePath());
220 }
221
222
223 InputStream is =
224 servlet.getServletContext().getResourceAsStream(pathname);
225 BufferedInputStream bis = new BufferedInputStream(is, 1024);
226 FileOutputStream os =
227 new FileOutputStream(file);
228 BufferedOutputStream bos = new BufferedOutputStream(os, 1024);
229 byte buffer[] = new byte[1024];
230 while (true) {
231 int n = bis.read(buffer);
232 if (n <= 0) {
233 break;
234 }
235 bos.write(buffer, 0, n);
236 }
237 bos.close();
238 bis.close();
239 return (file.getAbsolutePath());
240
241 }
242
243
244 }