1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.eclipse.jetty.server.handler;
20
21 import java.io.IOException;
22
23 import javax.servlet.DispatcherType;
24 import javax.servlet.ServletException;
25 import javax.servlet.http.HttpServletRequest;
26 import javax.servlet.http.HttpServletResponse;
27
28 import org.eclipse.jetty.continuation.Continuation;
29 import org.eclipse.jetty.continuation.ContinuationListener;
30 import org.eclipse.jetty.server.AsyncContinuation;
31 import org.eclipse.jetty.server.Request;
32 import org.eclipse.jetty.server.RequestLog;
33 import org.eclipse.jetty.server.Response;
34 import org.eclipse.jetty.server.Server;
35 import org.eclipse.jetty.util.component.AbstractLifeCycle;
36 import org.eclipse.jetty.util.log.Log;
37 import org.eclipse.jetty.util.log.Logger;
38
39
40
41
42
43
44
45
46 public class RequestLogHandler extends HandlerWrapper
47 {
48 private static final Logger LOG = Log.getLogger(RequestLogHandler.class);
49
50 private RequestLog _requestLog;
51
52
53
54
55
56 @Override
57 public void handle(String target, final Request baseRequest, HttpServletRequest request, final HttpServletResponse response)
58 throws IOException, ServletException
59 {
60 AsyncContinuation continuation = baseRequest.getAsyncContinuation();
61 if (!continuation.isInitial())
62 {
63 baseRequest.setDispatchTime(System.currentTimeMillis());
64 }
65
66 try
67 {
68 super.handle(target, baseRequest, request, response);
69 }
70 finally
71 {
72 if (_requestLog != null && baseRequest.getDispatcherType().equals(DispatcherType.REQUEST))
73 {
74 if (continuation.isAsync())
75 {
76 if (continuation.isInitial())
77 continuation.addContinuationListener(new ContinuationListener()
78 {
79
80 public void onTimeout(Continuation continuation)
81 {
82
83 }
84
85 public void onComplete(Continuation continuation)
86 {
87 _requestLog.log(baseRequest, (Response)response);
88 }
89 });
90 }
91 else
92 _requestLog.log(baseRequest, (Response)response);
93 }
94 }
95 }
96
97
98 public void setRequestLog(RequestLog requestLog)
99 {
100
101 try
102 {
103 if (_requestLog != null)
104 _requestLog.stop();
105 }
106 catch (Exception e)
107 {
108 LOG.warn (e);
109 }
110
111 if (getServer()!=null)
112 getServer().getContainer().update(this, _requestLog, requestLog, "logimpl",true);
113
114 _requestLog = requestLog;
115
116
117 try
118 {
119 if (isStarted() && (_requestLog != null))
120 _requestLog.start();
121 }
122 catch (Exception e)
123 {
124 throw new RuntimeException (e);
125 }
126 }
127
128
129
130
131
132 @Override
133 public void setServer(Server server)
134 {
135 if (_requestLog!=null)
136 {
137 if (getServer()!=null && getServer()!=server)
138 getServer().getContainer().update(this, _requestLog, null, "logimpl",true);
139 super.setServer(server);
140 if (server!=null && server!=getServer())
141 server.getContainer().update(this, null,_requestLog, "logimpl",true);
142 }
143 else
144 super.setServer(server);
145 }
146
147
148 public RequestLog getRequestLog()
149 {
150 return _requestLog;
151 }
152
153
154
155
156
157 @Override
158 protected void doStart() throws Exception
159 {
160 if (_requestLog==null)
161 {
162 LOG.warn("!RequestLog");
163 _requestLog=new NullRequestLog();
164 }
165 super.doStart();
166 _requestLog.start();
167 }
168
169
170
171
172
173 @Override
174 protected void doStop() throws Exception
175 {
176 super.doStop();
177 _requestLog.stop();
178 if (_requestLog instanceof NullRequestLog)
179 _requestLog=null;
180 }
181
182
183
184
185 private static class NullRequestLog extends AbstractLifeCycle implements RequestLog
186 {
187 public void log(Request request, Response response)
188 {
189 }
190 }
191
192 }