Fix the annotation for the refresh endpoint (#12075)

airbytehq · Apr 11, 2024 · 58a7969 · 58a7969
1 parent d5bd6d2
commit 58a7969
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/airbyte-server/src/main/java/io/airbyte/server/apis/ConnectionApiController.java b/airbyte-server/src/main/java/io/airbyte/server/apis/ConnectionApiController.java
@@ -131,8 +131,11 @@ public ConnectionReadList listConnectionsForWorkspacesPaginated(
     return ApiHelper.execute(() -> connectionsHandler.listConnectionsForWorkspaces(listConnectionsForWorkspacesRequestBody));
   }
 
+  @Post(uri = "/refresh")
+  @Secured({WORKSPACE_EDITOR, ORGANIZATION_EDITOR})
+  @ExecuteOn(AirbyteTaskExecutors.SCHEDULER)
   @Override
-  public BooleanRead refreshConnectionStream(ConnectionStreamRefreshRequestBody connectionStreamRefreshRequestBody) {
+  public BooleanRead refreshConnectionStream(@Body final ConnectionStreamRefreshRequestBody connectionStreamRefreshRequestBody) {
     return ApiHelper.execute(() -> new BooleanRead().value(streamRefreshesHandler.createRefreshesForConnection(
         connectionStreamRefreshRequestBody.getConnectionId(),
         connectionStreamRefreshRequestBody.getStreams() != null ? connectionStreamRefreshRequestBody.getStreams() : new ArrayList<>())));
@@ -252,15 +255,15 @@ public JobInfoRead resetConnectionStream(@Body final ConnectionStreamRequestBody
   @Post(uri = "/clear")
   @Secured({WORKSPACE_EDITOR, ORGANIZATION_EDITOR})
   @ExecuteOn(AirbyteTaskExecutors.SCHEDULER)
-  public JobInfoRead clearConnection(ConnectionIdRequestBody connectionIdRequestBody) {
+  public JobInfoRead clearConnection(@Body ConnectionIdRequestBody connectionIdRequestBody) {
     return ApiHelper.execute(() -> schedulerHandler.resetConnection(connectionIdRequestBody));
   }
 
   @Override
   @Post(uri = "/clear/stream")
   @Secured({WORKSPACE_EDITOR, ORGANIZATION_EDITOR})
   @ExecuteOn(AirbyteTaskExecutors.SCHEDULER)
-  public JobInfoRead clearConnectionStream(ConnectionStreamRequestBody connectionStreamRequestBody) {
+  public JobInfoRead clearConnectionStream(@Body ConnectionStreamRequestBody connectionStreamRequestBody) {
     return ApiHelper.execute(() -> schedulerHandler.resetConnectionStream(connectionStreamRequestBody));
   }