Add support for "balance" option

Closes #462

Author: nickkkccc

src/main/java/io/tarantool/driver/api/space/options/SelectOptions.java

@@ -1,5 +1,6 @@
 package io.tarantool.driver.api.space.options;
 
+import io.tarantool.driver.api.space.options.crud.OperationWithBalanceOptions;
 import io.tarantool.driver.api.space.options.crud.OperationWithBatchSizeOptions;
 import io.tarantool.driver.api.space.options.crud.OperationWithBucketIdOptions;
 import io.tarantool.driver.api.space.options.crud.OperationWithFieldsOptions;
@@ -16,5 +17,5 @@
  */
 public interface SelectOptions<T extends SelectOptions<T>>
     extends OperationWithBucketIdOptions<T>, OperationWithTimeoutOptions<T>, OperationWithFieldsOptions<T>,
-    OperationWithModeOptions<T>, OperationWithBatchSizeOptions<T> {
+            OperationWithModeOptions<T>, OperationWithBatchSizeOptions<T>, OperationWithBalanceOptions<T> {
 }

src/main/java/io/tarantool/driver/api/space/options/crud/OperationWithBalanceOptions.java

@@ -0,0 +1,33 @@
+package io.tarantool.driver.api.space.options.crud;
+
+import java.util.Optional;
+
+import io.tarantool.driver.api.space.options.Options;
+import io.tarantool.driver.api.space.options.Self;
+import io.tarantool.driver.api.space.options.crud.enums.ProxyOption;
+
+/**
+ * Base interface for all operation options that may have a configurable "balance" option.
+ *
+ * @author Belonogov Nikolay
+ */
+public interface OperationWithBalanceOptions<T extends OperationWithBalanceOptions<T>> extends Options, Self<T> {
+
+    /**
+     * Sets "balance" option value to true. If true then use replica according to vshard load balancing policy. Default
+     * value is false.
+     *
+     * @return this options instance.
+     */
+    default T balance() {
+        addOption(ProxyOption.BALANCE, true);
+        return self();
+    }
+
+    /**
+     * @return "balance" option value.
+     */
+    default Optional<Boolean> getBalance() {
+        return getOption(ProxyOption.BALANCE, Boolean.class);
+    }
+}

src/main/java/io/tarantool/driver/api/space/options/crud/enums/ProxyOption.java

@@ -23,7 +23,9 @@ public enum ProxyOption {
 
     AFTER("after"),
 
-    FIRST("first");
+    FIRST("first"),
+
+    BALANCE("balance");
 
     private final String name;
 

src/test/java/io/tarantool/driver/integration/proxy/options/ProxySpaceSelectOptionsIT.java

@@ -8,6 +8,7 @@
 import io.tarantool.driver.api.space.options.SelectOptions;
 import io.tarantool.driver.api.space.TarantoolSpaceOperations;
 import io.tarantool.driver.api.space.options.ProxySelectOptions;
+import io.tarantool.driver.api.space.options.crud.enums.ProxyOption;
 import io.tarantool.driver.api.tuple.DefaultTarantoolTupleFactory;
 import io.tarantool.driver.api.tuple.TarantoolTuple;
 import io.tarantool.driver.api.tuple.TarantoolTupleFactory;
@@ -28,6 +29,7 @@
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 /**
  * @author Artyom Dubinin
@@ -174,4 +176,20 @@ public void withModeTest() throws ExecutionException, InterruptedException {
         crudSelectOpts = client.eval("return crud_select_opts").get();
         assertEquals(Mode.WRITE.value(), ((HashMap<?, ?>) crudSelectOpts.get(0)).get("mode"));
     }
+
+    @Test
+    public void withBalanceTest() {
+        TarantoolSpaceOperations<TarantoolTuple, TarantoolResult<TarantoolTuple>> operations =
+            client.space(TEST_SPACE_NAME);
+
+        SelectOptions<ProxySelectOptions> options = ProxySelectOptions.create().balance();
+
+        assertTrue(options.getBalance().isPresent());
+        assertTrue(options.getBalance().get());
+
+        operations.select(Conditions.any(), options).join();
+        List<?> crudSelectOpts = client.eval("return crud_select_opts").join();
+
+        assertEquals(true, ((HashMap<?, ?>) crudSelectOpts.get(0)).get(ProxyOption.BALANCE.toString()));
+    }
 }