修改Rest帮助类

13693261870

2023-10-07 ff16d1213d5e43b21d3f72551c83dd007bec2261

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package com.smartearth.poiexcel.service;
 
import com.google.common.collect.Lists;
import com.smartearth.poiexcel.entity.*;
import com.smartearth.poiexcel.mapper.EntMapper;
import com.smartearth.poiexcel.utils.RestHelper;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.*;
import org.springframework.stereotype.Service;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
 
import javax.annotation.Resource;
import java.util.List;
 
/**
 * 企业服务类
 * @author WWW
 * @date 2023-10-05
 */
@Service
@SuppressWarnings("ALL")
public class EntService {
    @Resource
    EntMapper entMapper;
 
    @Value("${qylweb.host}")
    String host;
 
    @Value("${qylweb.user}")
    String user;
 
    @Value("${qylweb.pwd}")
    String pwd;
 
    private final static Log log = LogFactory.getLog(EntService.class);
 
    private final static String GENERATE_TOKEN = "%s/yqfwg/app/generateToken";
 
    private final static String GET_ENT_BASE_INFO = "%s/yqfwg/api/project/getEntBaseInfoForOtherSysListPage?ctoken=%s&showCount=%d&currentPage=%d";
 
    /**
     * 获取令牌
     */
    public String selectToken() {
        try {
            String url = String.format(GENERATE_TOKEN, host);
 
            //Map<String, String> map = new HashMap<>(2);
            //map.put("userName", user);
            //map.put("passWord", pwd);
 
            //String map = "userName=shikong001&passWord=123!@#qwe";
 
            // 请求头信息
            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.valueOf("application/x-www-form-urlencoded"));
 
            //设置为表单提交，按需求加
            headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
            // 组装请求信息
            MultiValueMap<String, String> map = new LinkedMultiValueMap<String, String>();
            map.add("userName", user);
            map.add("passWord", pwd);
            HttpEntity<MultiValueMap<String, String>> requestEntity = new HttpEntity<MultiValueMap<String, String>>(map, headers);
 
            // String str = RestHelper.postForRest(url, map)
            Result rs = RestHelper.postForRest(url, map, Result.class);
            if (null == rs || StaticData.I200 != rs.getCode()) {
                return null;
            }
 
            // TokenResult tr = JSONObject.parseObject(rs.getData().toJSONString(), TokenResult.class)
            TokenResult tr = rs.getData().toJavaObject(TokenResult.class);
 
            return tr.getCtoken();
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return null;
        }
    }
 
    /**
     * 查询企业
     */
    public List<EntEntity> selectEnts(String token, String startDate, String endDate, String qylabel, Integer showCount, Integer currentPage) {
        try {
            String url = String.format(GET_ENT_BASE_INFO, host, token, showCount, currentPage);
            if (null != startDate) {
                url += "&buildDate_startdate=" + startDate;
            }
            if (null != endDate) {
                url += "&buildDate_enddate=" + endDate;
            }
            if (null != qylabel) {
                url += "&qylabel=" + qylabel;
            }
 
            Result rs = RestHelper.getForRest(url, Result.class);
            if (null == rs || StaticData.I200 != rs.getCode()) {
                return null;
            }
 
            // EntResult er = JSONObject.parseObject(rs.getData().toJSONString(), EntResult.class)
            EntResult er = rs.getData().toJavaObject(EntResult.class);
 
            return er.getPd().getList();
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return null;
        }
    }
 
    /**
     * 插入企业
     */
    public Integer insertEnts(List<EntEntity> list) {
        try {
            int rows = 0;
            List<List<EntEntity>> subLists = Lists.partition(list, StaticData.I200);
            for (List<EntEntity> sub : subLists) {
                try {
                    rows += entMapper.insertBatch(sub);
                } catch (Exception ex) {
                    log.error(ex);
                }
            }
 
            return rows;
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return 0;
        }
    }
}