add checks for authprovider

DynamoDS · Apr 16, 2024 · cb08cc7 · cb08cc7
1 parent 4e0dad7
commit cb08cc7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/DynamoMLDataPipeline/DynamoMLDataPipeline.cs b/src/DynamoMLDataPipeline/DynamoMLDataPipeline.cs
@@ -63,7 +63,7 @@ internal string ProductionCollectionID
         // Id of the user sending the post request.
         private string GetUserId()
         {
-            return AuthUserInfoProvider.UserId;
+            return AuthUserInfoProvider?.UserId;
         }
 
         // Authorization token needed for the restsharp post request in this pipeline.

diff --git a/src/DynamoMLDataPipeline/DynamoMLDataPipelineExtension.cs b/src/DynamoMLDataPipeline/DynamoMLDataPipelineExtension.cs
@@ -23,9 +23,17 @@ public string UniqueId
         public void Startup(StartupParams sp)
         {
             DynamoMLDataPipeline = new DynamoMLDataPipeline();
-            DynamoMLDataPipeline.AuthTokenProvider = (IOAuth2AccessTokenProvider)sp.AuthProvider;
-            DynamoMLDataPipeline.AuthUserInfoProvider = (IOAuth2UserIDProvider)sp.AuthProvider;
             DynamoMLDataPipeline.DynamoVersion = sp.DynamoVersion;
+
+            if (sp.AuthProvider is IOAuth2AccessTokenProvider tokenProvider)
+            {
+                DynamoMLDataPipeline.AuthTokenProvider = tokenProvider;
+            }
+
+            if (sp.AuthProvider is IOAuth2UserIDProvider userIdProvider)
+            {
+                DynamoMLDataPipeline.AuthUserInfoProvider = userIdProvider;
+            }
         }
 
         public void Ready(ReadyParams sp)