码迷,mamicode.com
首页 > 其他好文 > 详细

OSChina 源码之 ActionServlet 控制类

时间:2014-12-17 09:10:40      阅读:299      评论:0      收藏:0      [点我收藏+]

标签:des   style   http   ar   io   color   os   sp   for   

ActionServlet 这个类在 OSChina 是负责处理表单请求的,所有以 /action 开头的请求,自己感觉还不甚满意,别拍我砖头。示例action类: FileAction
标签: OSCHINA MVC Servlet

[1].[代码] ActionServlet.java 跳至 [1]

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
package my.mvc;
 
import java.io.*;
import java.lang.reflect.*;
import java.net.URLDecoder;
import java.util.*;
 
import javax.servlet.*;
import javax.servlet.http.*;
 
import my.db.DBException;
import my.util.ResourceUtils;
 
import org.apache.commons.lang.StringUtils;
import org.apache.commons.lang.math.NumberUtils;
 
/**
 * 业务处理方法入口,URI的映射逻辑:
 * /action/xxxxxx/xxxx -> com.dlog4j.action.XxxxxxAction.xxxx(req,res)
 * <pre>
    林花谢了春红,
    太匆匆,
    无奈朝来寒雨晚来风。
 
    胭脂泪,
    相留醉,
    几时重,
    自是人生长恨水长东。
 * </pre>
 * @author Winter Lau (http://my.oschina.net/javayou)<br> */
public final class ActionServlet extends HttpServlet {
 
    private final static String ERROR_PAGE = "error_page";
    private final static String GOTO_PAGE = "goto_page";
    private final static String THIS_PAGE = "this_page";
    private final static String ERROR_MSG = "error_msg";
     
    private final static String UTF_8 = "utf-8";   
    private List<String> action_packages = null;
    private final static ThreadLocal<Boolean> g_json_enabled = new ThreadLocal<Boolean>();
     
    @Override
    public void init() throws ServletException {
        String tmp = getInitParameter("packages");
        action_packages = Arrays.asList(StringUtils.split(tmp,‘,‘));
        String initial_actions = getInitParameter("initial_actions");
        for(String action : StringUtils.split(initial_actions,‘,‘))
            try {
                _LoadAction(action);
            } catch (Exception e) {
                log("Failed to initial action : " + action, e);
            }
    }
 
