func (s *ManifoldSuite) SetUpTest(c *gc.C) { s.IsolationSuite.SetUpTest(c) s.manifold = upgradewaiter.Manifold(upgradewaiter.ManifoldConfig{ UpgradeStepsWaiterName: "steps-waiter", UpgradeCheckWaiterName: "check-waiter", }) }
// Manifolds returns a set of co-configured manifolds covering the // various responsibilities of a machine agent. // // Thou Shalt Not Use String Literals In This Function. Or Else. func Manifolds(config ManifoldsConfig) dependency.Manifolds { return dependency.Manifolds{ // The agent manifold references the enclosing agent, and is the // foundation stone on which most other manifolds ultimately depend. agentName: agent.Manifold(config.Agent), // The termination worker returns ErrTerminateAgent if a // termination signal is received by the process it's running // in. It has no inputs and its only output is the error it // returns. terminationName: terminationworker.Manifold(), // The api caller is a thin concurrent wrapper around a connection // to some API server. It's used by many other manifolds, which all // select their own desired facades. It will be interesting to see // how this works when we consolidate the agents; might be best to // handle the auth changes server-side..? apiCallerName: apicaller.Manifold(apicaller.ManifoldConfig{ AgentName: agentName, }), // The upgrade steps gate is used to coordinate workers which // shouldn't do anything until the upgrade-steps worker has // finished running any required upgrade steps. upgradeStepsGateName: gate.ManifoldEx(config.UpgradeStepsLock), // The upgrade check gate is used to coordinate workers which // shouldn't do anything until the upgrader worker has // completed it's first check for a new tools version to // upgrade to. upgradeCheckGateName: gate.ManifoldEx(config.UpgradeCheckLock), // The upgrader is a leaf worker that returns a specific error // type recognised by the machine agent, causing other workers // to be stopped and the agent to be restarted running the new // tools. We should only need one of these in a consolidated // agent, but we'll need to be careful about behavioural // differences, and interactions with the upgrade-steps // worker. upgraderName: upgrader.Manifold(upgrader.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeStepsGateName: upgradeStepsGateName, UpgradeCheckGateName: upgradeCheckGateName, PreviousAgentVersion: config.PreviousAgentVersion, }), // The upgradesteps worker runs soon after the machine agent // starts and runs any steps required to upgrade to the // running jujud version. Once upgrade steps have run, the // upgradesteps gate is unlocked and the worker exits. upgradeStepsName: upgradesteps.Manifold(upgradesteps.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeStepsGateName: upgradeStepsGateName, OpenStateForUpgrade: config.OpenStateForUpgrade, PreUpgradeSteps: config.PreUpgradeSteps, }), // The uninstaller manifold checks if the machine is dead. If // it is it writes the agent uninstall file and returns // ErrTerminateAgent which causes the agent to remove itself. uninstallerName: uninstallerManifold(uninstallerManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, WriteUninstallFile: config.WriteUninstallFile, }), // The serving-info-setter manifold sets grabs the state // serving info from the API connection and writes it to the // agent config. servingInfoSetterName: ServingInfoSetterManifold(ServingInfoSetterConfig{ AgentName: agentName, APICallerName: apiCallerName, }), // The upgradewaiter manifold aggregates the // upgrade-steps-gate and upgrade-check-gate manifolds into // one boolean output. It makes it easy to create manifolds // which must only run after these upgrade events have // occured. upgradeWaiterName: upgradewaiter.Manifold(upgradewaiter.ManifoldConfig{ UpgradeStepsWaiterName: upgradeStepsGateName, UpgradeCheckWaiterName: upgradeCheckGateName, }), // The apiworkers manifold starts workers which rely on the // machine agent's API connection but have not been converted // to work directly under the dependency engine. It waits for // upgrades to be finished before starting these workers. apiWorkersName: APIWorkersManifold(APIWorkersConfig{ APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, StartAPIWorkers: config.StartAPIWorkers, }), // The reboot manifold manages a worker which will reboot the // machine when requested. It needs an API connection and // waits for upgrades to be complete. rebootName: reboot.Manifold(reboot.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), // The logging config updater is a leaf worker that indirectly // controls the messages sent via the log sender or rsyslog, // according to changes in environment config. We should only need // one of these in a consolidated agent. loggingConfigUpdaterName: logger.Manifold(logger.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), } }
// Manifolds returns a set of co-configured manifolds covering the // various responsibilities of a machine agent. // // Thou Shalt Not Use String Literals In This Function. Or Else. func Manifolds(config ManifoldsConfig) dependency.Manifolds { // connectFilter exists: // 1) to let us retry api connections immeduately on password change, // rather than causing the dependency engine to wait for a while; // 2) to ensure that certain connection failures correctly trigger // complete agent removal. (It's not safe to let any agent other // than the machine mess around with SetCanUninstall). connectFilter := func(err error) error { cause := errors.Cause(err) if cause == apicaller.ErrConnectImpossible { err2 := coreagent.SetCanUninstall(config.Agent) if err2 != nil { return errors.Trace(err2) } return worker.ErrTerminateAgent } else if cause == apicaller.ErrChangedPassword { return dependency.ErrBounce } return err } return dependency.Manifolds{ // The agent manifold references the enclosing agent, and is the // foundation stone on which most other manifolds ultimately depend. agentName: agent.Manifold(config.Agent), // The termination worker returns ErrTerminateAgent if a // termination signal is received by the process it's running // in. It has no inputs and its only output is the error it // returns. It depends on the uninstall file having been // written *by the manual provider* at install time; it would // be Very Wrong Indeed to use SetCanUninstall in conjunction // with this code. terminationName: terminationworker.Manifold(), // The stateconfigwatcher manifold watches the machine agent's // configuration and reports if state serving info is // present. It will bounce itself if state serving info is // added or removed. It is intended as a dependency just for // the state manifold. stateConfigWatcherName: stateconfigwatcher.Manifold(stateconfigwatcher.ManifoldConfig{ AgentName: agentName, AgentConfigChanged: config.AgentConfigChanged, }), // The state manifold creates a *state.State and makes it // available to other manifolds. It pings the mongodb session // regularly and will die if pings fail. stateName: workerstate.Manifold(workerstate.ManifoldConfig{ AgentName: agentName, StateConfigWatcherName: stateConfigWatcherName, OpenState: config.OpenState, }), // The stateworkers manifold starts workers which rely on a // *state.State but which haven't been converted to run // directly under the dependency engine yet. This manifold // will be removed once all such workers have been converted. stateWorkersName: StateWorkersManifold(StateWorkersConfig{ StateName: stateName, StartStateWorkers: config.StartStateWorkers, }), // The api caller is a thin concurrent wrapper around a connection // to some API server. It's used by many other manifolds, which all // select their own desired facades. It will be interesting to see // how this works when we consolidate the agents; might be best to // handle the auth changes server-side..? apiCallerName: apicaller.Manifold(apicaller.ManifoldConfig{ AgentName: agentName, APIOpen: apicaller.APIOpen, NewConnection: apicaller.ScaryConnect, Filter: connectFilter, }), // The upgrade steps gate is used to coordinate workers which // shouldn't do anything until the upgrade-steps worker has // finished running any required upgrade steps. upgradeStepsGateName: gate.ManifoldEx(config.UpgradeStepsLock), // The upgrade check gate is used to coordinate workers which // shouldn't do anything until the upgrader worker has // completed it's first check for a new tools version to // upgrade to. upgradeCheckGateName: gate.ManifoldEx(config.UpgradeCheckLock), // The upgrader is a leaf worker that returns a specific error // type recognised by the machine agent, causing other workers // to be stopped and the agent to be restarted running the new // tools. We should only need one of these in a consolidated // agent, but we'll need to be careful about behavioural // differences, and interactions with the upgrade-steps // worker. upgraderName: upgrader.Manifold(upgrader.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeStepsGateName: upgradeStepsGateName, UpgradeCheckGateName: upgradeCheckGateName, PreviousAgentVersion: config.PreviousAgentVersion, }), // The upgradesteps worker runs soon after the machine agent // starts and runs any steps required to upgrade to the // running jujud version. Once upgrade steps have run, the // upgradesteps gate is unlocked and the worker exits. upgradeStepsName: upgradesteps.Manifold(upgradesteps.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeStepsGateName: upgradeStepsGateName, OpenStateForUpgrade: config.OpenStateForUpgrade, PreUpgradeSteps: config.PreUpgradeSteps, }), // The serving-info-setter manifold sets grabs the state // serving info from the API connection and writes it to the // agent config. servingInfoSetterName: ServingInfoSetterManifold(ServingInfoSetterConfig{ AgentName: agentName, APICallerName: apiCallerName, }), // The upgradewaiter manifold aggregates the // upgrade-steps-gate and upgrade-check-gate manifolds into // one boolean output. It makes it easy to create manifolds // which must only run after these upgrade events have // occured. upgradeWaiterName: upgradewaiter.Manifold(upgradewaiter.ManifoldConfig{ UpgradeStepsWaiterName: upgradeStepsGateName, UpgradeCheckWaiterName: upgradeCheckGateName, }), // The apiworkers manifold starts workers which rely on the // machine agent's API connection but have not been converted // to work directly under the dependency engine. It waits for // upgrades to be finished before starting these workers. apiWorkersName: APIWorkersManifold(APIWorkersConfig{ APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, StartAPIWorkers: config.StartAPIWorkers, }), // The reboot manifold manages a worker which will reboot the // machine when requested. It needs an API connection and // waits for upgrades to be complete. rebootName: reboot.Manifold(reboot.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), // The logging config updater is a leaf worker that indirectly // controls the messages sent via the log sender or according to // changes in environment config. We should only need one of these // in a consolidated agent. loggingConfigUpdaterName: logger.Manifold(logger.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), // The diskmanager worker periodically lists block devices on the // machine it runs on. This worker will be run on all Juju-managed // machines (one per machine agent). diskmanagerName: diskmanager.Manifold(diskmanager.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), // The proxy config updater is a leaf worker that sets http/https/apt/etc // proxy settings. proxyConfigUpdater: proxyupdater.Manifold(proxyupdater.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), // The api address updater is a leaf worker that rewrites agent config // as the state server addresses change. We should only need one of // these in a consolidated agent. apiAddressUpdaterName: apiaddressupdater.Manifold(apiaddressupdater.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), // The machiner Worker will wait for the identified machine to become // Dying and make it Dead; or until the machine becomes Dead by other // means. machinerName: machiner.Manifold(machiner.ManifoldConfig{ PostUpgradeManifoldConfig: util.PostUpgradeManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }, }), // The log sender is a leaf worker that sends log messages to some // API server, when configured so to do. We should only need one of // these in a consolidated agent. logSenderName: logsender.Manifold(logsender.ManifoldConfig{ LogSource: config.LogSource, PostUpgradeManifoldConfig: util.PostUpgradeManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }, }), // The deployer worker is responsible for deploying and recalling unit // agents, according to changes in a set of state units; and for the // final removal of its agents' units from state when they are no // longer needed. deployerName: deployer.Manifold(deployer.ManifoldConfig{ NewDeployContext: config.NewDeployContext, PostUpgradeManifoldConfig: util.PostUpgradeManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }, }), authenticationworkerName: authenticationworker.Manifold(authenticationworker.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), // The storageProvisioner worker manages provisioning // (deprovisioning), and attachment (detachment) of first-class // volumes and filesystems. storageprovisionerName: storageprovisioner.MachineManifold(storageprovisioner.MachineManifoldConfig{ PostUpgradeManifoldConfig: util.PostUpgradeManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName}, Clock: config.Clock, }), resumerName: resumer.Manifold(resumer.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), identityFileWriterName: identityfilewriter.Manifold(identityfilewriter.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), toolsversioncheckerName: toolsversionchecker.Manifold(toolsversionchecker.ManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, UpgradeWaiterName: upgradeWaiterName, }), machineActionName: machineactions.Manifold(machineactions.ManifoldConfig{ AgentApiManifoldConfig: util.AgentApiManifoldConfig{ AgentName: agentName, APICallerName: apiCallerName, }, NewFacade: machineactions.NewFacade, NewWorker: machineactions.NewMachineActionsWorker, }), } }