添加 CoroutineScope 扩展方法

shenbengit · Oct 25, 2023 · db128b2 · db128b2
1 parent d525734
commit db128b2
Show file tree

Hide file tree

Showing 4 changed files with 88 additions and 16 deletions.
diff --git a/lib/build.gradle b/lib/build.gradle
@@ -85,7 +85,7 @@ afterEvaluate {
                 // You can then customize attributes of the publication as shown below.
                 groupId = 'com.shencoder'
                 artifactId = 'mvvmkit'
-                version = '1.1.5'
+                version = '1.1.6'
             }
         }
     }

diff --git a/lib/src/main/java/com/shencoder/mvvmkit/ext/BaseViewModelExt.kt b/lib/src/main/java/com/shencoder/mvvmkit/ext/BaseViewModelExt.kt
@@ -66,15 +66,15 @@ fun <T> BaseViewModel<*>.httpRequest(
             onSuccess(ResultStatus.onSuccess(it.getResponseData()))
         } else {
             //处理公共的失败事件
-            BaseViewModelExt.defaultHttpRequestHandleFailure?.invoke(
+            DefaultHttpRequestHandle.defaultHttpRequestFailure?.invoke(
                 it.getResponseCode(),
                 it.getResponseMsg()
             )
             onFailure(ResultStatus.onFailure(it.getResponseCode(), it.getResponseMsg()))
         }
     }, {
         //处理公共的异常事件
-        BaseViewModelExt.defaultHttpRequestHandleError?.invoke(it)
+        DefaultHttpRequestHandle.defaultHttpRequestError?.invoke(it)
         onError(ResultStatus.onError(it))
     })
 
@@ -139,16 +139,4 @@ fun BaseViewModel<*>.toastWarning(
 fun BaseViewModel<*>.toastNormal(
     text: String,
     @ToastDuration duration: Int = Toast.LENGTH_SHORT
-) = applicationContext.toastNormal(text, duration)
-
-object BaseViewModelExt {
-    /**
-     * 默认处理公共的失败结果
-     */
-    var defaultHttpRequestHandleFailure: ((code: Int, msg: String) -> Unit)? = null
-
-    /**
-     * 默认处理异常的失败结果
-     */
-    var defaultHttpRequestHandleError: ((error: Throwable) -> Unit)? = null
-}
+) = applicationContext.toastNormal(text, duration)
diff --git a/lib/src/main/java/com/shencoder/mvvmkit/ext/CoroutineScopeExt.kt b/lib/src/main/java/com/shencoder/mvvmkit/ext/CoroutineScopeExt.kt
@@ -0,0 +1,64 @@
+package com.shencoder.mvvmkit.ext
+
+import com.shencoder.mvvmkit.http.ResultStatus
+import com.shencoder.mvvmkit.http.bean.BaseResponse
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.launch
+import kotlinx.coroutines.withContext
+
+/**
+ *
+ * @author  ShenBen
+ * @date    2023/10/25 21:40
+ * @email   714081644@qq.com
+ */
+
+fun <T> CoroutineScope.httpRequest(
+    block: suspend CoroutineScope.() -> BaseResponse<T>,
+    onSuccess: (ResultStatus.Success<T?>) -> Unit = {},
+    onFailure: (ResultStatus.Failure) -> Unit = {},
+    onError: (ResultStatus.Error) -> Unit = {},
+    onComplete: () -> Unit = {}
+) = launch(Dispatchers.Main) {
+    val result = runCatching {
+        withContext(Dispatchers.IO, block)
+    }
+
+    result.fold({
+        if (it.isSuccess()) {
+            onSuccess(ResultStatus.onSuccess(it.getResponseData()))
+        } else {
+            //处理公共的失败事件
+            DefaultHttpRequestHandle.defaultHttpRequestFailure?.invoke(
+                it.getResponseCode(),
+                it.getResponseMsg()
+            )
+            onFailure(ResultStatus.onFailure(it.getResponseCode(), it.getResponseMsg()))
+        }
+    }, {
+        //处理公共的异常事件
+        DefaultHttpRequestHandle.defaultHttpRequestError?.invoke(it)
+        onError(ResultStatus.onError(it))
+    })
+
+    onComplete()
+}
+
+fun <T> CoroutineScope.request(
+    block: suspend CoroutineScope.() -> T,
+    onSuccess: (ResultStatus.Success<T?>) -> Unit = {},
+    onError: (ResultStatus.Error) -> Unit = {},
+    onComplete: () -> Unit = {}
+) = launch(Dispatchers.Main) {
+    val result = runCatching {
+        withContext(Dispatchers.IO, block)
+    }
+    result.fold({
+        onSuccess(ResultStatus.onSuccess(it))
+    }, {
+        onError(ResultStatus.onError(it))
+    })
+
+    onComplete()
+}
diff --git a/lib/src/main/java/com/shencoder/mvvmkit/ext/DefaultHttpRequestHandle.kt b/lib/src/main/java/com/shencoder/mvvmkit/ext/DefaultHttpRequestHandle.kt
@@ -0,0 +1,20 @@
+package com.shencoder.mvvmkit.ext
+
+/**
+ *
+ * @author  ShenBen
+ * @date    2023/10/25 21:41
+ * @email   714081644@qq.com
+ */
+object DefaultHttpRequestHandle {
+
+    /**
+     * 默认处理公共的失败结果
+     */
+    var defaultHttpRequestFailure: ((code: Int, msg: String) -> Unit)? = null
+
+    /**
+     * 默认处理异常的失败结果
+     */
+    var defaultHttpRequestError: ((error: Throwable) -> Unit)? = null
+}