Add to upgrade pod.

openlattice · Mar 4, 2022 · 9650a0f · 9650a0f
1 parent d30e2c4
commit 9650a0f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/main/kotlin/com/openlattice/mechanic/pods/MechanicUpgradePod.kt b/src/main/kotlin/com/openlattice/mechanic/pods/MechanicUpgradePod.kt
@@ -72,6 +72,7 @@ import com.openlattice.linking.graph.PostgresLinkingQueryService
 import com.openlattice.mechanic.MechanicCli.Companion.UPGRADE
 import com.openlattice.mechanic.Toolbox
 import com.openlattice.mechanic.upgrades.DeleteOrgMetadataEntitySets
+import com.openlattice.mechanic.upgrades.ExportOrganizationMembers
 import com.openlattice.mechanic.upgrades.V3StudyMigrationUpgrade
 import com.openlattice.organizations.roles.HazelcastPrincipalService
 import com.openlattice.organizations.roles.SecurePrincipalsManager
@@ -379,6 +380,11 @@ class MechanicUpgradePod {
         )
     }
 
+    @Bean
+    fun exportOrganizationMembers() : ExportOrganizationMembers {
+        return ExportOrganizationMembers(toolbox, hikariDataSource, principalService(), hazelcastInstance)
+    }
+
     @PostConstruct
     fun post() {
         Principals.init(principalService(), hazelcastInstance)

diff --git a/src/main/kotlin/com/openlattice/mechanic/upgrades/ExportOrganizationMembers.kt b/src/main/kotlin/com/openlattice/mechanic/upgrades/ExportOrganizationMembers.kt
@@ -3,7 +3,7 @@ package com.openlattice.mechanic.upgrades
 import com.hazelcast.core.HazelcastInstance
 import com.openlattice.hazelcast.HazelcastMap
 import com.openlattice.mechanic.Toolbox
-import com.openlattice.organizations.roles.HazelcastPrincipalService
+import com.openlattice.organizations.roles.SecurePrincipalsManager
 import com.zaxxer.hikari.HikariDataSource
 import org.slf4j.LoggerFactory
 
@@ -14,7 +14,7 @@ import org.slf4j.LoggerFactory
 class ExportOrganizationMembers(
     private val toolbox: Toolbox,
     private val hds: HikariDataSource,
-    private val principalService: HazelcastPrincipalService,
+    private val principalService: SecurePrincipalsManager,
     hazelcast :HazelcastInstance
 ) : Upgrade {
     companion object {