SiZyFeeOprnSystem/src/main/java/thyyxxk/sizyfeeoprnsystm/service/ExecService.java

package thyyxxk.sizyfeeoprnsystm.service;

import com.alibaba.fastjson.JSONObject;
import org.apache.commons.codec.binary.Base64;
import org.springframework.http.HttpHeaders;
import thyyxxk.sizyfeeoprnsystm.config.MedinsurConfig;
import thyyxxk.sizyfeeoprnsystm.dao.SiZyDao;
import thyyxxk.sizyfeeoprnsystm.dicts.SiFunction;
import thyyxxk.sizyfeeoprnsystm.utils.DateUtil;
import thyyxxk.sizyfeeoprnsystm.utils.StringUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpEntity;
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;
import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * @description: 签到签退
 * @author: DingJie
 * @create: 2021-05-28 15:57:26
 **/
@Slf4j
@Service
public class ExecService {
    private static final String RESULT_CODE = "infcode";
    private static final String OUTPUT = "output";
    private static int serial = 1000;
    private static final int MIN_VAL = 1000;
    private static final int MAX_VAL = 9999;
    private final SiZyDao siZyDao;
    private final RestTemplate template;
    private final MedinsurConfig cfg;
    private volatile String signNo;

    @Autowired
    public ExecService(SiZyDao siZyDao, RestTemplate template, MedinsurConfig cfg) {
        this.siZyDao = siZyDao;
        this.template = template;
        this.cfg = cfg;
    }

    private synchronized String getSignNo() {
        if (StringUtil.isBlank(signNo)) {
            signIn();
        }
        return signNo;
    }

    public synchronized void signIn() {
        String dbSignNo = siZyDao.getSignInNo();
        if (StringUtil.notBlank(dbSignNo)) {
            signNo = dbSignNo;
            log.info("获取历史签到号成功：{}", dbSignNo);
            return;
        }
        JSONObject input = makeSignHeader(SiFunction.SIGN_IN);
        JSONObject signIn = new JSONObject();
        signIn.put("opter_no", "99999");
        signIn.put("mac", cfg.getMacAddress());
        signIn.put("ip", cfg.getIpAddress());
        input.getJSONObject("input").put("signIn", signIn);
        JSONObject result = executeTrade(input, SiFunction.SIGN_IN);
        log.info("医保签到：\n参数：{}，\n结果：{}", input, result);
        if (null != result && result.getIntValue(RESULT_CODE) == 0) {
            try {
                JSONObject output = result.getJSONObject(OUTPUT);
                signNo = output.getJSONObject("signinoutb").getString("sign_no");
                siZyDao.updateSignNo(signNo);
                log.info("签到成功，签到号：{}", signNo);
            } catch (Exception e) {
                log.error("签到出错", e);
            }
        }
    }

    public void signOut() {
        if (StringUtil.notBlank(signNo)) {
            JSONObject input = makeSignHeader(SiFunction.SIGN_OUT);
            JSONObject signOut = new JSONObject();
            signOut.put("sign_no", signNo);
            signOut.put("opter_no", "99999");
            input.getJSONObject("input").put("signOut", signOut);
            JSONObject result = executeTrade(input, SiFunction.SIGN_OUT);
            log.info("医保签退：\n参数：{}，\n结果：{}", input, result);
            if (null != result && result.getIntValue(RESULT_CODE) == 0) {
                signNo = null;
            }
        }
    }

    public String getInstitutionArea(String insureArea) {
//        String hospArea = cfg.getHospArea();
//        String prefix = hospArea.substring(0, 4);
//        return insureArea.startsWith(prefix) ? hospArea : prefix.substring(0, 2) + "9900";
        return cfg.getHospArea();
    }

    private String timestamp() {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmss");
        return sdf.format(new Date());
    }

    public synchronized String makeMsgId() {
        serial += 1;
        if (serial > MAX_VAL) {
            serial = MIN_VAL;
        }
        return cfg.getHospId() + timestamp() + serial;
    }