    @Override
    public void destroy() {
        for(Object action : actions.values()){
            try{
                Method dm = action.getClass().getMethod("destroy");
                if(dm != null){
                    dm.invoke(action);
                    log("!!!!!!!!! " + action.getClass().getSimpleName() +
                        " destroy !!!!!!!!!");
                }
            }catch(NoSuchMethodException e){
            }catch(Exception e){
                log("Unabled to destroy action: " + action.getClass().getSimpleName(), e);
            }
        }
        super.destroy();
    }
     
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp)
            throws ServletException, IOException {
        process(RequestContext.get(), false);
    }
 
    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp)
            throws ServletException, IOException {
        process(RequestContext.get(), true);
    }
     
    /**
     * 执行Action方法并进行返回处理、异常处理
     * @param req
     * @param resp
     * @param is_post
     * @throws ServletException
     * @throws IOException
     */
    protected void process(RequestContext req, boolean is_post)
        throws ServletException, IOException
    {
        try{
            req.response().setContentType("text/html;charset=utf-8");
            if(_process(req, is_post)){
                String gp = req.param(GOTO_PAGE);
                if(StringUtils.isNotBlank(gp))
                    req.redirect(gp);
            }
        }catch(InvocationTargetException e){
            Throwable t = e.getCause();
            if(t instanceof ActionException)
                handleActionException(req, (ActionException)t);
            else if(t instanceof DBException)
                handleDBException(req, (DBException)t);
            else
                throw new ServletException(t);
        }catch(ActionException t){
            handleActionException(req, t);
        }catch(IOException e){
            throw e;
        }catch(DBException e){
            handleDBException(req, e);
        }catch(Exception e){
            log("Exception in action process.", e);
            throw new ServletException(e);
        }finally{
            g_json_enabled.remove();
        }
    }
     
    /**
     * Action业务异常
     * @param req
     * @param resp
     * @param t
     * @throws ServletException
     * @throws IOException
     */
    protected void handleActionException(RequestContext req, ActionException t)
        throws ServletException, IOException
    {      
        handleException(req, t.getMessage());
    }
     
    protected void handleDBException(RequestContext req, DBException e)
        throws ServletException, IOException
    {
        log("DBException in action process.", e);
        handleException(req, ResourceUtils.getString("error",
            "database_exception", e.getCause().getMessage()));
    }
     
    /**
     * URL解码
     *
     * @param url
     * @param charset
     * @return
     */
    private static String _DecodeURL(String url, String charset) {
        if (StringUtils.isEmpty(url))
            return "";
        try {
            return URLDecoder.decode(url, charset);
        } catch (Exception e) {
        }
        return url;
    }
 
    protected void handleException(RequestContext req, String msg)
        throws ServletException, IOException
    {
        String ep = req.param(ERROR_PAGE);
        if(StringUtils.isNotBlank(ep)){
            if(ep.charAt(0)==‘%‘)
                ep = _DecodeURL(ep, UTF_8);
            ep = ep.trim();
            if(ep.charAt(0)!=‘/‘){
                req.redirect(req.contextPath()+"/");
            }
            else{
                req.request().setAttribute(ERROR_MSG, msg);
                req.forward(ep.trim());
            }
        }
        else{
            if(g_json_enabled.get())
                req.output_json("msg", msg);
            else
                req.print(msg);
        }
    }  
     
    /**
     * 业务逻辑处理
     * @param req
     * @param resp
     * @param is_post_method
     * @throws IllegalAccessException
     * @throws InstantiationException
     * @throws IOException
     * @throws ServletException
     * @throws IOException
     * @throws InvocationTargetException
     * @throws IllegalArgumentException
     */
    private boolean _process(RequestContext req, boolean is_post)
             throws InstantiationException,
                    IllegalAccessException,
                    IOException,
                    IllegalArgumentException,
                    InvocationTargetException
    {
        String requestURI = req.uri();
        String[] parts = StringUtils.split(requestURI, ‘/‘);
        if(parts.length<2){
            req.not_found();
            return false;
        }
        //加载Action类
        Object action = this._LoadAction(parts[1]);
        if(action == null){
            req.not_found();
            return false;
        }
        String action_method_name = (parts.length>2)?parts[2]:"index";
        Method m_action = this._GetActionMethod(action, action_method_name);
        if(m_action == null){
            req.not_found();
            return false;
        }
         
        //判断action方法是否只支持POST
        if (!is_post && m_action.isAnnotationPresent(Annotation.PostMethod.class)){
            req.not_found();
            return false;
        }
         
        g_json_enabled.set(m_action.isAnnotationPresent(Annotation.JSONOutputEnabled.class));
         
        if(m_action.isAnnotationPresent(Annotation.UserRoleRequired.class)){
            IUser loginUser = req.user();
            if(loginUser == null){
                String this_page = req.param(THIS_PAGE, "");
                throw req.error("user_not_login", this_page);
            }
            if(loginUser.IsBlocked())
                throw req.error("user_blocked");
             
            Annotation.UserRoleRequired urr = (Annotation.UserRoleRequired)
                m_action.getAnnotation(Annotation.UserRoleRequired.class);
            if(loginUser.getRole() < urr.role())
                throw req.error("user_role_deny");         
        }
         
        //调用Action方法之准备参数
        int arg_c = m_action.getParameterTypes().length;
        switch(arg_c){
        case 0: // login()
            m_action.invoke(action);
            break ;
        case 1:
            m_action.invoke(action, req);
            break;
        case 2: // login(HttpServletRequest req, HttpServletResponse res)
            m_action.invoke(action, req.request(), req.response());
            break ;
        case 3: // login(HttpServletRequest req, HttpServletResponse res, String[] extParams)
            StringBuilder args = new StringBuilder();
            for(int i=3;i<parts.length;i++){
                if(StringUtils.isBlank(parts[i]))
                    continue;
                if(args.length() > 0)
                    args.append(‘/‘);
                args.append(parts[i]);
            }
            boolean isLong = m_action.getParameterTypes()[2].equals(long.class);
            m_action.invoke(action, req.request(), req.response(), isLong ? NumberUtils.toLong(
                    args.toString(), -1L) : args.toString());
            break ;
        default:
            req.not_found();
            return false;
        }
         
        return true;
    }
     
    /**
     * 加载Action类
     * @param act_name
     * @return
     * @throws InstantiationException
     * @throws IllegalAccessException
     * @throws ClassNotFoundException
     */
    protected Object _LoadAction(String act_name)
        throws InstantiationException,IllegalAccessException
    {
        Object action = actions.get(act_name);
        if(action == null){
            for(String pkg : action_packages){
                String cls = pkg + ‘.‘ + StringUtils.capitalize(act_name) + "Action";
                action = _LoadActionOfFullname(act_name, cls);
                if(action != null)
                    break;
            }
        }
        return action ;
    }
     
    private Object _LoadActionOfFullname(String act_name, String cls)
        throws IllegalAccessException, InstantiationException
    {
        Object action = null;
        try {                              
            action = Class.forName(cls).newInstance();
            try{
                Method action_init_method = action.getClass().getMethod("init", ServletContext.class);
                action_init_method.invoke(action, getServletContext());
            }catch(NoSuchMethodException e){
            }catch(InvocationTargetException excp) {
                excp.printStackTrace();
            }
            if(!actions.containsKey(act_name)){
                synchronized(actions){
                    actions.put(act_name, action);
                }
            }
        } catch (ClassNotFoundException excp) {}
        return action;
    }
     
    /**
     * 获取名为{method}的方法
     * @param action
     * @param method
     * @return
     */
    private Method _GetActionMethod(Object action, String method) {
        String key = action.getClass().getSimpleName() + ‘.‘ + method;
        Method m = methods.get(key);
        if(m != null) return m;
        for(Method m1 : action.getClass().getMethods()){
            if(m1.getModifiers()==Modifier.PUBLIC && m1.getName().equals(method)){
                synchronized(methods){
                    methods.put(key, m1);
                }
                return m1 ;
            }
        }
        return null;
    }
 
    private final static HashMap<String, Object> actions = new HashMap<String, Object>();
    private final static HashMap<String, Method> methods = new HashMap<String, Method>();
 
}

OSChina 源码之 ActionServlet 控制类

标签:des   style   http   ar   io   color   os   sp   for   

原文地址:http://blog.csdn.net/u014311051/article/details/41977107

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!