ZhouXY108
/
plusone-commons
forked from plusone/plusone-commons
1
0
Fork 0
Code Pull Requests Activity

将 ValueSet 中的 Map 改为不可变集合。

feature/net-util
ZhouXY108 2023-04-29 15:08:26 +08:00
parent a771b207f8
commit c6ddb8ab1e
2 changed files with 14 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/main/java/xyz/zhouxy/plusone/commons/util/Enumeration.java
View File

@ -17,11 +17,10 @@
package xyz.zhouxy.plusone.commons.util;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
import javax.annotation.Nonnull;
/**
* 枚举类
@ -71,26 +70,25 @@ public abstract class Enumeration<T extends Enumeration<T>> implements Comparabl
}
protected static final class ValueSet<T extends Enumeration<T>> {
private final Map<Integer, @Nonnull T> values = new ConcurrentHashMap<>();
private final Map<Integer, T> valueMap;
@SafeVarargs
public ValueSet(T... values) {
Map<Integer, T> valueMap = new HashMap<>(values.length);
for (T value : values) {
put(value);
Assert.notNull(value, "Value must not be null.");
valueMap.put(value.getId(), value);
}
}
private void put(final T value) {
this.values.put(value.getId(), Objects.requireNonNull(value, "Value must not be null."));
this.valueMap = Collections.unmodifiableMap(valueMap);
}
public T get(int id) {
Assert.isTrue(this.values.containsKey(id), "%s 对应的值不存在", id);
return this.values.get(id);
Assert.isTrue(this.valueMap.containsKey(id), "%s 对应的值不存在", id);
return this.valueMap.get(id);
}
public Collection<@Nonnull T> getValues() {
return this.values.values();
public Collection<T> getValues() {
return this.valueMap.values();
}
}
}

6
src/test/java/xyz/zhouxy/plusone/commons/EnumerationTests.java
View File

@ -1,6 +1,6 @@
package xyz.zhouxy.plusone.commons;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.junit.jupiter.api.Assertions.assertSame;
import com.google.common.collect.Lists;
import org.junit.jupiter.api.Test;
@ -17,8 +17,8 @@ class EnumerationTests {
@Test
void testEnumeration() {
assertTrue(EntityStatus.AVAILABLE == EntityStatus.of(0));
assertTrue(Result.SUCCESSFUL == Result.of(1));
assertSame(EntityStatus.AVAILABLE, EntityStatus.of(0));
assertSame(Result.SUCCESSFUL, Result.of(1));
Collection<Comparable<? extends Enumeration<?>>> enums = Lists.newArrayList();
enums.addAll(EntityStatus.constants());
enums.addAll(Result.constants());