package io.cattle.platform.servicediscovery.process;
import io.cattle.platform.activity.ActivityService;
import io.cattle.platform.core.constants.ServiceConstants;
import io.cattle.platform.core.model.Service;
import io.cattle.platform.engine.handler.HandlerResult;
import io.cattle.platform.engine.process.ProcessInstance;
import io.cattle.platform.engine.process.ProcessState;
import io.cattle.platform.json.JsonMapper;
import io.cattle.platform.object.util.DataAccessor;
import io.cattle.platform.process.base.AbstractDefaultProcessHandler;
import io.cattle.platform.servicediscovery.upgrade.UpgradeManager;
import javax.inject.Inject;
import javax.inject.Named;
@Named
public class ServiceFinishUpgrade extends AbstractDefaultProcessHandler {
@Inject
UpgradeManager upgradeManager;
@Inject
ActivityService activityService;
@Inject
JsonMapper jsonMapper;
@Override
public String[] getProcessNames() {
return new String[] { ServiceConstants.PROCESS_SERVICE_FINISH_UPGRADE };
}
@Override
public HandlerResult handle(ProcessState state, ProcessInstance process) {
final Service service = (Service) state.getResource();
activityService.run(service, "service.finishupgrade", "Finishing upgrade", new Runnable() {
@Override
public void run() {
io.cattle.platform.core.addon.ServiceUpgrade upgrade = DataAccessor.field(service,
ServiceConstants.FIELD_UPGRADE, jsonMapper,
io.cattle.platform.core.addon.ServiceUpgrade.class);
if (upgrade != null) {
upgradeManager.upgrade(service, upgrade.getStrategy(), ServiceConstants.STATE_FINISHING_UPGRADE);
}
upgradeManager.finishUpgrade(service, true);
}
});
return null;
}
}