[Add] 添加 MapCacheStore 的本地实现;

[Add] HashMapCacheStore 添加基于 Hashtable 的 MapCacheStore 实现;
[Add] HashMapCacheStoreTest 添加针对 HashMapCacheStore 的完整单元测试;
This commit is contained in:
LamGC 2020-09-04 19:58:45 +08:00
parent ba422ef90e
commit 688b99d198
Signed by: LamGC
GPG Key ID: 6C5AE2A913941E1D
2 changed files with 247 additions and 0 deletions

View File

@ -0,0 +1,117 @@
/*
* Copyright (C) 2020 LamGC
*
* ContentGrabbingJi is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* ContentGrabbingJi is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
package net.lamgc.cgj.bot.cache.local;
import net.lamgc.cgj.bot.cache.MapCacheStore;
import java.util.*;
import java.util.function.Function;
/**
*
* @param <V>
* @see net.lamgc.cgj.bot.cache.CacheStore
* @see net.lamgc.cgj.bot.cache.MapCacheStore
* @author LamGC
*/
public class HashMapCacheStore<V> extends HashCacheStore<Map<String, V>> implements MapCacheStore<V> {
@Override
public int mapSize(String key) {
return getMap(key, false, Map::size, -1);
}
@Override
public Set<String> mapFieldSet(String key) {
return getMap(key, false, Map::keySet, null);
}
@Override
public Set<V> mapValueSet(String key) {
return getMap(key, false, map -> new HashSet<>(map.values()), null);
}
@Override
public boolean put(String key, String field, V value) {
return getMap(key, true, map -> {
map.put(Objects.requireNonNull(field), Objects.requireNonNull(value));
return true;
}, false);
}
@Override
public boolean putAll(String key, Map<String, V> map) {
return getMap(key, true, keyMap -> {
keyMap.putAll(Objects.requireNonNull(map));
return true;
}, false);
}
@Override
public boolean putIfNotExist(String key, String field, V value) {
return getMap(key, true, map -> {
if (map.containsKey(Objects.requireNonNull(field))) {
return false;
}
map.put(Objects.requireNonNull(field), Objects.requireNonNull(value));
return true;
}, false);
}
@Override
public V get(String key, String field) {
return getMap(key, false, map -> map.get(Objects.requireNonNull(field)), null);
}
@Override
public boolean removeField(String key, String field) {
return getMap(key, false, map -> map.remove(Objects.requireNonNull(field)) != null, false);
}
@Override
public boolean containsField(String key, String field) {
return getMap(key, false, map -> map.containsKey(Objects.requireNonNull(field)), false);
}
@Override
public boolean mapIsEmpty(String key) {
return getMap(key, false, Map::isEmpty, false);
}
@Override
public boolean clearMap(String key) {
return getMap(key, false, map -> {
map.clear();
return true;
}, false);
}
private <R> R getMap(String key, boolean create, Function<Map<String, V>, R> notNull, R isNull) {
Objects.requireNonNull(key);
Map<String, CacheItem<Map<String, V>>> cacheMap = getCacheMap();
if (!cacheMap.containsKey(key)) {
if (create) {
cacheMap.put(key, new CacheItem<>(new Hashtable<>()));
} else {
return isNull;
}
}
return notNull.apply(cacheMap.get(key).getValue());
}
}

View File

