diff options
author | Morten Tokle <mortent@verizonmedia.com> | 2021-03-04 10:27:43 +0100 |
---|---|---|
committer | Morten Tokle <mortent@verizonmedia.com> | 2021-03-04 10:27:43 +0100 |
commit | 0964cd8a731d2628f03199fc2478140f1f84760f (patch) | |
tree | 833004c32b55d9ea71c146a548dab1b63bb43210 /controller-server | |
parent | a6c06485fd2894c88e30c6dc58efcb4669721ccc (diff) |
Add new methods to RoleService, create maintainer for maintaining roles
Diffstat (limited to 'controller-server')
2 files changed, 38 insertions, 0 deletions
diff --git a/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/maintenance/ControllerMaintenance.java b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/maintenance/ControllerMaintenance.java index 9f9a0f6d56f..d9a233eb475 100644 --- a/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/maintenance/ControllerMaintenance.java +++ b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/maintenance/ControllerMaintenance.java @@ -66,6 +66,7 @@ public class ControllerMaintenance extends AbstractComponent { maintainers.add(new EndpointCertificateMaintainer(controller, intervals.endpointCertificateMaintainer)); maintainers.add(new TrafficShareUpdater(controller, intervals.trafficFractionUpdater)); maintainers.add(new ArchiveUriUpdater(controller, intervals.archiveUriUpdater)); + maintainers.add(new TenantRoleMaintainer(controller, intervals.tenantRoleMaintainer)); } public Upgrader upgrader() { return upgrader; } @@ -117,6 +118,7 @@ public class ControllerMaintenance extends AbstractComponent { private final Duration endpointCertificateMaintainer; private final Duration trafficFractionUpdater; private final Duration archiveUriUpdater; + private final Duration tenantRoleMaintainer; public Intervals(SystemName system) { this.system = Objects.requireNonNull(system); @@ -145,6 +147,7 @@ public class ControllerMaintenance extends AbstractComponent { this.endpointCertificateMaintainer = duration(12, HOURS); this.trafficFractionUpdater = duration(5, MINUTES); this.archiveUriUpdater = duration(5, MINUTES); + this.tenantRoleMaintainer = duration(5, MINUTES); } private Duration duration(long amount, TemporalUnit unit) { diff --git a/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/maintenance/TenantRoleMaintainer.java b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/maintenance/TenantRoleMaintainer.java new file mode 100644 index 00000000000..e8b50a6b604 --- /dev/null +++ b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/maintenance/TenantRoleMaintainer.java @@ -0,0 +1,35 @@ +// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. + +package com.yahoo.vespa.hosted.controller.maintenance; + +import com.yahoo.vespa.flags.BooleanFlag; +import com.yahoo.vespa.flags.FetchVector; +import com.yahoo.vespa.flags.Flags; +import com.yahoo.vespa.hosted.controller.Controller; +import com.yahoo.vespa.hosted.controller.tenant.Tenant; + +import java.time.Duration; +import java.util.stream.Collectors; + +public class TenantRoleMaintainer extends ControllerMaintainer { + + private final BooleanFlag provisionTenantRoles; + + public TenantRoleMaintainer(Controller controller, Duration tenantRoleMaintainer) { + super(controller, tenantRoleMaintainer); + provisionTenantRoles = Flags.PROVISION_TENANT_ROLES.bindTo(controller.flagSource()); + } + + @Override + protected boolean maintain() { + var roleService = controller().serviceRegistry().roleService(); + var tenants = controller().tenants().asList(); + var tenantsWithRoles = tenants.stream() + .map(Tenant::name) + // Only maintain a subset of the tenants + .filter(name -> provisionTenantRoles.with(FetchVector.Dimension.TENANT_ID, name.value()).value()) + .collect(Collectors.toList()); + roleService.maintainRoles(tenantsWithRoles); + return true; + } +} |