[QoS] Feature/qos client

atlasdb-api/build.gradle

@@ -13,6 +13,7 @@ dependencies {
   compile group: 'javax.validation', name: 'validation-api'
 
   compile group: 'com.palantir.remoting-api', name: 'ssl-config'
+  compile group: 'com.palantir.remoting3', name: 'jaxrs-clients'
 
   processor group: 'org.immutables', name: 'value'
 

atlasdb-api/src/main/java/com/palantir/atlasdb/qos/FakeQosClient.java

@@ -0,0 +1,31 @@
+/*
+ * Copyright 2017 Palantir Technologies, Inc. All rights reserved.
+ *
+ * Licensed under the BSD-3 License (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://opensource.org/licenses/BSD-3-Clause
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.palantir.atlasdb.qos;
+
+public class FakeQosClient implements QosClient {
+
+    private static final FakeQosClient DEFAULT = new FakeQosClient();
+
+    public static FakeQosClient getDefault() {
+        return DEFAULT;
+    }
+
+    @Override
+    public void checkLimit() {
+        // no op
+    }
+}

atlasdb-api/src/main/java/com/palantir/atlasdb/qos/QosClient.java

@@ -0,0 +1,21 @@
+/*
+ * Copyright 2017 Palantir Technologies, Inc. All rights reserved.
+ *
+ * Licensed under the BSD-3 License (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://opensource.org/licenses/BSD-3-Clause
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.palantir.atlasdb.qos;
+
+public interface QosClient {
+    void checkLimit();
+}

atlasdb-api/src/main/java/com/palantir/atlasdb/spi/AtlasDbFactory.java

@@ -23,6 +23,8 @@
 import com.palantir.atlasdb.config.LeaderConfig;
 import com.palantir.atlasdb.keyvalue.api.KeyValueService;
 import com.palantir.atlasdb.keyvalue.api.TableReference;
+import com.palantir.atlasdb.qos.FakeQosClient;
+import com.palantir.atlasdb.qos.QosClient;
 import com.palantir.timestamp.TimestampService;
 import com.palantir.timestamp.TimestampStoreInvalidator;
 
@@ -34,7 +36,8 @@ public interface AtlasDbFactory {
 
     default KeyValueService createRawKeyValueService(
             KeyValueServiceConfig config, Optional<LeaderConfig> leaderConfig) {
-        return createRawKeyValueService(config, leaderConfig, Optional.empty(), DEFAULT_INITIALIZE_ASYNC);
+        return createRawKeyValueService(config, leaderConfig, Optional.empty(), DEFAULT_INITIALIZE_ASYNC,
+                FakeQosClient.getDefault());
     }
 
     /**
@@ -46,13 +49,15 @@ default KeyValueService createRawKeyValueService(
      * absent. If both are present, they must match.
      * @param initializeAsync If the implementations supports it, and initializeAsync is true, the KVS will initialize
      * asynchronously when synchronous initialization fails.
+     * @param qosClient the client for checking limits from the Quality-of-Service service.
      * @return The requested KeyValueService instance
      */
     KeyValueService createRawKeyValueService(
             KeyValueServiceConfig config,
             Optional<LeaderConfig> leaderConfig,
             Optional<String> namespace,
-            boolean initializeAsync);
+            boolean initializeAsync,
+            QosClient qosClient);
 
     default TimestampService createTimestampService(KeyValueService rawKvs) {
         return createTimestampService(rawKvs, Optional.empty(), DEFAULT_INITIALIZE_ASYNC);