basic retry functions

Author: ggilbert-unicon

pom.xml

@@ -5,7 +5,7 @@
 
   <groupId>apereo</groupId>
   <artifactId>data-loader-library</artifactId>
-  <version>1.0.3</version>
+  <version>1.0.4</version>
   <packaging>jar</packaging>
 
   <name>apereo-data-loader-library</name>

src/main/java/unicon/matthews/dataloader/MatthewsClient.java

@@ -124,8 +124,16 @@ public void postEnrollment(Enrollment enrollment) {
     String path = "/api/classes/{classSourcedId}/enrollments";
     String url = this.baseUrl + StringUtils.replace(path, "{classSourcedId}", enrollment.getKlass().getSourcedId());
 
-    restTemplate
-    .exchange(url, HttpMethod.POST, he, JsonObject.class);
+    ResponseEntity<JsonObject> response =
+        restTemplate
+          .exchange(url, HttpMethod.POST, he, JsonObject.class);
+    
+    if (response != null && response.getStatusCode() == HttpStatus.UNAUTHORIZED) {
+      httpHeaders.add("Authorization", "Bearer "+ getToken());
+      restTemplate
+      .exchange(url, HttpMethod.POST, he, JsonObject.class);      
+    }
+
   }
 
   public void postUser(User user) {
@@ -139,7 +147,7 @@ public void postUser(User user) {
         .exchange(url, HttpMethod.POST, he, JsonObject.class);
 
 
-    if (response != null && response.getStatusCode() == HttpStatus.FORBIDDEN) {
+    if (response != null && response.getStatusCode() == HttpStatus.UNAUTHORIZED) {
       httpHeaders.add("Authorization", "Bearer "+ getToken());
       restTemplate
       .exchange(url, HttpMethod.POST, he, JsonObject.class);      
@@ -153,9 +161,16 @@ public void postUserMapping(UserMapping userMapping) {
     String path = "/api/users/mapping";
     String url = this.baseUrl + path;
 
-    restTemplate
+    ResponseEntity<JsonObject> response = 
+        restTemplate
         .exchange(url, HttpMethod.POST, he, JsonObject.class);
 
+    if (response != null && response.getStatusCode() == HttpStatus.UNAUTHORIZED) {
+      httpHeaders.add("Authorization", "Bearer "+ getToken());
+      restTemplate
+      .exchange(url, HttpMethod.POST, he, JsonObject.class);      
+    }
+
   }
 
   public void postLineItem(LineItem lineItem) {