oauth2 v2 update

src/main/java/love/sola/netsupport/session/MapSession.java

@@ -0,0 +1,88 @@
+/*
+ * Copyright 2002-2015 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not
+ * use this file except in compliance with the License. You may obtain a copy of
+ * the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations under
+ * the License.
+ */
+package love.sola.netsupport.session;
+
+import lombok.EqualsAndHashCode;
+import lombok.Getter;
+import lombok.Setter;
+
+import java.io.Serializable;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Set;
+import java.util.UUID;
+
+/**
+ * @author Sola
+ */
+@EqualsAndHashCode(of = "id")
+public final class MapSession implements WxSession, Serializable {
+
+	@Getter
+	private final String id;
+	private Map<String, Object> sessionAttrs = new HashMap<String, Object>();
+	@Getter
+	private long creationTime = System.currentTimeMillis();
+	@Getter
+	@Setter
+	private long lastAccessedTime = creationTime;
+	@Getter
+	private boolean invalidated = false;
+
+	/**
+	 * Creates a new instance with a secure randomly generated identifier.
+	 */
+	public MapSession() {
+		this(UUID.randomUUID().toString());
+	}
+
+	/**
+	 * Creates a new instance with the specified id. This is preferred to the
+	 * default constructor when the id is known to prevent unnecessary consumption on
+	 * entropy which can be slow.
+	 *
+	 * @param id the identifier to use
+	 */
+	public MapSession(String id) {
+		this.id = id;
+	}
+
+	@SuppressWarnings("unchecked")
+	public <T> T getAttribute(String attributeName) {
+		return (T) sessionAttrs.get(attributeName);
+	}
+
+	public Set<String> getAttributeNames() {
+		return sessionAttrs.keySet();
+	}
+
+	public void setAttribute(String attributeName, Object attributeValue) {
+		if (attributeValue == null) {
+			removeAttribute(attributeName);
+		} else {
+			sessionAttrs.put(attributeName, attributeValue);
+		}
+	}
+
+	public void removeAttribute(String attributeName) {
+		sessionAttrs.remove(attributeName);
+	}
+
+	public void invalidate() {
+		invalidated = true;
+	}
+
+}

src/main/java/love/sola/netsupport/session/MapSessionRepository.java

@@ -0,0 +1,85 @@
+/*
+ * Copyright 2002-2015 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not
+ * use this file except in compliance with the License. You may obtain a copy of
+ * the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations under
+ * the License.
+ */
+package love.sola.netsupport.session;
+
+import com.google.common.cache.CacheBuilder;
+import com.google.common.cache.CacheLoader;
+import com.google.common.cache.LoadingCache;
+import love.sola.netsupport.config.Settings;
+import org.apache.commons.lang3.Validate;
+
+import javax.annotation.Nonnull;
+import java.util.Map;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * @author Sola
+ */
+public class MapSessionRepository {
+
+	private final LoadingCache<String, MapSession> sessions;
+
+	public MapSessionRepository() {
+		this(CacheBuilder.newBuilder()
+				.concurrencyLevel(4)
+				.maximumSize(65535)
+				.expireAfterAccess(Settings.I.User_Session_Max_Inactive, TimeUnit.SECONDS)
+				.build(new CacheLoader<String, MapSession>() {
+					       @Override
+					       public MapSession load(@Nonnull String key) throws Exception {
+						       return new MapSession(key);
+					       }
+				       }
+				)
+		);
+	}
+
+	public MapSessionRepository(LoadingCache<String, MapSession> sessions) {
+		Validate.notNull(sessions);
+		this.sessions = sessions;
+	}
+
+	public void save(MapSession session) {
+		sessions.put(session.getId(), session);
+	}
+
+	public MapSession getSession(String id) {
+		MapSession saved = sessions.getIfPresent(id);
+		if (saved == null) {
+			return null;
+		}
+		if (saved.isInvalidated()) {
+			delete(saved.getId());
+			return null;
+		}
+		return saved;
+	}
+
+	public void delete(String id) {
+		sessions.invalidate(id);
+	}
+
+	public MapSession createSession() {
+		MapSession session = new MapSession();
+		save(session);
+		return session;
+	}
+
+	public Map<String, MapSession> asMap() {
+		return sessions.asMap();
+	}
+
+}

src/main/java/love/sola/netsupport/session/WechatSession.java

@@ -0,0 +1,32 @@
+package love.sola.netsupport.session;
+
+
+import java.util.Collection;
+
+/**
+ * ***********************************************
+ * Created by Sola on 2015/12/14.
+ * Don't modify this source without my agreement
+ * ***********************************************
+ */
+public class WechatSession {
+
+	private static MapSessionRepository repository;
+
+	static {
+		repository = new MapSessionRepository();
+	}
+
+	public static WxSession get(String id) {
+		return repository.getSession(id);
+	}
+
+	public static WxSession create() {
+		return repository.createSession();
+	}
+
+	public static Collection<? extends WxSession> list() {
+		return repository.asMap().values();
+	}
+
+}

src/main/java/love/sola/netsupport/session/WxSession.java

@@ -0,0 +1,22 @@
+package love.sola.netsupport.session;
+
+import java.util.Set;
+
+/**
+ * @author Sola
+ */
+public interface WxSession {
+
+	String getId();
+
+	<T> T getAttribute(String name);
+
+	Set<String> getAttributeNames();
+
+	void setAttribute(String name, Object value);
+
+	void removeAttribute(String name);
+
+	void invalidate();
+
+}