diff --git a/mars-server/src/main/java/com/mars/server/server/request/HttpMarsRequest.java b/mars-server/src/main/java/com/mars/server/server/request/HttpMarsRequest.java index 80b02d4..971b49b 100644 --- a/mars-server/src/main/java/com/mars/server/server/request/HttpMarsRequest.java +++ b/mars-server/src/main/java/com/mars/server/server/request/HttpMarsRequest.java @@ -76,7 +76,20 @@ public class HttpMarsRequest { * @param key 键 * @return 头数据 */ - public List getHeader(String key) { + public String getHeader(String key) { + List headers = getHeaders(key); + if(headers == null || headers.size() < 1){ + return null; + } + return headers.get(0); + } + + /** + * 获取请求头数据 + * @param key 键 + * @return 头数据 + */ + public List getHeaders(String key) { return httpExchange.getRequestHeaders().get(key); }