package com.engine.salary.util; import com.alibaba.fastjson.JSONObject; import com.alibaba.fastjson.serializer.SerializerFeature; import com.engine.common.service.HrmCommonService; import com.engine.common.service.impl.HrmCommonServiceImpl; import com.engine.core.exception.ECException; import com.engine.salary.exception.SalaryRunTimeException; import com.fasterxml.jackson.core.JsonProcessingException; import com.fasterxml.jackson.databind.ObjectMapper; import lombok.extern.slf4j.Slf4j; import weaver.general.BaseBean; import weaver.hrm.User; import java.util.HashMap; import java.util.Map; import java.util.function.Consumer; import java.util.function.Function; import java.util.function.Supplier; @Slf4j public class ResponseResult { private static final long serialVersionUID = 1L; private final User user; private final HrmCommonService hrmCommonService = new HrmCommonServiceImpl(); private final BaseBean baseBean = new BaseBean(); private final Boolean isLog = "true".equals(baseBean.getPropValue("hrmSalary", "log")); public ResponseResult(User user) { this.user = user; } /** * 统一返回方法 */ public String run(Function f, T t) { try { if (isLog) { log.info("run salary api , param {}", t); } return Ok(f.apply(t)); } catch (SalaryRunTimeException e) { return Error(e.getMessage()); } catch (ECException e) { log.error("salary api run fail", e); Throwable cause = e.getCause(); return Error(cause.getMessage()); } catch (Exception e) { log.error("salary api run fail", e); return Error(e.getMessage()); } } /** * 统一返回方法(有参无返回) */ public String run(Consumer f, T t) { try { if (isLog) { log.info("run salary api , param {}", t); } f.accept(t); return Ok(); } catch (SalaryRunTimeException e) { return Error(e.getMessage()); } catch (ECException e) { log.error("salary api run fail", e); Throwable cause = e.getCause(); return Error(cause.getMessage()); } catch (Exception e) { log.error("salary api run fail", e); return Error(e.getMessage()); } } /** * 统一返回方法(无参有返回) */ public String run(Supplier f) { try { if (isLog) { log.info("run salary api"); } return Ok(f.get()); } catch (SalaryRunTimeException e) { return Error(e.getMessage()); } catch (ECException e) { log.error("salary api run fail", e); Throwable cause = e.getCause(); return Error(cause.getMessage()); } catch (Exception e) { log.error("salary api run fail", e); return Error(e.getMessage()); } } /** * 成功返回 */ private String Ok(R r) { Map apidatas = new HashMap<>(); apidatas.put("status", true); apidatas.put("data", r); String success = getJsonString(apidatas); if (isLog) { log.info("run salary api success return {}", success); } return success; } private static String getJsonString(Object apidatas) { ObjectMapper mapper = new ObjectMapper(); try { return mapper.writeValueAsString(apidatas); } catch (JsonProcessingException e) { e.printStackTrace(); } return ""; } /** * 成功返回 */ private String Ok() { Map apidatas = new HashMap<>(); apidatas.put("status", true); return JSONObject.toJSONString(apidatas, SerializerFeature.DisableCircularReferenceDetect); } /** * 失败返回 */ private static String Error(String message) { Map apidatas = new HashMap<>(); apidatas.put("status", false); apidatas.put("errormsg", message); return JSONObject.toJSONString(apidatas, SerializerFeature.DisableCircularReferenceDetect); } }