Add post and options requests for version mismatch (#3535)

* Add post and options requests for version mismatch
airbytehq · May 21, 2021 · 38c1d6c · 38c1d6c
1 parent 8862fba
commit 38c1d6c
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/airbyte-server/src/main/java/io/airbyte/server/version_mismatch/VersionMismatchServer.java b/airbyte-server/src/main/java/io/airbyte/server/version_mismatch/VersionMismatchServer.java
@@ -80,18 +80,34 @@ public static class VersionMismatchServlet extends HttpServlet {
     // this error message should be overwritten before any requests are served
     public static String ERROR_MESSAGE = "Versions don't match!";
 
+    public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException {
+      this.serveDefaultRequest(response);
+    }
+
     public void doGet(HttpServletRequest request, HttpServletResponse response) throws IOException {
+      this.serveDefaultRequest(response);
+    }
+
+    public void doOptions(HttpServletRequest request, HttpServletResponse response) throws IOException {
+      this.addCorsHeaders(response);
+    }
+
+    private void serveDefaultRequest(HttpServletResponse response) throws IOException {
       var outputMap = ImmutableMap.of("error", ERROR_MESSAGE);
 
-      for (Map.Entry<String, String> entry : CorsFilter.MAP.entrySet()) {
-        response.setHeader(entry.getKey(), entry.getValue());
-      }
+      this.addCorsHeaders(response);
 
       response.setContentType("application/json");
       response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
       response.getWriter().println(Jsons.serialize(outputMap));
     }
 
+    private void addCorsHeaders(HttpServletResponse response) {
+      for (Map.Entry<String, String> entry : CorsFilter.MAP.entrySet()) {
+        response.setHeader(entry.getKey(), entry.getValue());
+      }
+    }
+
   }
 
 }