refactor: add default implementation ExchangeClientFactory string method

namjug-kim · Jun 20, 2019 · 9bdf772 · 9bdf772
1 parent bc6ba4b
commit 9bdf772
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 1 deletion.
diff --git a/...pto-binance/src/test/kotlin/com/njkim/reactivecrypto/binance/ExchangeClientFactoryTest.kt b/...pto-binance/src/test/kotlin/com/njkim/reactivecrypto/binance/ExchangeClientFactoryTest.kt
@@ -1,3 +1,19 @@
+/*
+ * Copyright 2019 namjug-kim
+ *
+ * LINE Corporation licenses this file to you under the Apache License,
+ * version 2.0 (the "License"); you may not use this file except in compliance
+ * with the License. You may obtain a copy of the License at:
+ *
+ *   https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * 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.njkim.reactivecrypto.binance
 
 import com.njkim.reactivecrypto.core.ExchangeClientFactory
@@ -15,4 +31,13 @@ class ExchangeClientFactoryTest {
         assertThat(exchangeWebsocketClient).isInstanceOf(ExchangeWebsocketClient::class.java)
         assertThat(exchangeWebsocketClient).isExactlyInstanceOf(BinanceWebsocketClient::class.java)
     }
-}
+
+    @Test
+    fun `create binance websocket client using string`() {
+        val exchangeWebsocketClient = ExchangeClientFactory.websocket("BINANCE")
+
+        assertThat(exchangeWebsocketClient).isNotNull
+        assertThat(exchangeWebsocketClient).isInstanceOf(ExchangeWebsocketClient::class.java)
+        assertThat(exchangeWebsocketClient).isExactlyInstanceOf(BinanceWebsocketClient::class.java)
+    }
+}
diff --git a/reactive-crypto-core/src/main/kotlin/com/njkim/reactivecrypto/core/ExchangeClientFactory.kt b/reactive-crypto-core/src/main/kotlin/com/njkim/reactivecrypto/core/ExchangeClientFactory.kt
@@ -25,6 +25,16 @@ import kotlin.reflect.full.createInstance
 
 class ExchangeClientFactory {
     companion object {
+        init {
+            ExchangeVendor.values()
+                .forEach { exchangeVendor ->
+                    ReactiveCryptoPlugins.customClientFactory
+                        .addHttpCustomFactory(exchangeVendor.name) { http(exchangeVendor) }
+                    ReactiveCryptoPlugins.customClientFactory
+                        .addWsCustomFactory(exchangeVendor.name) { websocket(exchangeVendor) }
+                }
+        }
+
         @JvmStatic
         fun websocket(exchangeVendor: ExchangeVendor): ExchangeWebsocketClient {
             return websocket(exchangeVendor.websocketClientName)