Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
41 changes: 33 additions & 8 deletions src/main/java/com/iemr/admin/utils/RestTemplateUtil.java
Original file line number Diff line number Diff line change
Expand Up @@ -25,14 +25,9 @@ public static HttpEntity<Object> createRequestEntity(Object body, String authori
return new HttpEntity<>(body, headers);
}
HttpServletRequest requestHeader = servletRequestAttributes.getRequest();
String jwtTokenFromCookie = null;
try {
jwtTokenFromCookie = CookieUtil.getJwtTokenFromCookie(requestHeader);

} catch (Exception e) {
logger.error("Error while getting jwtToken from Cookie" + e.getMessage() );
}


String jwtTokenFromCookie = extractJwttoken(requestHeader);

MultiValueMap<String, String> headers = new LinkedMultiValueMap<>();
headers.add(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE + ";charset=utf-8");
if(null != UserAgentContext.getUserAgent()) {
Expand All @@ -49,4 +44,34 @@ public static HttpEntity<Object> createRequestEntity(Object body, String authori
return new HttpEntity<>(body, headers);
}

private static String extractJwttoken(HttpServletRequest requestHeader) {
String jwtTokenFromCookie = null;
try {
jwtTokenFromCookie = CookieUtil.getJwtTokenFromCookie(requestHeader);

} catch (Exception e) {
logger.error("Error while getting jwtToken from Cookie" + e.getMessage() );
}
return jwtTokenFromCookie;
}

public static void getJwttokenFromHeaders(HttpHeaders headers) {
ServletRequestAttributes servletRequestAttributes = ((ServletRequestAttributes) RequestContextHolder
.getRequestAttributes());
if(servletRequestAttributes == null) {
return;
}
HttpServletRequest requestHeader = servletRequestAttributes.getRequest();
String jwtTokenFromCookie = extractJwttoken(requestHeader);
if (null != UserAgentContext.getUserAgent()) {
headers.add(HttpHeaders.USER_AGENT, UserAgentContext.getUserAgent());
}
if (null != jwtTokenFromCookie) {
headers.add(HttpHeaders.COOKIE, Constants.JWT_TOKEN+"="+jwtTokenFromCookie);
} else if (null != requestHeader.getHeader(Constants.JWT_TOKEN)) {
headers.add(Constants.JWT_TOKEN, requestHeader.getHeader(Constants.JWT_TOKEN));
}

}

}
12 changes: 11 additions & 1 deletion src/main/java/com/iemr/admin/utils/http/HttpUtils.java
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;

import com.iemr.admin.utils.RestTemplateUtil;
import com.sun.jersey.multipart.FormDataBodyPart;
import com.sun.jersey.multipart.FormDataMultiPart;

Expand Down Expand Up @@ -76,7 +77,10 @@ public HttpUtils() {

public String get(String uri) {
String body;
HttpEntity<String> requestEntity = new HttpEntity<String>("", headers);
HttpHeaders localheaders = new HttpHeaders();
localheaders.add("Content-Type", "application/json");
RestTemplateUtil.getJwttokenFromHeaders(localheaders);
HttpEntity<String> requestEntity = new HttpEntity<String>("", localheaders);
ResponseEntity<String> responseEntity = rest.exchange(uri, HttpMethod.GET, requestEntity, String.class);
setStatus((HttpStatus) responseEntity.getStatusCode());
// if (status == HttpStatus.OK){
Expand All @@ -98,6 +102,7 @@ public String get(String uri, HashMap<String, Object> header) {
} else {
headers.add("Content-Type", MediaType.APPLICATION_JSON);
}
RestTemplateUtil.getJwttokenFromHeaders(headers);
HttpEntity<String> requestEntity = new HttpEntity<String>("", headers);
ResponseEntity<String> responseEntity = rest.exchange(uri, HttpMethod.GET, requestEntity, String.class);
setStatus((HttpStatus) responseEntity.getStatusCode());
Expand All @@ -107,6 +112,7 @@ public String get(String uri, HashMap<String, Object> header) {

public String post(String uri, String json) {
String body;
RestTemplateUtil.getJwttokenFromHeaders(headers);
HttpEntity<String> requestEntity = new HttpEntity<String>(json, headers);
ResponseEntity<String> responseEntity = rest.exchange(uri, HttpMethod.POST, requestEntity, String.class);
setStatus((HttpStatus) responseEntity.getStatusCode());
Expand All @@ -126,6 +132,7 @@ public String post(String uri, String data, HashMap<String, Object> header) {
if (header.containsKey("X-APIkey-Header")) {
headers.add("X-APIkey-Header", header.get("X-APIkey-Header").toString());
}
RestTemplateUtil.getJwttokenFromHeaders(headers);
// headers.add("Content-Type", MediaType.APPLICATION_JSON);
ResponseEntity<String> responseEntity = new ResponseEntity(HttpStatus.BAD_REQUEST);
HttpEntity<String> requestEntity;
Expand All @@ -137,6 +144,7 @@ public String post(String uri, String data, HashMap<String, Object> header) {
}
public String put(String uri, String json) {
String body;
RestTemplateUtil.getJwttokenFromHeaders(headers);
HttpEntity<String> requestEntity = new HttpEntity<String>(json, headers);
ResponseEntity<String> responseEntity = rest.exchange(uri, HttpMethod.PUT, requestEntity, String.class);
setStatus((HttpStatus) responseEntity.getStatusCode());
Expand All @@ -159,6 +167,7 @@ public String put(String uri, String data, HashMap<String, Object> header) {
// headers.add("Content-Type", MediaType.APPLICATION_JSON);
ResponseEntity<String> responseEntity = new ResponseEntity(HttpStatus.BAD_REQUEST);
HttpEntity<String> requestEntity;
RestTemplateUtil.getJwttokenFromHeaders(headers);
requestEntity = new HttpEntity<String>(data, headers);
responseEntity = rest.exchange(uri, HttpMethod.PUT, requestEntity, String.class);
setStatus((HttpStatus) responseEntity.getStatusCode());
Expand Down Expand Up @@ -188,6 +197,7 @@ public String uploadFile(String uri, String data, HashMap<String, Object> header
multiPart.bodyPart(filePart);
multiPart.field("docPath", data);
headers.add("Content-Type", MediaType.APPLICATION_JSON);
RestTemplateUtil.getJwttokenFromHeaders(headers);
requestEntity = new HttpEntity<FormDataMultiPart>(multiPart, headers);// new
// HttpEntity<String>(multiPart,
// headers);
Expand Down
Loading