lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加RetCodes错误码加载器

Browse Source
This commit is contained in:
redkale
2024-06-12 16:18:57 +08:00
parent 0f110fca33
commit 15a501b2cd
2 changed files with 162 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

154
src/main/java/org/redkale/service/RetCodes.java Normal file
View File

@@ -0,0 +1,154 @@
/*
*/
package org.redkale.service;
import java.text.MessageFormat;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.locks.ReentrantLock;
/**
* 错误码加载器
*
* <p>详情见: https://redkale.org
*
* @author zhangjx
* @since 2.8.0
*/
public abstract class RetCodes {
private static final ReentrantLock loadLock = new ReentrantLock();
private static Map<String, Map<Integer, String>> rets = new LinkedHashMap<>();
private static Map<Integer, String> defret = new LinkedHashMap<>();
protected RetCodes() {
throw new IllegalStateException();
}
public static void load(Class codeClass) {
load(RetLabel.RetLoader.loadMap(codeClass));
}
public static void load(Map<String, Map<Integer, String>> map) {
if (map.isEmpty()) {
return;
}
loadLock.lock();
try {
Map<String, Map<Integer, String>> newMap = new LinkedHashMap<>();
rets.forEach((k, v) -> newMap.put(k, new LinkedHashMap<>(v)));
map.forEach((k, v) -> {
Map<Integer, String> m = newMap.get(k);
if (m != null) {
m.putAll(v);
} else {
newMap.put(k, v);
}
});
rets = newMap;
defret = rets.get("");
} finally {
loadLock.unlock();
}
}
public static RetResult retResult(int retcode) {
if (retcode == 0) {
return RetResult.success();
}
return new RetResult(retcode, retInfo(retcode));
}
public static RetResult retResult(String locale, int retcode) {
if (retcode == 0) {
return RetResult.success();
}
return new RetResult(retcode, retInfo(locale, retcode));
}
public static RetResult retResult(int retcode, Object... args) {
if (retcode == 0) {
return RetResult.success();
}
if (args == null || args.length < 1) {
return new RetResult(retcode, retInfo(retcode));
}
String info = MessageFormat.format(retInfo(retcode), args);
return new RetResult(retcode, info);
}
public static RetResult retResult(String locale, int retcode, Object... args) {
if (retcode == 0) {
return RetResult.success();
}
if (args == null || args.length < 1) {
return new RetResult(retcode, retInfo(locale, retcode));
}
String info = MessageFormat.format(retInfo(locale, retcode), args);
return new RetResult(retcode, info);
}
public static <T> CompletableFuture<RetResult<T>> retResultFuture(int retcode) {
return CompletableFuture.completedFuture(retResult(retcode));
}
public static <T> CompletableFuture<RetResult<T>> retResultFuture(String locale, int retcode) {
return CompletableFuture.completedFuture(retResult(locale, retcode));
}
public static <T> CompletableFuture<RetResult<T>> retResultFuture(int retcode, Object... args) {
return CompletableFuture.completedFuture(retResult(retcode, args));
}
public static <T> CompletableFuture<RetResult<T>> retResultFuture(String locale, int retcode, Object... args) {
return CompletableFuture.completedFuture(retResult(locale, retcode, args));
}
public static RetResult retInfo(RetResult result, int retcode, Object... args) {
if (retcode == 0) {
return result.retcode(0).retinfo("");
}
if (args == null || args.length < 1) {
return result.retcode(retcode).retinfo(retInfo(retcode));
}
String info = MessageFormat.format(retInfo(retcode), args);
return result.retcode(retcode).retinfo(info);
}
public static RetResult retInfo(RetResult result, String locale, int retcode, Object... args) {
if (retcode == 0) {
return result.retcode(0).retinfo("");
}
if (args == null || args.length < 1) {
return result.retcode(retcode).retinfo(retInfo(locale, retcode));
}
String info = MessageFormat.format(retInfo(locale, retcode), args);
return result.retcode(retcode).retinfo(info);
}
public static String retInfo(int retcode) {
if (retcode == 0) {
return "Success";
}
return defret.getOrDefault(retcode, "Error");
}
public static String retInfo(String locale, int retcode) {
if (locale == null || locale.isEmpty()) {
return retInfo(retcode);
}
if (retcode == 0) {
return "Success";
}
Map<Integer, String> map = rets.get(locale);
if (map == null) {
return "Error";
}
return map.getOrDefault(retcode, "Error");
}
}

13
src/main/java/org/redkale/service/RetLabel.java
View File

@@ -5,17 +5,16 @@
*/
package org.redkale.service;
import static java.lang.annotation.ElementType.*;
import static java.lang.annotation.RetentionPolicy.RUNTIME;
import static org.redkale.boot.Application.SYSNAME_APP_CONF_DIR;
import static org.redkale.boot.Application.SYSNAME_APP_HOME;
import java.io.*;
import java.lang.annotation.*;
import static java.lang.annotation.ElementType.*;
import static java.lang.annotation.RetentionPolicy.RUNTIME;
import java.lang.reflect.*;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.function.BiFunction;
import static org.redkale.boot.Application.SYSNAME_APP_CONF_DIR;
import static org.redkale.boot.Application.SYSNAME_APP_HOME;
import org.redkale.util.RedkaleClassLoader;
/**
@@ -50,6 +49,10 @@ public @interface RetLabel {
public static interface RetInfoTransfer extends BiFunction<Integer, String, String> {}
public abstract static class RetLoader {
private RetLoader() {
throw new IllegalStateException();
}
public static Map<String, Map<Integer, String>> loadMap(Class clazz) {
final Map<String, Map<Integer, String>> rets = new LinkedHashMap<>();