forked from fanruan/finekit
Mars.Ma
5 years ago
5 changed files with 276 additions and 21 deletions
@ -0,0 +1,36 @@ |
|||||||
|
package com.fanruan.api.net; |
||||||
|
|
||||||
|
import com.fr.general.CloudCenter; |
||||||
|
import org.jetbrains.annotations.NotNull; |
||||||
|
import org.jetbrains.annotations.Nullable; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author richie |
||||||
|
* @version 10.0 |
||||||
|
* Created by richie on 2019-08-30 |
||||||
|
* 云中心交互工具类 |
||||||
|
*/ |
||||||
|
public class CloudKit { |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* 从云中心获取值 |
||||||
|
* |
||||||
|
* @param kind 键 |
||||||
|
* @param defaultSiteUrl 默认值 |
||||||
|
* @return 键对应的值,如果没有值则返回默认值 |
||||||
|
*/ |
||||||
|
public @Nullable String acquireUrlByKind(@NotNull String kind, @NotNull String defaultSiteUrl) { |
||||||
|
return CloudCenter.getInstance().acquireUrlByKind(kind, defaultSiteUrl); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 从云中心获取值 |
||||||
|
* |
||||||
|
* @param kind 键 |
||||||
|
* @return 键对应的值,如果没有值则返回空 |
||||||
|
*/ |
||||||
|
public @Nullable String acquireUrlByKind(@NotNull String kind) { |
||||||
|
return CloudCenter.getInstance().acquireUrlByKind(kind); |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,47 @@ |
|||||||
|
package com.fanruan.api.session; |
||||||
|
|
||||||
|
import com.fr.stable.script.CalculatorProvider; |
||||||
|
import com.fr.stable.web.WebContextProvider; |
||||||
|
import com.fr.web.session.SessionIDInfo; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author richie |
||||||
|
* @version 10.0 |
||||||
|
* Created by richie on 2019-08-30 |
||||||
|
*/ |
||||||
|
class MockSession extends SessionIDInfo { |
||||||
|
|
||||||
|
MockSession(String sessionID) { |
||||||
|
this.sessionID = sessionID; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public String getWebTitle() { |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public String getRelativePath() { |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public WebContextProvider getWebContext() { |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public Object resolveVariable(Object var, CalculatorProvider cal) { |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public Object getParameterValue(String parameterName) { |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void release() { |
||||||
|
|
||||||
|
} |
||||||
|
} |
@ -1,18 +1,186 @@ |
|||||||
package com.fanruan.api.session; |
package com.fanruan.api.session; |
||||||
|
|
||||||
import com.fanruan.api.Prepare; |
import com.fanruan.api.Prepare; |
||||||
|
import com.fr.general.web.ParameterConstants; |
||||||
|
import com.fr.invoke.Reflect; |
||||||
|
import com.fr.stable.StringUtils; |
||||||
|
import com.fr.stable.web.SessionProvider; |
||||||
|
import com.fr.stable.web.SessionSource; |
||||||
|
import com.fr.stable.web.Weblet; |
||||||
|
import com.fr.web.core.RemoteAddressManager; |
||||||
|
import com.fr.web.core.SessionPoolManager; |
||||||
|
import com.fr.web.session.SessionIDInfo; |
||||||
|
import com.fr.web.utils.WebUtils; |
||||||
|
import org.easymock.EasyMock; |
||||||
|
import org.easymock.IAnswer; |
||||||
|
import org.easymock.IMocksControl; |
||||||
import org.junit.Assert; |
import org.junit.Assert; |
||||||
import org.junit.Test; |
import org.junit.Test; |
||||||
|
import org.junit.runner.RunWith; |
||||||
|
import org.powermock.api.easymock.PowerMock; |
||||||
|
import org.powermock.core.classloader.annotations.PrepareForTest; |
||||||
|
import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor; |
||||||
|
import org.powermock.modules.junit4.PowerMockRunner; |
||||||
|
import org.powermock.reflect.Whitebox; |
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletRequest; |
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.HashSet; |
||||||
|
import java.util.List; |
||||||
|
import java.util.Map; |
||||||
|
import java.util.Set; |
||||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||||
|
import java.util.concurrent.ExecutorService; |
||||||
|
import java.util.concurrent.Executors; |
||||||
|
|
||||||
|
import static org.junit.Assert.assertEquals; |
||||||
|
import static org.junit.Assert.assertFalse; |
||||||
|
import static org.junit.Assert.assertNotNull; |
||||||
|
import static org.junit.Assert.assertTrue; |
||||||
|
|
||||||
/** |
/** |
||||||
* @author richie |
* @author richie |
||||||
* @version 10.0 |
* @version 10.0 |
||||||
* Created by richie on 2019-08-09 |
* Created by richie on 2019-08-09 |
||||||
*/ |
*/ |
||||||
|
@RunWith(PowerMockRunner.class) |
||||||
|
@PrepareForTest({SessionPoolManager.class, WebUtils.class}) |
||||||
|
@SuppressStaticInitializationFor({"com.fr.web.core.SessionPoolManager"}) |
||||||
public class SessionKitTest extends Prepare { |
public class SessionKitTest extends Prepare { |
||||||
|
|
||||||
|
private static final long MIN_ACCESS_INTERVAL = 900; |
||||||
|
private int taskCount = 0; |
||||||
|
private Map<String, SessionProvider> sessions = new HashMap<String, SessionProvider>(); |
||||||
|
|
||||||
|
@Test |
||||||
|
public void testGenerateSession() throws Exception { |
||||||
|
final SessionIDInfo sessionInfo = new MockSession("session"); |
||||||
|
final Map<String, Object> params = new HashMap<String, Object>(); |
||||||
|
PowerMock.mockStatic(SessionPoolManager.class); |
||||||
|
PowerMock.mockStatic(WebUtils.class); |
||||||
|
IMocksControl control = EasyMock.createControl(); |
||||||
|
final HttpServletRequest request = control.createMock(HttpServletRequest.class); |
||||||
|
EasyMock.expect(WebUtils.getIpAddr(request)).andReturn("127.0.0.1").anyTimes(); |
||||||
|
EasyMock.expect(WebUtils.getHTTPRequestParameter(request, ParameterConstants.OP)).andReturn("view").anyTimes(); |
||||||
|
EasyMock.expect(WebUtils.parameters4SessionIDInforContainMPCache(request)).andReturn(params).anyTimes(); |
||||||
|
final Weblet weblet = control.createMock(Weblet.class); |
||||||
|
EasyMock.expect(weblet.isSessionOccupy()).andReturn(false).anyTimes(); |
||||||
|
EasyMock.expect(weblet.createSessionIDInfor(request, "127.0.0.1", params)).andReturn(sessionInfo).anyTimes(); |
||||||
|
EasyMock.expect(addSessionIDInfo(sessionInfo)).andReturn(sessionInfo.getSessionID()).anyTimes(); |
||||||
|
EasyMock.expect(SessionPoolManager.generateSessionID(request, null, weblet)).andAnswer(new IAnswer<String>() { |
||||||
|
@Override |
||||||
|
public String answer() throws Throwable { |
||||||
|
SessionProvider sessionIDInfor = weblet.createSessionIDInfor(request, "127.0.0.1", params); |
||||||
|
String sessionID = addSessionIDInfo(sessionIDInfor); |
||||||
|
if (weblet.isSessionOccupy()) { |
||||||
|
assert sessionID != null; |
||||||
|
RemoteAddressManager.add("127.0.0.1", sessionID); |
||||||
|
} |
||||||
|
return sessionID; |
||||||
|
} |
||||||
|
}).anyTimes(); |
||||||
|
control.replay(); |
||||||
|
PowerMock.replayAll(); |
||||||
|
sessionInfo.setSource(SessionSource.SINGLE); |
||||||
|
SessionPoolManager.generateSessionID(request, null, weblet); |
||||||
|
Assert.assertSame(sessionInfo.getSource(), SessionSource.SINGLE); |
||||||
|
sessionInfo.setSource(SessionSource.INTEGRATION); |
||||||
|
SessionPoolManager.generateSessionID(request, null, weblet); |
||||||
|
Assert.assertSame(sessionInfo.getSource(), SessionSource.INTEGRATION); |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void testAddDelayTask() { |
||||||
|
String sessionID = "test"; |
||||||
|
sessions.put("test", new MockSession("test")); |
||||||
|
for (int i = 0; i < 100; i++) { |
||||||
|
accessSession(sessionID); |
||||||
|
} |
||||||
|
Assert.assertEquals(1, this.taskCount); |
||||||
|
for (int i = 0; i < 2; i++) { |
||||||
|
try { |
||||||
|
Thread.sleep(1000); |
||||||
|
} catch (InterruptedException e) { |
||||||
|
//do nothing
|
||||||
|
} |
||||||
|
accessSession(sessionID); |
||||||
|
} |
||||||
|
Assert.assertEquals(3, this.taskCount); |
||||||
|
} |
||||||
|
|
||||||
|
private void accessSession(String sessionId) { |
||||||
|
if (StringUtils.isBlank(sessionId)) { |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
SessionProvider sessionIDInfor = sessions.get(sessionId); |
||||||
|
if (sessionIDInfor != null) { |
||||||
|
long accessInterval = System.currentTimeMillis() - sessionIDInfor.getLastAccessTime(); |
||||||
|
sessionIDInfor.updateTime(); |
||||||
|
if (accessInterval > MIN_ACCESS_INTERVAL) { |
||||||
|
taskCount++; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void testGetAllSession() { |
||||||
|
Whitebox.setInternalState( |
||||||
|
SessionPoolManager.class, |
||||||
|
"sessionIDMap", |
||||||
|
new ConcurrentHashMap<String, SessionProvider>() |
||||||
|
); |
||||||
|
Whitebox.setInternalState(SessionPoolManager.class, "sessionToDelete", new HashSet<String>()); |
||||||
|
Map<String, SessionProvider> sessionIDMap = Reflect.on(SessionPoolManager.class).get("sessionIDMap"); |
||||||
|
String sessionID1 = "Test1"; |
||||||
|
sessionIDMap.put(sessionID1, new MockSession(sessionID1)); |
||||||
|
sessionIDMap.put("Test2", new MockSession("Test2")); |
||||||
|
closeSession(sessionID1); |
||||||
|
List<SessionProvider> list = SessionPoolManager.getAllSession(SessionProvider.class); |
||||||
|
assertEquals(1, list.size()); |
||||||
|
assertEquals("Test2", list.get(0).getSessionID()); |
||||||
|
assertNotNull(SessionKit.getSession("Test2")); |
||||||
|
} |
||||||
|
|
||||||
@Test |
@Test |
||||||
public void getSession() throws Exception { |
public void testHasSessionID() { |
||||||
Assert.assertNull(SessionKit.getSession("abc")); |
Whitebox.setInternalState( |
||||||
|
SessionPoolManager.class, |
||||||
|
"sessionIDMap", |
||||||
|
new ConcurrentHashMap<String, SessionProvider>() |
||||||
|
); |
||||||
|
Whitebox.setInternalState(SessionPoolManager.class, "sessionToDelete", new HashSet<String>()); |
||||||
|
Map<String, SessionProvider> sessionIDMap = Reflect.on(SessionPoolManager.class).get("sessionIDMap"); |
||||||
|
String sessionID1 = "Test1"; |
||||||
|
sessionIDMap.put(sessionID1, new MockSession(sessionID1)); |
||||||
|
sessionIDMap.put("Test2", new MockSession("Test2")); |
||||||
|
closeSession(sessionID1); |
||||||
|
assertFalse(SessionPoolManager.hasSessionID(sessionID1)); |
||||||
|
assertTrue(SessionPoolManager.hasSessionID("Test2")); |
||||||
|
} |
||||||
|
|
||||||
|
private void closeSession(final String sessionID) { |
||||||
|
final Set<String> sessionToDelete = Reflect.on(SessionPoolManager.class).get("sessionToDelete"); |
||||||
|
sessionToDelete.add(sessionID); |
||||||
|
ExecutorService closeSessionPool = Executors.newFixedThreadPool(1); |
||||||
|
closeSessionPool.submit(new Runnable() { |
||||||
|
@Override |
||||||
|
public void run() { |
||||||
|
try { |
||||||
|
//模拟删除session对象的过程
|
||||||
|
Thread.sleep(1000); |
||||||
|
} catch (InterruptedException e) { |
||||||
|
//do nothing
|
||||||
|
} |
||||||
|
synchronized (SessionPoolManager.class) { |
||||||
|
sessionToDelete.remove(sessionID); |
||||||
|
} |
||||||
|
} |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
private String addSessionIDInfo(SessionProvider sessionIDInfo) { |
||||||
|
return Reflect.on(SessionPoolManager.class).call("addSessionIDInfor", sessionIDInfo).get(); |
||||||
} |
} |
||||||
} |
} |
Loading…
Reference in new issue