fix spotless

streamnative · Nov 12, 2024 · 9f8a427 · 9f8a427
1 parent e42411d
commit 9f8a427
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/client/src/main/java/io/streamnative/oxia/client/grpc/WriteStreamKey.java b/client/src/main/java/io/streamnative/oxia/client/grpc/WriteStreamKey.java
@@ -0,0 +1,3 @@
+package io.streamnative.oxia.client.grpc;
+
+public record WriteStreamKey(long shard, int mod) {}
diff --git a/client/src/test/java/io/streamnative/oxia/client/batch/BatchTest.java b/client/src/test/java/io/streamnative/oxia/client/batch/BatchTest.java
@@ -174,7 +174,9 @@ public void setUp() throws Exception {
         final WriteStreamWrapper writeStreamWrapper = new WriteStreamWrapper(stub.async());
         clientByShardId = mock(OxiaStubProvider.class);
         lenient().when(clientByShardId.getStubForShard(anyLong())).thenReturn(stub);
-        lenient().when(clientByShardId.getWriteStreamForShard(anyLong())).thenReturn(writeStreamWrapper);
+        lenient()
+                .when(clientByShardId.getWriteStreamForShard(anyLong()))
+                .thenReturn(writeStreamWrapper);
     }
 
     @AfterEach
@@ -328,7 +330,8 @@ public void sendFail() {
         @Test
         public void sendFailNoClient() {
             var stubProvider = mock(OxiaStubProvider.class);
-            when(stubProvider.getWriteStreamForShard(anyLong())).thenThrow(new NoShardAvailableException(1));
+            when(stubProvider.getWriteStreamForShard(anyLong()))
+                    .thenThrow(new NoShardAvailableException(1));
 
             batch =
                     new WriteBatch(