package com.vincent.rsf.server.common.datasource;
|
|
import java.util.ArrayDeque;
|
import java.util.Deque;
|
|
/**
|
* 数据源上下文
|
*/
|
public final class DataSourceContextHolder {
|
|
private static final ThreadLocal<Deque<String>> CONTEXT = ThreadLocal.withInitial(ArrayDeque::new);
|
|
private DataSourceContextHolder() {
|
}
|
|
public static void push(String dataSource) {
|
CONTEXT.get().push(dataSource);
|
}
|
|
public static String peek() {
|
Deque<String> deque = CONTEXT.get();
|
return deque.isEmpty() ? null : deque.peek();
|
}
|
|
public static void poll() {
|
Deque<String> deque = CONTEXT.get();
|
if (!deque.isEmpty()) {
|
deque.pop();
|
}
|
if (deque.isEmpty()) {
|
CONTEXT.remove();
|
}
|
}
|
}
|
