新增模块和注册中

puzhibing

2023-06-13 11b4c2cf16ff93e4e4955f88db3ae9c28d6f7782

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
package cn.mb.cloud.gateway.filter;
 
 
import cn.mb.cloud.gateway.model.HeaderParams;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.util.MultiValueMap;
import reactor.core.publisher.Flux;
 
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
 
public class QueryParamsUtil {
 
 
    /**
     * 获取请求参数
     * @param request
     * @return
     */
    public static Map<String, Object> getQueryParams(ServerHttpRequest request) {
        HttpMethod method = request.getMethod();
        Map<String, Object> map = new HashMap<>();
        if("POST".equals(method.name())){
            MultiValueMap<String, String> queryParams = request.getQueryParams();
            Set<String> keySet = queryParams.keySet();
            for (String s : keySet) {
                try {
                    map.put(s, URLDecoder.decode(queryParams.getFirst(s), "UTF-8"));
                } catch (UnsupportedEncodingException e) {
                    e.printStackTrace();
                }
            }
            Flux<DataBuffer> body = request.getBody();
            body.subscribe(dataBuffer -> {
                byte[] bytes = new byte[dataBuffer.readableByteCount()];
                dataBuffer.read(bytes);
                String bodyStr = new String(bytes, StandardCharsets.UTF_8);
                String[] param = bodyStr.split("&");
                for (String s : param) {
                    String[] split = s.split("=");
                    try {
                        map.put(split[0], split.length == 1 ? "" : URLDecoder.decode(split[1], "UTF-8"));
                    } catch (UnsupportedEncodingException e) {
                        e.printStackTrace();
                    }
                }
            });
        }else{
            MultiValueMap<String, String> queryParams = request.getQueryParams();
            Set<String> keySet = queryParams.keySet();
            for (String s : keySet) {
                try {
                    map.put(s, URLDecoder.decode(queryParams.getFirst(s), "UTF-8"));
                } catch (UnsupportedEncodingException e) {
                    e.printStackTrace();
                }
            }
        }
        return map;
    }
 
 
    /**
     * 获取原始Header
     * @param request
     * @return
     */
    public static Map<String, Object> getOriginalHeaderParams(ServerHttpRequest request) {
        HttpHeaders headers = request.getHeaders();
        Map<String, Object> map = new HashMap<>();
        Set<String> keySet = headers.keySet();
        for (String s : keySet) {
            map.put(s, headers.getFirst(s));
        }
        return map;
    }
 
 
 
    /**
     * 获取请求header
     * @param request
     * @return
     */
    public static HeaderParams getHeaderParams(ServerHttpRequest request) {
        String authorization = request.getHeaders().getFirst("Authorization");
        String role = request.getHeaders().getFirst("role");
        String from = request.getHeaders().getFirst("from");
        String version = request.getHeaders().getFirst("version");
        String deviceid = request.getHeaders().getFirst("deviceid");
        String nonce = request.getHeaders().getFirst("nonce");
        String timestamp = request.getHeaders().getFirst("timestamp");
        HeaderParams headerParams = new HeaderParams();
        headerParams.setAuthorization(authorization);
        headerParams.setRole(role);
        headerParams.setFrom(from);
        headerParams.setVersion(version);
        headerParams.setDeviceid(deviceid);
        headerParams.setNonce(nonce);
        headerParams.setTimestamp(timestamp);
        return headerParams;
    }
 
 
 
}