org.yx.redis.RedisPool Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sumk Show documentation

A quick developing framewort for internet company

There is a newer version: 4.0.2

/**
 * Copyright (C) 2016 - 2030 youtongluan.
 *
 * 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 org.yx.redis;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import org.yx.log.Logs;

public final class RedisPool {
	private static final ConcurrentMap map = new ConcurrentHashMap<>();

	private static Redis _defaultRedis;

	static void setDefaultRedis(Redis r) {
		_defaultRedis = r;
	}

	static List keys() {
		return new ArrayList(map.keySet());
	}

	public static Redis get(String alias) {
		if (alias == null) {
			return _defaultRedis;
		}
		return map.getOrDefault(alias, _defaultRedis);
	}

	public static Redis getRedisExactly(String alias) {
		return map.get(alias);
	}

	public static Redis defaultRedis() {
		return _defaultRedis;
	}

	public static void put(String alias, Redis redis) {
		Redis old = map.put(alias, Objects.requireNonNull(redis));
		Logs.redis().trace("redis name {} : {}", alias, redis);
		if (old != null) {
			old.shutdownPool();
			Logs.redis().info("shutdown old redis {} : {}", alias, redis);
		}
	}

	public static void shutdown() {
		Set redises = new HashSet<>(map.values());
		if (_defaultRedis != null) {
			redises.add(_defaultRedis);
		}

		for (Redis r : redises) {
			r.shutdownPool();
		}
		map.clear();
	}
}