updated UserApiExample class

paypay · Oct 12, 2020 · d9cca4b · d9cca4b
1 parent e3c4546
commit d9cca4b
Showing 1 changed file with 10 additions and 13 deletions.
diff --git a/src/main/java/jp/ne/paypay/example/UserApiExample.java b/src/main/java/jp/ne/paypay/example/UserApiExample.java
@@ -20,23 +20,20 @@ public static void main(String[] args) throws ApiException {
 
     UserApi userApi = new UserApi(apiClient);
 
-    getUserAuthorizationStatus(userAuthorizationId, userApi);
+    getOrUnlinkUser(userAuthorizationId, userApi, false);
     //Replace the "userAuthorizationId" with the actual one. Please note that this will Unlink the user from the client
-    unlinkUser("userAuthorizationId", userApi);
+    getOrUnlinkUser("userAuthorizationId", userApi, true);
   }
 
-  private static void getUserAuthorizationStatus(String userAuthorizationId, UserApi userApi) {
+  private static void getOrUnlinkUser(String userAuthorizationId, UserApi userApi, boolean unlinkUser) {
     try{
-      UserAuthorizationStatus userAuthorizationStatus = userApi.getUserAuthorizationStatus(userAuthorizationId);
-      System.out.println(userAuthorizationStatus);
-    }catch (ApiException e){
-      System.out.println(e.getResponseBody());
-    }
-  }
-  private static void unlinkUser(String userAuthorizationId, UserApi userApi) {
-    try{
-      NotDataResponse notDataResponse = userApi.unlinkUser(userAuthorizationId);
-      System.out.println(notDataResponse);
+      if(unlinkUser){
+        NotDataResponse notDataResponse = userApi.unlinkUser(userAuthorizationId);
+        System.out.println(notDataResponse);
+      }else{
+        UserAuthorizationStatus userAuthorizationStatus = userApi.getUserAuthorizationStatus(userAuthorizationId);
+        System.out.println(userAuthorizationStatus);
+      }
     }catch (ApiException e){
       System.out.println(e.getResponseBody());
     }