Skip to content

Commit 7bf7692

Browse files
committed
2 parents 7a02d8a + 415005f commit 7bf7692

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

core/graphman/src/commands/deployment/pause.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,12 +39,11 @@ pub fn load_active_deployment(
3939
) -> Result<ActiveDeployment, PauseDeploymentError> {
4040
let mut primary_conn = primary_pool.get().map_err(GraphmanError::from)?;
4141

42-
let locator = crate::deployment::load_deployment(
42+
let locator = crate::deployment::load_deployment_locator(
4343
&mut primary_conn,
4444
deployment,
4545
&DeploymentVersionSelector::All,
46-
)?
47-
.locator();
46+
)?;
4847

4948
let mut catalog_conn = catalog::Connection::new(primary_conn);
5049

core/graphman/src/commands/deployment/resume.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,12 +39,11 @@ pub fn load_paused_deployment(
3939
) -> Result<PausedDeployment, ResumeDeploymentError> {
4040
let mut primary_conn = primary_pool.get().map_err(GraphmanError::from)?;
4141

42-
let locator = crate::deployment::load_deployment(
42+
let locator = crate::deployment::load_deployment_locator(
4343
&mut primary_conn,
4444
deployment,
4545
&DeploymentVersionSelector::All,
46-
)?
47-
.locator();
46+
)?;
4847

4948
let mut catalog_conn = catalog::Connection::new(primary_conn);
5049

core/graphman/src/deployment.rs

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -127,13 +127,15 @@ pub(crate) fn load_deployments(
127127
query.load(primary_conn).map_err(Into::into)
128128
}
129129

130-
pub(crate) fn load_deployment(
130+
pub(crate) fn load_deployment_locator(
131131
primary_conn: &mut PgConnection,
132132
deployment: &DeploymentSelector,
133133
version: &DeploymentVersionSelector,
134-
) -> Result<Deployment, GraphmanError> {
135-
let deployment = load_deployments(primary_conn, deployment, version)?
134+
) -> Result<DeploymentLocator, GraphmanError> {
135+
let deployment_locator = load_deployments(primary_conn, deployment, version)?
136136
.into_iter()
137+
.map(|deployment| deployment.locator())
138+
.unique()
137139
.exactly_one()
138140
.map_err(|err| {
139141
let count = err.into_iter().count();
@@ -142,5 +144,5 @@ pub(crate) fn load_deployment(
142144
))
143145
})?;
144146

145-
Ok(deployment)
147+
Ok(deployment_locator)
146148
}

0 commit comments

Comments
 (0)