/*
* Copyright 2012 Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://aws.amazon.com/apache2.0
*
* or in the "license" file accompanying this file. This file 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 com.amazonaws.services.simpleworkflow.flow.worker;
import com.amazonaws.services.simpleworkflow.flow.ActivityExecutionContext;
import com.amazonaws.services.simpleworkflow.flow.ActivityExecutionContextProviderImpl;
/**
* Thread local store of the context object passed to an activity
* implementation. Avoid using this class directly. Use
* {@link ActivityExecutionContextProviderImpl} instead.
*
* @author fateev
*/
public class CurrentActivityExecutionContext {
private final static ThreadLocal<ActivityExecutionContext> CURRENT = new ThreadLocal<ActivityExecutionContext>();
/**
* This is used by activity implementation to get access to the current
* ActivityExecutionContext
*/
public static ActivityExecutionContext get() {
ActivityExecutionContext result = CURRENT.get();
if (result == null) {
throw new IllegalStateException("ActivityExecutionContext can be used only inside of acitivty implementation methods");
}
return result;
};
public static boolean isSet() {
return CURRENT.get() != null;
}
public static void set(ActivityExecutionContext context) {
if (context == null) {
throw new IllegalArgumentException("null context");
}
CURRENT.set(context);
}
public static void unset() {
CURRENT.set(null);
}
private CurrentActivityExecutionContext() {
}
}