@ -0,0 +1,130 @@
/*
* Copyright (C) 2020 LamGC
*
* ContentGrabbingJi is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* ContentGrabbingJi is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
package net.lamgc.cgj.bot.cache.local;
import net.lamgc.cgj.bot.cache.MapCacheStore;
import org.junit.Assert;
import org.junit.Test;
import java.util.HashMap;
import java.util.Map;
/**
* @see HashMapCacheStore
*/
public class HashMapCacheStoreTest {
@Test
public void nullThrowTest() {
final MapCacheStore<String> cacheStore = new HashMapCacheStore<>();
Assert.assertThrows(NullPointerException.class, () -> cacheStore.mapSize(null));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.mapFieldSet(null));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.mapValueSet(null));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.put(null, "field", "value"));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.put("testKey", null, "value"));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.put("testKey", "field", null));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.putAll(null, new HashMap<>()));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.putAll("testKey", null));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.putIfNotExist(null, "field", "value"));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.putIfNotExist("testKey", null, "value"));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.putIfNotExist("testKey", "field", null));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.get("testKey", null));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.get(null, "field"));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.removeField("testKey", null));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.removeField(null, "field"));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.mapIsEmpty(null));
Assert.assertThrows(NullPointerException.class, () -> cacheStore.clearMap(null));
}
@Test
public void keyNotExistTest() {
final MapCacheStore<String> cacheStore = new HashMapCacheStore<>();
final String key = "testKey";
Assert.assertEquals(-1, cacheStore.mapSize(key));
Assert.assertFalse(cacheStore.mapIsEmpty(key));
Assert.assertFalse(cacheStore.clearMap(key));
Assert.assertFalse(cacheStore.containsField(key, "Field"));
Assert.assertFalse(cacheStore.removeField(key, "Field"));
Assert.assertNull(cacheStore.get(key, "Field"));
Assert.assertTrue(cacheStore.put(key, "Field", "value"));
Assert.assertTrue("clearMap operation failed!", cacheStore.remove(key));
Assert.assertTrue(cacheStore.putAll(key, new HashMap<>()));
Assert.assertTrue("clearMap operation failed!", cacheStore.remove(key));
Assert.assertTrue(cacheStore.putIfNotExist(key, "Field", "value"));
}
@Test
public void putAndGetTest() {
final MapCacheStore<String> cacheStore = new HashMapCacheStore<>();
final String key = "testKey";
final Map<String, String> expectedMap = new HashMap<>();
expectedMap.put("test01", "testValue01");
expectedMap.put("test02", "testValue02");
expectedMap.put("test03", "testValue03");
expectedMap.put("test04", "testValue04");
expectedMap.put("test05", "testValue05");
expectedMap.put("test06", "testValue06");
// put/get, mapIsEmpty, containsField
Assert.assertTrue("put operation failed!", cacheStore.put(key, "test00", "testValue00"));
Assert.assertTrue(cacheStore.containsField(key, "test00"));
Assert.assertEquals("testValue00", cacheStore.get(key, "test00"));
Assert.assertTrue("removeField operation failed!", cacheStore.removeField(key, "test00"));
// putIfNotExist
Assert.assertTrue(cacheStore.putIfNotExist(key, "test00", "testValue00"));
Assert.assertFalse(cacheStore.putIfNotExist(key, "test00", "testValue00"));
Assert.assertTrue("clearMap operation failed!", cacheStore.clearMap(key));
// putAll
Assert.assertTrue(cacheStore.putAll(key, expectedMap));
Assert.assertTrue(expectedMap.keySet().containsAll(cacheStore.mapFieldSet(key)));
Assert.assertTrue(expectedMap.values().containsAll(cacheStore.mapValueSet(key)));
}
@Test
public void fieldChangeTest() {
final MapCacheStore<String> cacheStore = new HashMapCacheStore<>();
final String key = "testKey";
final Map<String, String> expectedMap = new HashMap<>();
expectedMap.put("test01", "testValue01");
expectedMap.put("test02", "testValue02");
expectedMap.put("test03", "testValue03");
expectedMap.put("test04", "testValue04");
expectedMap.put("test05", "testValue05");
expectedMap.put("test06", "testValue06");
// mapSize, clearMap, mapIsEmpty 测试
Assert.assertTrue("putAll operation failed!", cacheStore.putAll(key, expectedMap));
Assert.assertEquals(expectedMap.size(), cacheStore.mapSize(key));
Assert.assertTrue(cacheStore.clearMap(key));
Assert.assertTrue(cacheStore.mapIsEmpty(key));
// removeField 多分支测试
Assert.assertTrue("put operation failed!", cacheStore.put(key, "test00", "testValue00"));
Assert.assertTrue(cacheStore.containsField(key, "test00"));
Assert.assertEquals("testValue00", cacheStore.get(key, "test00"));
Assert.assertTrue("removeField operation failed!", cacheStore.removeField(key, "test00"));
Assert.assertFalse(cacheStore.removeField(key, "test00"));
}
}