    public HttpHeaders getHttpHeaders(SiFunction function) {
        String timestamp = String.valueOf(System.currentTimeMillis());
        String signature = getSignature(timestamp, function.getCode());
        HttpHeaders headers = new HttpHeaders();
//        headers.add("apiAccessKey", cfg.getAccessKey());
//        headers.add("apiSecreKey", cfg.getSecretKey());
//        headers.add("apiName", cfg.getApiName());
//        headers.add("apiVersion", cfg.getApiVersion());
//        headers.add("apiTimestamp", timestamp);
//        headers.add("apiSignature", signature);

        // 医保原始地址添加header
        headers.add("_api_access_key", cfg.getAccessKey());
//        headers.add("secretKey", cfg.getSecretKey());
        headers.add("_api_name", cfg.getApiName());
        headers.add("_api_version", cfg.getApiVersion());
        headers.add("_api_timestamp", timestamp);
        headers.add("_api_signature", signature);
        return headers;
    }

    public String getSignature(String timestamp, String function) {
        String source =
                "_api_access_key=" + cfg.getAccessKey() +
                        "&_api_name=" + cfg.getApiName() +
                        "&_api_timestamp=" + timestamp +
                        "&_api_version=" + cfg.getApiVersion();
        return hmacSha1(source);
    }

    private String hmacSha1(String src) {
        byte[] result = null;
        try {
            SecretKeySpec signInKey = new SecretKeySpec(cfg.getSecretKey().getBytes(), "HmacSHA1");
            Mac mac = Mac.getInstance("HmacSHA1");
            mac.init(signInKey);
            byte[] rawHmac = mac.doFinal(src.getBytes());
            result = Base64.encodeBase64(rawHmac);
        } catch (Exception e) {
            log.error("HmacSHA1加密出错", e);
        }
        return null == result ? null : new String(result);
    }
    
    
    private JSONObject makeSignHeader(SiFunction function) {
        JSONObject header = makePublicHeader("");
        header.put("infno", function.getCode());
        header.put("opter", "99999");
        header.put("opter_name", "全院");
        return header;
    }

    public JSONObject makeTradeHeaderWithInsureArea(SiFunction function, String insureArea, String staffId) {
        String staffName = siZyDao.selectStaffName(staffId);
        JSONObject header = makePublicHeader(insureArea);
        header.put("infno", function.getCode());
        header.put("opter", staffId);
        header.put("opter_name", staffName);
        header.replace("sign_no", getSignNo());
        return header;
    }

    public JSONObject makePublicHeader(String insureArea) {
        if (null == insureArea) {
            insureArea = "";
        }
        JSONObject header = new JSONObject();
        JSONObject input = new JSONObject();
        header.put("msgid", makeMsgId());
        header.put("mdtrtarea_admvs", getInstitutionArea(insureArea));
        header.put("insuplc_admdvs", insureArea);
        header.put("recer_sys_code", cfg.getHospArea());
        header.put("dev_no", "");
        header.put("dev_safe_info", "");
        header.put("cainfo", "");
        header.put("signtype", "");
        header.put("infver", cfg.getApiVersion());
        header.put("opter_type", "1");
        header.put("inf_time", DateUtil.now());
        header.put("fixmedins_code", cfg.getHospId());
        header.put("fixmedins_name", cfg.getHospName());
        header.put("sign_no", "");
        if (StringUtil.notBlank(cfg.getSoftDeveloper())) {
            header.put("fixmedins_soft_fcty", cfg.getSoftDeveloper());
        }
        header.put("input", input);
        return header;
    }

    public JSONObject executeTrade(JSONObject input, SiFunction function) {
        return template.postForObject(cfg.getApiUrl(),
                new HttpEntity<>(input, getHttpHeaders(function)), JSONObject.class);
    }
    
    /**
     * 工伤接口URL
     */
    // 模拟接口地址（当前使用）
    private static final String WORK_INJURY_API_URL = "http://130.150.161.72:9206/thyy/api/public/injury/workinjury";
    // 真实接口地址（注释掉，需要时手动切换）
    // private static final String WORK_INJURY_API_URL = "http://localhost:8321/api/entry/workinjury";
    
    /**
     * 执行工伤接口调用（工伤接口使用不同的请求头格式）
     */
    public JSONObject executeWorkInjuryTrade(JSONObject input) {
        HttpHeaders headers = new HttpHeaders();
        headers.add("Content-Type", "application/json");
        // 工伤接口可能不需要医保接口的认证头，直接使用简单的JSON头
        return template.postForObject(WORK_INJURY_API_URL,
                new HttpEntity<>(input, headers), JSONObject.class);
    }
}