add sessionStat support

2023-03-04 19:17:16 +00:00 · 2012-08-13 08:15:50 +08:00
parent 41120f6ecb
commit 3193ae8d2d
7 changed files with 361 additions and 48 deletions
@@ -0,0 +1,52 @@
+package com.alibaba.druid.bvt.support.http;
+
+import junit.framework.TestCase;
+
+import org.junit.Assert;
+import org.springframework.mock.web.MockFilterChain;
+import org.springframework.mock.web.MockFilterConfig;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+import org.springframework.mock.web.MockServletContext;
+
+import com.alibaba.druid.filter.stat.StatFilterContext;
+import com.alibaba.druid.support.http.WebStatFilter;
+import com.alibaba.druid.support.http.stat.WebAppStat;
+import com.alibaba.druid.support.http.stat.WebAppStatManager;
+
+public class WebStatFilterTest extends TestCase {
+
+    public void test_sessionStatDisable() throws Exception {
+        MockServletContext servletContext = new MockServletContext();
+
+        MockFilterConfig filterConfig = new MockFilterConfig(servletContext);
+        filterConfig.addInitParameter(WebStatFilter.PARAM_NAME_SESSION_STAT_ENABLE, "false");
+
+        WebStatFilter filter = new WebStatFilter();
+
+        Assert.assertNull(filter.getWebAppStat());
+
+        filter.init(filterConfig);
+        
+        WebAppStat appStat = filter.getWebAppStat();
+
+        Assert.assertFalse(filter.isSessionStatEnable());
+        Assert.assertTrue(WebAppStatManager.getInstance().getWebAppStatSet().contains(appStat));
+        Assert.assertTrue(StatFilterContext.getInstance().getListeners().contains(filter.getStatFilterContextListener()));
+
+        MockHttpServletRequest request = new MockHttpServletRequest();
+        MockHttpServletResponse response = new MockHttpServletResponse();
+        MockFilterChain chain = new MockFilterChain();
+
+        Assert.assertNull(filter.getSessionStat(request));
+
+        filter.doFilter(request, response, chain);
+
+        Assert.assertEquals(0, appStat.getSessionStatDataList().size());
+
+        filter.destroy();
+        
+        Assert.assertFalse(WebAppStatManager.getInstance().getWebAppStatSet().contains(appStat));
+        Assert.assertFalse(StatFilterContext.getInstance().getListeners().contains(filter.getStatFilterContextListener()));
+    }
+}
@@ -0,0 +1,150 @@
+package com.alibaba.druid.bvt.support.http;
+
+import org.junit.Assert;
+import org.springframework.mock.web.MockFilterChain;
+import org.springframework.mock.web.MockFilterConfig;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+import org.springframework.mock.web.MockHttpSession;
+import org.springframework.mock.web.MockServletContext;
+
+import com.alibaba.druid.support.http.WebStatFilter;
+import com.alibaba.druid.support.http.stat.WebAppStat;
+
+import junit.framework.TestCase;
+
+public class WebStatFilterTest2 extends TestCase {
+
+    public void test_lru() throws Exception {
+        MockServletContext servletContext = new MockServletContext();
+
+        MockFilterConfig filterConfig = new MockFilterConfig(servletContext);
+        filterConfig.addInitParameter(WebStatFilter.PARAM_NAME_SESSION_STAT_MAX_COUNT, "3");
+
+        WebStatFilter filter = new WebStatFilter();
+
+        filter.init(filterConfig);
+
+        WebAppStat appStat = filter.getWebAppStat();
+
+        Assert.assertEquals(3, filter.getSessionStatMaxCount());
+        Assert.assertEquals(0, appStat.getSessionStatDataList().size());
+
+        final MockHttpSession session_0 = new MockHttpSession(servletContext);
+        final MockHttpSession session_1 = new MockHttpSession(servletContext);
+        final MockHttpSession session_2 = new MockHttpSession(servletContext);
+        final MockHttpSession session_3 = new MockHttpSession(servletContext);
+        final MockHttpSession session_4 = new MockHttpSession(servletContext);
+
+        // 第一个session请求2次
+        {
+            MockHttpServletRequest request = new MockHttpServletRequest();
+            MockHttpServletResponse response = new MockHttpServletResponse();
+            MockFilterChain chain = new MockFilterChain() {
+
+                public void doFilter(javax.servlet.ServletRequest request, javax.servlet.ServletResponse response) {
+                    ((MockHttpServletRequest) request).setSession(session_0);
+                };
+            };
+
+            filter.doFilter(request, response, chain);
+
+            Assert.assertEquals(1, appStat.getSessionStatDataList().size());
+            Assert.assertEquals(1, appStat.getSessionStat(session_0.getId()).getRequestCount());
+        }
+        {
+            MockHttpServletRequest request = new MockHttpServletRequest();
+            MockHttpServletResponse response = new MockHttpServletResponse();
+            MockFilterChain chain = new MockFilterChain() {
+                
+                public void doFilter(javax.servlet.ServletRequest request, javax.servlet.ServletResponse response) {
+                    ((MockHttpServletRequest) request).setSession(session_0);
+                };
+            };
+            
+            filter.doFilter(request, response, chain);
+            
+            Assert.assertEquals(1, appStat.getSessionStatDataList().size());
+            Assert.assertEquals(2, appStat.getSessionStat(session_0.getId()).getRequestCount());
+        }
+        
+        
+        // 第2个sesion请求1次
+        {
+            MockHttpServletRequest request = new MockHttpServletRequest();
+            MockHttpServletResponse response = new MockHttpServletResponse();
+            MockFilterChain chain = new MockFilterChain() {
+
+                public void doFilter(javax.servlet.ServletRequest request, javax.servlet.ServletResponse response) {
+                    ((MockHttpServletRequest) request).setSession(session_1);
+                };
+            };
+
+            filter.doFilter(request, response, chain);
+
+            Assert.assertEquals(2, appStat.getSessionStatDataList().size());
+            Assert.assertEquals(2, appStat.getSessionStat(session_0.getId()).getRequestCount());
+            Assert.assertEquals(1, appStat.getSessionStat(session_1.getId()).getRequestCount());
+        }
+        
+        // 第3个sesion请求1次
+        {
+            MockHttpServletRequest request = new MockHttpServletRequest();
+            MockHttpServletResponse response = new MockHttpServletResponse();
+            MockFilterChain chain = new MockFilterChain() {
+                
+                public void doFilter(javax.servlet.ServletRequest request, javax.servlet.ServletResponse response) {
+                    ((MockHttpServletRequest) request).setSession(session_2);
+                };
+            };
+            
+            filter.doFilter(request, response, chain);
+            
+            Assert.assertEquals(3, appStat.getSessionStatDataList().size());
+            Assert.assertEquals(2, appStat.getSessionStat(session_0.getId()).getRequestCount());
+            Assert.assertEquals(1, appStat.getSessionStat(session_1.getId()).getRequestCount());
+            Assert.assertEquals(1, appStat.getSessionStat(session_2.getId()).getRequestCount());
+        }
+        
+        // 第4个sesion请求1次
+        {
+            MockHttpServletRequest request = new MockHttpServletRequest();
+            MockHttpServletResponse response = new MockHttpServletResponse();
+            MockFilterChain chain = new MockFilterChain() {
+                
+                public void doFilter(javax.servlet.ServletRequest request, javax.servlet.ServletResponse response) {
+                    ((MockHttpServletRequest) request).setSession(session_3);
+                };
+            };
+            
+            filter.doFilter(request, response, chain);
+            
+            Assert.assertEquals(3, appStat.getSessionStatDataList().size());
+            Assert.assertNull(appStat.getSessionStat(session_0.getId()));
+            Assert.assertEquals(1, appStat.getSessionStat(session_1.getId()).getRequestCount());
+            Assert.assertEquals(1, appStat.getSessionStat(session_2.getId()).getRequestCount());
+            Assert.assertEquals(1, appStat.getSessionStat(session_3.getId()).getRequestCount());
+        }
+        
+        // 第5个sesion请求1次
+        {
+            MockHttpServletRequest request = new MockHttpServletRequest();
+            MockHttpServletResponse response = new MockHttpServletResponse();
+            MockFilterChain chain = new MockFilterChain() {
+                
+                public void doFilter(javax.servlet.ServletRequest request, javax.servlet.ServletResponse response) {
+                    ((MockHttpServletRequest) request).setSession(session_4);
+                };
+            };
+            
+            filter.doFilter(request, response, chain);
+            
+            Assert.assertEquals(3, appStat.getSessionStatDataList().size());
+            Assert.assertNull(appStat.getSessionStat(session_0.getId()));
+            Assert.assertNull(appStat.getSessionStat(session_1.getId()));
+            Assert.assertEquals(1, appStat.getSessionStat(session_2.getId()).getRequestCount());
+            Assert.assertEquals(1, appStat.getSessionStat(session_3.getId()).getRequestCount());
+            Assert.assertEquals(1, appStat.getSessionStat(session_4.getId()).getRequestCount());
+        }
+    }
+}