From bf57bf717c000a501a7df6df7267e2a5186feb99 Mon Sep 17 00:00:00 2001
From: EMQ-YangM <yangm@emqx.io>
Date: Wed, 23 Feb 2022 18:37:34 +0800
Subject: [PATCH] feat(emqx_resource_instance): add wait_connected Opts

---
 apps/emqx_resource/src/emqx_resource_instance.erl | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/apps/emqx_resource/src/emqx_resource_instance.erl b/apps/emqx_resource/src/emqx_resource_instance.erl
index 115f8426a..e08c57467 100644
--- a/apps/emqx_resource/src/emqx_resource_instance.erl
+++ b/apps/emqx_resource/src/emqx_resource_instance.erl
@@ -178,6 +178,16 @@ do_recreate(InstId, ResourceType, NewConfig, Opts) ->
             {error, not_found}
     end.
 
+wait_for_resource_ready(InstId, 0) ->
+    force_lookup(InstId);
+wait_for_resource_ready(InstId, Retry) ->
+    case force_lookup(InstId) of
+        #{resource_data := #{status := connected}} = Data -> Data;
+        _ ->
+            timer:sleep(100),
+            wait_for_resource_ready(InstId, Retry-1)
+    end.
+
 do_create(InstId, Group, ResourceType, Config, Opts) ->
     case lookup(InstId) of
         {ok,_, _} ->
@@ -187,7 +197,8 @@ do_create(InstId, Group, ResourceType, Config, Opts) ->
                 ok ->
                     ok = emqx_plugin_libs_metrics:create_metrics(resource_metrics, InstId,
                             [matched, success, failed, exception], [matched]),
-                    {ok, force_lookup(InstId)};
+                    WaitTime = maps:get(wait_connected, Opts, 0),
+                    {ok, wait_for_resource_ready(InstId, WaitTime div 100)};
                 Error ->
                     Error
             end