com.corundumstudio.socketio.store.RedisTemplateMap Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2018, hiwepy (https://github.com/hiwepy).
 *
 * 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 com.corundumstudio.socketio.store;

import org.springframework.data.redis.core.BoundHashOperations;
import org.springframework.data.redis.core.RedisTemplate;

import java.util.Collection;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @author ： wandl
 */
@SuppressWarnings("unchecked")
public class RedisTemplateMap implements Map {

	private final String name;
	private final BoundHashOperations hashOperations;

	public RedisTemplateMap(RedisTemplate redisTemplate, String name) {
		this.name = name;
		this.hashOperations = redisTemplate.boundHashOps(name);
	}

	@Override
	public int size() {
		return hashOperations.size().intValue();
	}

	@Override
	public boolean isEmpty() {
		return size() > 0;
	}

	@Override
	public boolean containsKey(Object key) {
		return hashOperations.hasKey(key);
	}

	@Override
	public boolean containsValue(Object value) {
		return hashOperations.entries().containsValue(value);
	}

	@Override
	public V get(Object key) {
		return (V) hashOperations.get(key);
	}

	@Override
	public V put(Object key, Object value) {
		hashOperations.put(key, value);
		return (V) value;
	}

	@Override
	public V remove(Object key) {
		Object value = hashOperations.get(key);
		hashOperations.delete(name, key);
		return (V) value;
	}

	@Override
	public void putAll(Map m) {
		hashOperations.putAll(m);
	}

	@Override
	public void clear() {
		hashOperations.delete(name);
	}

	@Override
	public Set keySet() {
		return (Set) hashOperations.keys();
	}

	@Override
	public Collection values() {
		return (Collection) hashOperations.values();
	}

	@Override
	public Set> entrySet() {
		Set> sets = hashOperations.entries().entrySet();
		if(sets != null) {
			return sets.stream().map((m) -> {
				return (Entry) m;
			}).collect(Collectors.toSet());
		}
		return null;
	}

}