Use limitedParallelism and MainScope() in AbacusThreadingImp.

dydxprotocol · Mar 13, 2024 · d62ef74 · d62ef74
1 parent 9a5ff0a
commit d62ef74
Show file tree

Hide file tree

Showing 4 changed files with 136 additions and 5 deletions.
diff --git a/.idea/.name b/.idea/.name
diff --git a/.idea/codeStyles/Project.xml b/.idea/codeStyles/Project.xml
diff --git a/.idea/codeStyles/codeStyleConfig.xml b/.idea/codeStyles/codeStyleConfig.xml
diff --git a/...rc/main/java/exchange/dydx/dydxstatemanager/protocolImplementations/AbacusThreadingImp.kt b/...rc/main/java/exchange/dydx/dydxstatemanager/protocolImplementations/AbacusThreadingImp.kt
@@ -2,16 +2,18 @@ package exchange.dydx.dydxstatemanager.protocolImplementations
 
 import exchange.dydx.abacus.protocols.ThreadingProtocol
 import exchange.dydx.abacus.protocols.ThreadingType
-import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.ExperimentalCoroutinesApi
+import kotlinx.coroutines.MainScope
 import kotlinx.coroutines.launch
-import kotlinx.coroutines.newSingleThreadContext
 import kotlinx.coroutines.plus
 
+@OptIn(ExperimentalCoroutinesApi::class)
 class AbacusThreadingImp : ThreadingProtocol {
-    private val mainScope = CoroutineScope(Dispatchers.Main) // + Job())
-    private val abacusScope = CoroutineScope(newSingleThreadContext("AbacusScope"))
-    private val networkScope = CoroutineScope(Dispatchers.IO)
+    private val mainScope = MainScope()
+    // Abacus runs lots of computations, but needs to be run without parallelism
+    private val abacusScope = MainScope() + Dispatchers.Default.limitedParallelism(1)
+    private val networkScope = MainScope() + Dispatchers.IO
     override fun async(type: ThreadingType, block: () -> Unit) {
         when (type) {
             ThreadingType.main ->