✨ Add the channel param for enableTrafficControl function

zegoim · Oct 9, 2023 · 86de446 · 86de446
1 parent de465b1
commit 86de446
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/...id/src/main/java/im/zego/zego_express_engine/internal/ZegoExpressEngineMethodHandler.java b/...id/src/main/java/im/zego/zego_express_engine/internal/ZegoExpressEngineMethodHandler.java
@@ -1024,8 +1024,9 @@ public static void enableTrafficControl(MethodCall call, Result result) {
 
         boolean enable = ZegoUtils.boolValue((Boolean) call.argument("enable"));
         int property = ZegoUtils.intValue((Number) call.argument("property"));
+        ZegoPublishChannel channel = ZegoPublishChannel.getZegoPublishChannel(ZegoUtils.intValue((Number) call.argument("channel")));
 
-        ZegoExpressEngine.getEngine().enableTrafficControl(enable, property);
+        ZegoExpressEngine.getEngine().enableTrafficControl(enable, property, channel);
 
         result.success(null);
     }

diff --git a/ios/Classes/internal/ZegoExpressEngineMethodHandler.m b/ios/Classes/internal/ZegoExpressEngineMethodHandler.m
@@ -874,8 +874,9 @@ - (void)enableTrafficControl:(FlutterMethodCall *)call result:(FlutterResult)res
 
     BOOL enable = [ZegoUtils boolValue:call.arguments[@"enable"]];
     int property = [ZegoUtils intValue:call.arguments[@"property"]];
+    int channel = [ZegoUtils intValue:call.arguments[@"channel"]];
 
-    [[ZegoExpressEngine sharedEngine] enableTrafficControl:enable property:(ZegoTrafficControlProperty)property];
+    [[ZegoExpressEngine sharedEngine] enableTrafficControl:enable property:(ZegoTrafficControlProperty)property channel:(ZegoPublishChannel)channel];
 
     result(nil);
 }

diff --git a/windows/internal/ZegoExpressEngineMethodHandler.cpp b/windows/internal/ZegoExpressEngineMethodHandler.cpp
@@ -3216,8 +3216,10 @@ void ZegoExpressEngineMethodHandler::enableTrafficControl(
     std::unique_ptr<flutter::MethodResult<flutter::EncodableValue>> result) {
     auto enable = std::get<bool>(argument[FTValue("enable")]);
     auto property = std::get<int32_t>(argument[FTValue("property")]);
+    auto channel = std::get<int32_t>(argument[FTValue("channel")]);
 
-    EXPRESS::ZegoExpressSDK::getEngine()->enableTrafficControl(enable, property);
+    EXPRESS::ZegoExpressSDK::getEngine()->enableTrafficControl(enable, property, 
+        (EXPRESS::ZegoPublishChannel)channel);
     result->Success();
 }