package act.job;
/*-
* #%L
* ACT Framework
* %%
* Copyright (C) 2014 - 2017 ActFramework
* %%
* 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.
* #L%
*/
import act.Act;
import act.app.ActionContext;
import act.app.App;
import act.job.event.JobContextDestroyed;
import act.job.event.JobContextInitialized;
import act.util.ActContext;
import org.osgl.http.H;
import org.osgl.util.E;
import java.util.HashMap;
import java.util.Locale;
import java.util.Map;
import java.util.Set;
/**
* Communicate context data across threads
*/
public class JobContext extends ActContext.Base {
private static ThreadLocal current_ = new ThreadLocal();
private JobContext parent;
private JobContext(JobContext parent) {
super(App.instance());
current_.set(this);
this.parent = parent;
if (null != parent) {
bag_.putAll(parent.bag_);
} else {
ActContext actContext = ActContext.Base.currentContext();
if (null != actContext) {
bag_.put("locale", actContext.locale());
if (actContext instanceof ActionContext) {
H.Session session = ((ActionContext) actContext).session();
if (null != session) {
bag_.put("session", session);
}
}
}
}
}
private static Map m() {
return current_.get().bag_;
}
/**
* Whether JobContext of current thread initialized
* @return `true` if current job context is not null
*/
public static boolean initialized() {
return null != current_.get();
}
public static JobContext current() {
return current_.get();
}
@Override
public JobContext accept(H.Format fmt) {
return this;
}
@Override
public H.Format accept() {
throw E.unsupport();
}
@Override
public String methodPath() {
throw E.unsupport();
}
@Override
public Set paramKeys() {
throw E.unsupport();
}
@Override
public String paramVal(String s) {
throw E.unsupport();
}
@Override
public String[] paramVals(String s) {
throw E.unsupport();
}
/**
* Init JobContext of current thread
*/
static void init() {
JobContext parent = current_.get();
JobContext ctx = new JobContext(parent);
if (null == parent) {
Act.eventBus().trigger(new JobContextInitialized(ctx));
}
}
/**
* Clear JobContext of current thread
*/
static void clear() {
JobContext ctx = current_.get();
if (null != ctx) {
ctx.bag_.clear();
JobContext parent = ctx.parent;
if (null != parent) {
current_.set(parent);
} else {
current_.remove();
Act.eventBus().trigger(new JobContextDestroyed(ctx));
}
}
}
/**
* Get value by key from the JobContext of current thread
* @param key the key
* @param the val type
* @return attribute value associated with `key` in this job context
*/
public static T get(String key) {
return (T) m().get(key);
}
/**
* Generic version of getting value by key from the JobContext of current thread
* @param key the key
* @param clz the val class
* @param the val type
* @return the value
*/
public static T get(String key, Class clz) {
return (T)m().get(key);
}
/**
* Set value by key to the JobContext of current thread
* @param key the key
* @param val the value
*/
public static void put(String key, Object val) {
m().put(key, val);
}
/**
* Remove value by key from the JobContext of current thread
* @param key the key
*/
public static void remove(String key) {
m().remove(key);
}
/**
* Make a copy of JobContext of current thread
* @return the copy of current job context or an empty job context
*/
static JobContext copy() {
JobContext ctxt = new JobContext(null);
JobContext current = current_.get();
if (null != current) {
ctxt.bag_.putAll(current.bag_);
}
return ctxt;
}
/**
* Initialize current thread's JobContext using specified copy
* @param origin the original job context
*/
static void init(JobContext origin) {
current_.set(origin);
if (origin.bag_.isEmpty()) {
return;
}
ActContext actContext = ActContext.Base.currentContext();
if (null != actContext) {
Locale locale = (Locale) origin.bag_.get("locale");
if (null != locale) {
actContext.locale(locale);
}
H.Session session = (H.Session) origin.bag_.get("session");
if (null != session) {
actContext.attribute("__session", session);
}
}
}
private Map bag_ = new HashMap<>();
}