From a1162eaf89ab29ea95c1deb6d48190cd0705c340 Mon Sep 17 00:00:00 2001
From: tangwanchao <18284228392@163.com>
Date: Fri, 5 Feb 2021 11:01:38 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=E5=BC=BA=E5=88=B6=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=20Source?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .idea/gradle.xml                              |  1 +
 .../src/main/java/me/twc/source/ForceHttp.kt  | 37 +++++++++++++++++++
 2 files changed, 38 insertions(+)
 create mode 100644 source/src/main/java/me/twc/source/ForceHttp.kt

diff --git a/.idea/gradle.xml b/.idea/gradle.xml
index 0d1fb00..cba5394 100644
--- a/.idea/gradle.xml
+++ b/.idea/gradle.xml
@@ -1,5 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
+  <component name="GradleMigrationSettings" migrationVersion="1" />
   <component name="GradleSettings">
     <option name="linkedExternalProjectsSettings">
       <GradleProjectSettings>
diff --git a/source/src/main/java/me/twc/source/ForceHttp.kt b/source/src/main/java/me/twc/source/ForceHttp.kt
new file mode 100644
index 0000000..433ef28
--- /dev/null
+++ b/source/src/main/java/me/twc/source/ForceHttp.kt
@@ -0,0 +1,37 @@
+@file:Suppress("unused")
+
+package me.twc.source
+
+import kotlinx.coroutines.delay
+
+/**
+ * @author 唐万超
+ * @date 2021/02/05
+ */
+
+suspend fun <T> forceHttpSource(
+    block: suspend () -> ISource<T>
+): Source<T> = forceHttp(runInIo = false, block)
+
+suspend fun <T> forceHttpContext(
+    block: suspend () -> ISource<T>
+): Source<T> = forceHttp(runInIo = true, block)
+
+
+/**
+ * @param runInIo [true : 切换上下文到 IO],[false : 使用当前上下文]
+ *
+ * @param block 实际获取 ISource 的函数
+ */
+private suspend fun <T> forceHttp(
+    runInIo: Boolean,
+    block: suspend () -> ISource<T>
+): Source<T> {
+    while (true) {
+        val source = if (runInIo) httpSource(block) else httpContext(block)
+        if (source is SuccessSource<T>) {
+            return source
+        }
+        delay(1_000L)
+    }
+}
\ No newline at end of file