-
Notifications
You must be signed in to change notification settings - Fork 99
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
26 changed files
with
709 additions
and
97 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
160 changes: 160 additions & 0 deletions
160
...linkedin/glu/orchestration/engine/core/action/execution/ActionExecutionFactoryImpl.groovy
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,160 @@ | ||
| /* | ||
| * Copyright (c) 2011 Yan Pujante | ||
| * | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); you may not | ||
| * use this file except in compliance with the License. You may obtain a copy of | ||
| * the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | ||
| * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the | ||
| * License for the specific language governing permissions and limitations under | ||
| * the License. | ||
| */ | ||
|
|
||
| package org.linkedin.glu.orchestration.engine.core.action.execution | ||
|
|
||
| import org.linkedin.glu.orchestration.engine.action.execution.ActionExecutionFactory | ||
| import org.linkedin.glu.orchestration.engine.action.execution.ActionExecution | ||
| import org.linkedin.glu.orchestration.engine.action.descriptor.ActionDescriptor | ||
| import org.linkedin.glu.orchestration.engine.action.descriptor.NoOpActionDescriptor | ||
| import org.slf4j.Logger | ||
| import org.slf4j.LoggerFactory | ||
| import org.linkedin.glu.orchestration.engine.action.descriptor.ScriptTransitionActionDescriptor | ||
| import org.linkedin.glu.agent.rest.client.AgentFactory | ||
| import org.linkedin.util.annotations.Initializable | ||
| import org.linkedin.glu.orchestration.engine.action.descriptor.AgentActionDescriptor | ||
| import org.linkedin.glu.orchestration.engine.action.descriptor.AgentURIProvider | ||
| import org.linkedin.glu.agent.api.Agent | ||
| import org.linkedin.glu.agent.rest.client.EncryptionKeysProvider | ||
| import org.linkedin.util.clock.Timespan | ||
| import org.linkedin.glu.orchestration.engine.action.descriptor.ScriptLifecycleInstallActionDescriptor | ||
| import org.linkedin.glu.orchestration.engine.action.descriptor.ScriptLifecycleUninstallActionDescriptor | ||
|
|
||
| /** | ||
| * This implementation uses a convention: | ||
| * @author yan@pongasoft.com */ | ||
| public class ActionExecutionFactoryImpl implements ActionExecutionFactory | ||
| { | ||
| public static final String MODULE = ActionExecutionFactoryImpl.class.getName(); | ||
| public static final Logger log = LoggerFactory.getLogger(MODULE); | ||
|
|
||
| public static final String ENCRYPTION_KEYS = 'encryptionKeys' | ||
|
|
||
| @Initializable(required = true) | ||
| AgentFactory factory | ||
|
|
||
| @Initializable(required = true) | ||
| AgentURIProvider agentURIProvider | ||
|
|
||
| @Initializable(required = false) | ||
| EncryptionKeysProvider encryptionKeysProvider | ||
|
|
||
| /** | ||
| * the timeout for the agent operations */ | ||
| @Initializable(required = false) | ||
| Timespan timeout = Timespan.parse('10s') | ||
|
|
||
| /** | ||
| * For NoOpActionDescriptor: do nothing | ||
| */ | ||
| def NoOpActionDescriptor_execution = { NoOpActionDescriptor ad -> | ||
| // nothing to do (noop) | ||
| if(log.isDebugEnabled()) | ||
| log.debug("noop: ${ad.toMetadata()}") | ||
| } | ||
|
|
||
| /** | ||
| * For ScriptTransitionActionDescriptor: execute the action (transition) on the agent and wait | ||
| * for the action to complete | ||
| */ | ||
| def ScriptTransitionActionDescriptor_execution = { ScriptTransitionActionDescriptor ad -> | ||
| withAgent(ad) { Agent agent -> | ||
|
|
||
| String mountPoint = ad.mountPoint | ||
|
|
||
| // 1. we clear any error which could be left over | ||
| agent.clearError(mountPoint: mountPoint) | ||
|
|
||
| // 2. we execute the transition action | ||
| Map actionArgs = computeActionArgs(ad) | ||
| agent.executeAction(mountPoint: mountPoint, action: ad.action, actionArgs: actionArgs) | ||
|
|
||
| // 3. we wait for the action to be completed | ||
| def success = false | ||
| while(!success) | ||
| { | ||
| success = agent.waitForState(mountPoint: mountPoint, state: ad.endState, timeout: timeout) | ||
| if(!success) | ||
| { | ||
| if(Thread.currentThread().isInterrupted()) | ||
| { | ||
| agent.interruptAction(mountPoint: mountPoint, action: ad.action) | ||
| throw new InterruptedException() | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * For ScriptLifecycleInstallActionDescriptor: installs the script | ||
| */ | ||
| def ScriptLifecycleInstallActionDescriptor_execution = { ScriptLifecycleInstallActionDescriptor ad -> | ||
| withAgent(ad) { Agent agent -> | ||
| agent.installScript(mountPoint: ad.mountPoint, | ||
| scriptLocation: ad.script, | ||
| parent: ad.parent, | ||
| initParameters: ad.initParameters) | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * For ScriptLifecycleUninstallActionDescriptor: uninstalls the script | ||
| */ | ||
| def ScriptLifecycleUninstallActionDescriptor_execution = { ScriptLifecycleUninstallActionDescriptor ad -> | ||
| withAgent(ad) { Agent agent -> | ||
| agent.clearError(mountPoint: ad.mountPoint) | ||
|
|
||
| agent.uninstallScript(mountPoint: ad.mountPoint) | ||
| } | ||
| } | ||
|
|
||
| /** | ||
| * Compute the action args (automatically adds encryption keys if any present) | ||
| */ | ||
| Map computeActionArgs(ScriptTransitionActionDescriptor ad) | ||
| { | ||
| def Map args = [ : ] | ||
|
|
||
| Map encryptionKeys = encryptionKeysProvider?.getEncryptionKeys() | ||
|
|
||
| if (encryptionKeys) { | ||
| args.put(ENCRYPTION_KEYS, encryptionKeys) | ||
| } | ||
|
|
||
| if(ad.actionArgs) | ||
| args.putAll(ad.actionArgs) | ||
|
|
||
| return args | ||
| } | ||
|
|
||
| /** | ||
| * Compute the agent URI and call the closure with the {@link Agent}. | ||
| */ | ||
| private def withAgent(AgentActionDescriptor ad, Closure closure) | ||
| { | ||
| factory.withRemoteAgent(agentURIProvider.getAgentURI(ad.fabric, ad.agent), closure) | ||
| } | ||
|
|
||
| @Override | ||
| <V> ActionExecution<V> createAction(ActionDescriptor actionDescriptor) | ||
| { | ||
| Closure actionClosure = this."${actionDescriptor.class.name}_execution" | ||
|
|
||
| return new ClosureActionExecution<V>(actionDescriptor: actionDescriptor, | ||
| actionClosure: actionClosure) | ||
| } | ||
| } |
34 changes: 34 additions & 0 deletions
34
...org/linkedin/glu/orchestration/engine/core/action/execution/ClosureActionExecution.groovy
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,34 @@ | ||
| /* | ||
| * Copyright (c) 2011 Yan Pujante | ||
| * | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); you may not | ||
| * use this file except in compliance with the License. You may obtain a copy of | ||
| * the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | ||
| * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the | ||
| * License for the specific language governing permissions and limitations under | ||
| * the License. | ||
| */ | ||
|
|
||
| package org.linkedin.glu.orchestration.engine.core.action.execution | ||
|
|
||
| import org.linkedin.glu.orchestration.engine.action.execution.ActionExecution | ||
| import org.linkedin.glu.orchestration.engine.action.descriptor.ActionDescriptor | ||
|
|
||
| /** | ||
| * @author yan@pongasoft.com */ | ||
| public class ClosureActionExecution<V> implements ActionExecution<V> | ||
| { | ||
| ActionDescriptor actionDescriptor | ||
| Closure actionClosure | ||
|
|
||
| @Override | ||
| V call() | ||
| { | ||
| return actionClosure(actionDescriptor) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
33 changes: 33 additions & 0 deletions
33
...c/main/java/org/linkedin/glu/orchestration/engine/action/descriptor/AgentURIProvider.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,33 @@ | ||
| /* | ||
| * Copyright (c) 2011 Yan Pujante | ||
| * | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); you may not | ||
| * use this file except in compliance with the License. You may obtain a copy of | ||
| * the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | ||
| * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the | ||
| * License for the specific language governing permissions and limitations under | ||
| * the License. | ||
| */ | ||
|
|
||
| package org.linkedin.glu.orchestration.engine.action.descriptor; | ||
|
|
||
| import org.linkedin.glu.orchestration.engine.agents.NoSuchAgentException; | ||
|
|
||
| import java.net.URI; | ||
|
|
||
| /** | ||
| * @author yan@pongasoft.com | ||
| */ | ||
| public interface AgentURIProvider | ||
| { | ||
| /** | ||
| * @return the agent uri given the agent name | ||
| * @throws NoSuchAgentException if agent cannot be found | ||
| */ | ||
| URI getAgentURI(String fabric, String agent) throws NoSuchAgentException; | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.