ars.invoke.request.CacheSessionFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ars-core Show documentation
Ars framework core
The newest version!
package ars.invoke.request;

import java.util.Set;
import java.util.HashSet;
import java.util.Collections;

import ars.util.Cache;
import ars.util.SimpleCache;

/**
 * 基于缓存的会话工厂实现
 *
 * @author wuyongqiang
 */
public class CacheSessionFactory implements SessionFactory {
    /**
     * 默认会话超时时间（秒）
     */
    public final static int DEFAULT_TIMEOUT = 30 * 60;

    protected final Cache cache;
    protected final int timeout;

    public CacheSessionFactory() {
        this(DEFAULT_TIMEOUT);
    }

    public CacheSessionFactory(int timeout) {
        this(new SimpleCache(), timeout);
    }

    public CacheSessionFactory(Cache cache) {
        this(cache, DEFAULT_TIMEOUT);
    }

    public CacheSessionFactory(Cache cache, int timeout) {
        if (cache == null) {
            throw new IllegalArgumentException("Cache must not be null");
        }
        if (timeout < 1) {
            throw new IllegalArgumentException("Timeout must not be less than 1, got " + timeout);
        }
        this.cache = cache;
        this.timeout = timeout;
    }

    @Override
    public Session getSession(final Requester requester) {
        if (requester == null) {
            throw new IllegalArgumentException("Requester must not be null");
        }
        return new Session() {
            private Set names; // 所有属性名称集合

            /**
             * 获取缓存标识
             *
             * @return 缓存标识
             */
            private String key() {
                return this.key(null);
            }

            /**
             * 获取属性名称缓存标识
             *
             * @param name
             *            属性名称
             * @return 缓存标识
             */
            private String key(String name) {
                StringBuilder buffer = new StringBuilder();
                if (name == null) {
                    buffer.append("session_").append(this.getId());
                } else {
                    buffer.append("session_name_").append(this.getId()).append(name);
                }
                return buffer.toString();
            }

            @SuppressWarnings("unchecked")
            private void initializeNames() {
                if (this.names == null) {
                    this.names = (Set) cache.get(this.key());
                    if (this.names == null) {
                        this.names = new HashSet(0);
                    }
                }
            }

            @Override
            public String getId() {
                return requester.getClient();
            }

            @Override
            public int getTimeout() {
                return timeout;
            }

            @Override
            public SessionFactory getSessionFactory() {
                return CacheSessionFactory.this;
            }

            @Override
            public Set getAttributeNames() {
                this.initializeNames();
                return Collections.unmodifiableSet(this.names);
            }

            @Override
            public Object getAttribute(String name) {
                return name == null ? null : cache.get(this.key(name));
            }

            @Override
            public void setAttribute(String name, Object value) {
                if (name != null) {
                    cache.set(this.key(name), value);
                    this.initializeNames();
                    if (this.names.add(name)) {
                        cache.set(this.key(), this.names);
                    }
                }
            }

            @Override
            public void removeAttribute(String name) {
                if (name != null) {
                    cache.remove(this.key(name));
                    this.initializeNames();
                    if (this.names.remove(name)) {
                        cache.set(this.key(), this.names);
                    }
                }
            }

            @Override
            public String toString() {
                this.initializeNames();
                if (this.names.isEmpty()) {
                    return "{}";
                }
                int i = 0;
                StringBuilder sb = new StringBuilder();
                sb.append('{');
                for (String name : this.names) {
                    if (i++ > 0) {
                        sb.append(", ");
                    }
                    sb.append(name).append('=').append(this.getAttribute(name));
                }
                return sb.append('}').toString();
            }

        };
    }

    @Override
    public void destroy() {
        this.cache.destroy();
    }

}