checked for condition if port not present

rudderlabs · Jul 24, 2020 · 55cb09b · 55cb09b
1 parent 59763cb
commit 55cb09b
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/Rudder/Classes/RSConfigBuilder.m b/Rudder/Classes/RSConfigBuilder.m
@@ -17,7 +17,12 @@ - (instancetype) withEndPointUrl:(NSString *)endPointUrl{
         config = [[RSConfig alloc] init];
     }
     NSURL *url = [[NSURL alloc] initWithString:endPointUrl];
+    if([url port]){
     config.dataPlaneUrl = [[NSString alloc] initWithFormat:@"%@://%@:%@", [url scheme], [url host], [url port]];
+    }
+    else{
+        config.dataPlaneUrl = [[NSString alloc] initWithFormat:@"%@://%@", [url scheme], [url host]];
+    }
     return self;
 }
 
@@ -27,15 +32,24 @@ - (instancetype) withDataPlaneUrl: (NSString*) dataPlaneUrl {
     }
 
     NSURL *url = [[NSURL alloc] initWithString:dataPlaneUrl];
+    if([url port]){
     config.dataPlaneUrl = [[NSString alloc] initWithFormat:@"%@://%@:%@", [url scheme], [url host], [url port]];
+    }else{
+            config.dataPlaneUrl = [[NSString alloc] initWithFormat:@"%@://%@", [url scheme], [url host]];
+    }
     return self;
 }
 
 - (instancetype)withDataPlaneURL:(NSURL *) dataPlaneURL {
     if (config == nil) {
         config = [[RSConfig alloc] init];
     }
+    if([dataPlaneURL port]){
     config.dataPlaneUrl = [[NSString alloc] initWithFormat:@"%@://%@:%@", [dataPlaneURL scheme], [dataPlaneURL host] ,[dataPlaneURL port]];
+    }
+    else{
+        config.dataPlaneUrl = [[NSString alloc] initWithFormat:@"%@://%@", [dataPlaneURL scheme], [dataPlaneURL host]];
+    }
     return self;
 }