diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/R.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/R.java index 381a6f6..97e1ea5 100644 --- a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/R.java +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/R.java @@ -5,6 +5,10 @@ import lombok.Data; import lombok.NoArgsConstructor; import java.io.Serializable; +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; /** * 响应信息主体 @@ -32,6 +36,21 @@ public class R implements Serializable { private T data; + public R setCode(int code){ + this.code = code; + return this; + } + + public R setMsg(String msg){ + this.msg = msg; + return this; + } + + public R setData(T data) { + this.data = data; + return this; + } + public static R ok() { return restResult(null, SUCCESS, "操作成功"); } @@ -104,4 +123,36 @@ public class R implements Serializable { public static Boolean isSuccess(R ret) { return R.SUCCESS == ret.getCode(); } + + public static R> map(){ + R> r = new R<>(); + r.setData(new HashMap<>()); + return r; + } + + @SuppressWarnings("unchecked") + public R> put(String key,Object value){ + if(this.data instanceof Map) { + ((Map)this.data).put(key, value); + return (R>) this; + }else { + throw new RuntimeException("data does not belong to Map"); + } + } + + public static R> list(){ + R> r = new R<>(); + r.setData(new ArrayList<>()); + return r; + } + + @SuppressWarnings("unchecked") + public R> add(Object value){ + if(this.data instanceof List) { + ((List)this.data).add(value); + return (R>) this; + }else { + throw new RuntimeException("data does not belong to List"); + } + } }