diff --git a/src/main/kotlin/com/openlattice/mechanic/pods/MechanicUpgradePod.kt b/src/main/kotlin/com/openlattice/mechanic/pods/MechanicUpgradePod.kt index 3767ea48..9d03d914 100644 --- 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 index 414d5017..6f087ca1 100644 --- 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 {