今天看了看泛型,发现自己好像从来没用过。想到了以前自己苦逼的包装Result的日子,于是就用泛型和反射写了一个小轮子,可以自动包装Result,再也不用自己一个一个set啦。
功能比较简陋,请多担待。
public class U100 {}
import lombok.Data; @Data public class U32 extends U100 { String username; }
import lombok.Data; @Data public class U33 extends U100{ String username; }
public class Test { public static void main(String[] args) { Result result=new Result(); //设置两个POJO对象,并set值 U32 u32=new U32(); u32.setUsername("u32 Name"); U33 u33=new U33(); u33.setUsername("u33 Name"); //调用轮子 System.out.println(result.getResult(u32,u33).toString()); //设置两个List,并add值 List<U32> u32s=new ArrayList<>(); u32s.add(u32); List<U33> u33s=new ArrayList<>(); u33s.add(u33); //调用轮子 System.out.println(result.getResult(u32s,u33s).toString()); } }
class Result <T> { //传入两个对象,返回Result Map public Map getResult(T t1,T t2) { Map<String,T> dataInnerMap=new HashMap<>(); dataInnerMap.put(t1.getClass().getSimpleName(),t1); dataInnerMap.put(t2.getClass().getSimpleName(),t2); Map<String,Object> Result=new HashMap<>(); Result.put("code","200"); Result.put("Msg","Success!"); Result.put("data",dataInnerMap); return Result; } //传入两个List,返回Result Map public Map getResult(List<? extends U100> t1,List<? extends U100> t2) { Map<String,List<? extends U100>> dataInnerMap=new HashMap<>(); if(t1!=null){ dataInnerMap.put(t1.get(0).getClass().getSimpleName(),t1); } if(t2!=null){ dataInnerMap.put(t2.get(0).getClass().getSimpleName(),t2); } Map<String,Object> Result=new HashMap<>(); Result.put("code","200"); Result.put("Msg","Success!"); Result.put("data",dataInnerMap); return Result; } }
看一眼结果
还不错,可以少写很多代码,而且不用怕data内部Map命名错误前端看不懂了。后续加一些复杂功能,嘿嘿嘿。