org.nutz.boot.NbApp Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of nutzboot-core Show documentation
Show all versions of nutzboot-core Show documentation
NutzBoot, micoservice base on Nutz
package org.nutz.boot;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import org.nutz.boot.aware.AppContextAware;
import org.nutz.boot.aware.ClassLoaderAware;
import org.nutz.boot.aware.EnvHolderAware;
import org.nutz.boot.aware.IocAware;
import org.nutz.boot.aware.ResourceLoaderAware;
import org.nutz.boot.banner.SimpleBannerPrinter;
import org.nutz.boot.config.ConfigureLoader;
import org.nutz.boot.config.impl.PropertiesConfigureLoader;
import org.nutz.boot.env.SystemPropertiesEnvHolder;
import org.nutz.boot.ioc.IocLoaderProvider;
import org.nutz.boot.metrics.impl.MemoryCounterService;
import org.nutz.boot.resource.ResourceLoader;
import org.nutz.boot.resource.impl.SimpleResourceLoader;
import org.nutz.boot.tools.NbAppEventListener;
import org.nutz.boot.tools.NbAppEventListener.EventType;
import org.nutz.boot.tools.PropDocReader;
import org.nutz.ioc.IocLoader;
import org.nutz.ioc.impl.NutIoc;
import org.nutz.ioc.loader.annotation.AnnotationIocLoader;
import org.nutz.ioc.loader.annotation.IocBean;
import org.nutz.ioc.loader.combo.ComboIocLoader;
import org.nutz.lang.Lang;
import org.nutz.lang.Mirror;
import org.nutz.lang.Stopwatch;
import org.nutz.lang.Streams;
import org.nutz.lang.Strings;
import org.nutz.lang.util.LifeCycle;
import org.nutz.log.Log;
import org.nutz.log.LogAdapter;
import org.nutz.log.Logs;
import org.nutz.mvc.Mvcs;
import org.nutz.mvc.annotation.IocBy;
/**
* NutzBoot的主体
*
* @author wendal
*
*/
public class NbApp extends Thread {
/**
* 日志属性要等日志适配器准备好了才能加载,这里不可以使用Logs.get();
*/
protected static Log log;
/**
* 命令行参数
*/
protected String[] args;
/**
* 是否允许命令行下的 -Dxxx.xxx.xxx=转为配置参数
*/
protected boolean allowCommandLineProperties = true;
/**
* 是否打印配置文档
*/
protected boolean printProcDoc;
/**
* 主上下文
*/
protected AppContext ctx;
/**
* Starter共享的IocLoader
*/
protected AnnotationIocLoader starterIocLoader;
/**
* Starter类列表
*/
protected List> starterClasses;
protected boolean prepared;
protected Object lock = new Object();
protected List listeners = new LinkedList<>();
protected boolean started;
/**
* 创建一个NbApp,把调用本构造方法的类作为mainClass
*/
public NbApp() {
ctx = AppContext.getDefault();
StackTraceElement[] ts = Thread.currentThread().getStackTrace();
if (ts.length > 2) {
setMainClass(Lang.loadClassQuite(ts[2].getClassName()));
}
}
/**
* 基于mainClass创建一个NbApp
*
* @param mainClass
* 主启动类
*/
public NbApp(Class mainClass) {
ctx = AppContext.getDefault();
setMainClass(mainClass);
}
public NbApp(AppContext ctx) {
this.ctx = ctx;
StackTraceElement[] ts = Thread.currentThread().getStackTrace();
if (ts.length > 2) {
setMainClass(Lang.loadClassQuite(ts[2].getClassName()));
}
}
public NbApp(AppContext ctx, Class mainClass) {
this.ctx = ctx;
setMainClass(mainClass);
}
/**
* 设置命令行参数,可选
*/
public NbApp setArgs(String[] args) {
this.args = args;
return this;
}
/**
* 设置主启动类
*/
public NbApp setMainClass(Class mainClass) {
this.ctx.setMainClass(mainClass);
return this;
}
/**
* 是否允许从命令行读取配置信息, 例如 java -Dnutz.ioc.async.poolSize=128 xxx.jar
*
* @param flag
* 默认允许
*/
public NbApp setAllowCommandLineProperties(boolean flag) {
this.allowCommandLineProperties = flag;
return this;
}
/**
* 是否打印配置文档
*/
public NbApp setPrintProcDoc(boolean printProcDoc) {
this.printProcDoc = printProcDoc;
return this;
}
/**
* 获取全局上下文
*
* @return
*/
public AppContext getAppContext() {
return ctx;
}
public void run() {
try {
if (execute()) {
synchronized (lock) {
lock.wait();
}
}
// 收尾
_shutdown();
}
catch (Throwable e) {
Logs.get().error("something happen", e);
}
}
public boolean execute() {
Stopwatch sw = Stopwatch.begin();
try {
// 各种预备操作
listeners.forEach((listener)->listener.whenPrepare(this, EventType.before));
this.prepare();
listeners.forEach((listener)->listener.whenPrepare(this, EventType.after));
// 依次启动
listeners.forEach((listener)->listener.whenInitAppContext(this, EventType.before));
ctx.init();
listeners.forEach((listener)->listener.whenInitAppContext(this, EventType.after));
listeners.forEach((listener)->listener.whenStartServers(this, EventType.before));
ctx.startServers();
listeners.forEach((listener)->listener.whenStartServers(this, EventType.after));
if (ctx.getMainClass().getAnnotation(IocBean.class) != null)
ctx.getIoc().get(ctx.getMainClass());
listeners.forEach((listener)->listener.afterAppStated(this));
sw.stop();
log.infof("%s started : %sms", ctx.getConf().get("nutz.application.name", "NB"), sw.du());
started = true;
return true;
}
catch (Throwable e) {
log.error("something happen!!", e);
return false;
}
}
protected void _shutdown() {
try {
ctx.stopServers();
ctx.depose();
}
catch (Exception e) {
throw Lang.wrapThrow(e);
}
}
public void shutdown() {
log.info("ok, shutting down ...");
if (lock == null) {
_shutdown();
}
else {
synchronized (lock) {
lock.notify();
}
}
}
/**
* 执行预备操作
*/
public void prepare() throws Exception {
if (prepared)
return;
// 初始化上下文
listeners.forEach((listener)->listener.whenPrepareBasic(this, EventType.before));
this.prepareBasic();
listeners.forEach((listener)->listener.whenPrepareBasic(this, EventType.after));
// 打印Banner,暂时不可配置具体的类
new SimpleBannerPrinter().printBanner(ctx);
// 配置信息要准备好
listeners.forEach((listener)->listener.whenPrepareConfigureLoader(this, EventType.before));
this.prepareConfigureLoader();
listeners.forEach((listener)->listener.whenPrepareConfigureLoader(this, EventType.after));
// 创建IocLoader体系
listeners.forEach((listener)->listener.whenPrepareIocLoader(this, EventType.before));
prepareIocLoader();
listeners.forEach((listener)->listener.whenPrepareIocLoader(this, EventType.after));
// 加载各种starter
listeners.forEach((listener)->listener.whenPrepareStarterClassList(this, EventType.before));
prepareStarterClassList();
listeners.forEach((listener)->listener.whenPrepareStarterClassList(this, EventType.after));
// 打印配置文档
if (printProcDoc) {
PropDocReader docReader = new PropDocReader();
docReader.load(starterClasses);
Logs.get().info("Configure Manual:\r\n" + docReader.toMarkdown());
}
// 创建Ioc容器
listeners.forEach((listener)->listener.whenPrepareIoc(this, EventType.before));
prepareIoc();
listeners.forEach((listener)->listener.whenPrepareIoc(this, EventType.after));
// 生成Starter实例
listeners.forEach((listener)->listener.whenPrepareStarterInstance(this, EventType.before));
prepareStarterInstance();
listeners.forEach((listener)->listener.whenPrepareStarterInstance(this, EventType.after));
// 从Ioc容器检索Listener
listeners.addAll(ctx.getBeans(NbAppEventListener.class));
prepared = true;
}
public void prepareBasic() throws Exception {
// 检查ClassLoader的情况
if (ctx.getClassLoader() == null)
ctx.setClassLoader(NbApp.class.getClassLoader());
if (ctx.getEnvHolder() == null) {
ctx.setEnvHolder(new SystemPropertiesEnvHolder());
}
// 看看日志应该用哪个
String logAdapter = ctx.getEnvHolder().get("nutz.boot.base.LogAdapter");
if (!Strings.isBlank(logAdapter)) {
Logs.setAdapter((LogAdapter) ctx.getClassLoader().loadClass(logAdapter).newInstance());
}
log = Logs.get();
// 资源加载器
if (ctx.getResourceLoader() == null) {
ResourceLoader resourceLoader = new SimpleResourceLoader();
aware(resourceLoader);
ctx.setResourceLoader(resourceLoader);
}
}
public void prepareConfigureLoader() throws Exception {
if (ctx.getConfigureLoader() == null) {
ConfigureLoader configureLoader = null;
InputStream ins = ctx.getResourceLoader().get("META-INF/nutz/org.nutz.boot.config.ConfigureLoader");
if (ins != null) {
String cnfLoader = new String(Streams.readBytes(ins)).trim();
if (!Strings.isBlank(cnfLoader)) {
log.debugf("using %s as ConfigureLoader", cnfLoader);
configureLoader = (ConfigureLoader) ctx.getClassLoader().loadClass(cnfLoader).newInstance();
}
}
if (configureLoader == null) {
configureLoader = new PropertiesConfigureLoader();
}
configureLoader.setCommandLineProperties(allowCommandLineProperties, args);
aware(configureLoader);
ctx.setConfigureLoader(configureLoader);
if (configureLoader instanceof LifeCycle)
((LifeCycle) configureLoader).init();
}
}
public void prepareIocLoader() throws Exception {
if (ctx.getComboIocLoader() == null) {
int asyncPoolSize = ctx.getConfigureLoader().get().getInt("nutz.ioc.async.poolSize", 64);
List args = new ArrayList<>();
args.add("*js");
args.add("ioc/");
args.add("*tx");
args.add("*async");
args.add("" + asyncPoolSize);
args.add("*anno");
args.add(ctx.getPackage());
IocBy iocBy = ctx.getMainClass().getAnnotation(IocBy.class);
if (iocBy != null) {
String[] tmp = iocBy.args();
ArrayList _args = new ArrayList<>();
for (int i = 0; i < tmp.length; i++) {
if (tmp[i].startsWith("*")) {
if (!_args.isEmpty()) {
switch (_args.get(0)) {
case "*tx":
case "*async":
case "*anno":
case "*js":
break;
default:
args.addAll(_args);
}
_args.clear();
}
}
_args.add(tmp[i]);
}
if (_args.size() > 0) {
switch (_args.get(0)) {
case "*tx":
case "*async":
case "*anno":
case "*js":
break;
default:
args.addAll(_args);
}
}
}
ctx.setComboIocLoader(new ComboIocLoader(args.toArray(new String[args.size()])));
}
// 用于加载Starter的IocLoader
starterIocLoader = new AnnotationIocLoader(NbApp.class.getPackage().getName() + ".starter");
ctx.getComboIocLoader().addLoader(starterIocLoader);
}
public void prepareStarterClassList() throws Exception {
starterClasses = new ArrayList<>();
HashSet classNames = new HashSet<>();
Enumeration _en = ctx.getClassLoader().getResources("META-INF/nutz/org.nutz.boot.starter.NbStarter");
while (_en.hasMoreElements()) {
URL url = _en.nextElement();
log.debug("Found " + url);
try (InputStream ins = url.openStream()) {
InputStreamReader reader = new InputStreamReader(ins);
String tmp = Streams.readAndClose(reader);
if (!Strings.isBlank(tmp)) {
for (String _tmp : Strings.splitIgnoreBlank(tmp, "[\n]")) {
String className = _tmp.trim();
if (!classNames.add(className))
continue;
Class klass = ctx.getClassLoader().loadClass(className);
if (!klass.getPackage().getName().startsWith(NbApp.class.getPackage().getName()) && klass.getAnnotation(IocBean.class) != null) {
starterIocLoader.addClass(klass);
}
starterClasses.add(klass);
}
}
}
}
}
public void prepareIoc() {
if (ctx.getIoc() == null) {
ctx.setIoc(new NutIoc(ctx.getComboIocLoader()));
}
// 把核心对象放进ioc容器
if (!ctx.ioc.has("appContext")) {
ctx.ioc.addBean("appContext", ctx);
ctx.ioc.addBean("conf", ctx.getConf());
ctx.ioc.addBean("nbApp", this);
// 添加更多扩展bean
ctx.ioc.addBean("counterService", new MemoryCounterService());
}
Mvcs.ctx().iocs.put("nutz", ctx.getIoc());
}
public void prepareStarterInstance() {
for (Class klass : starterClasses) {
Object obj;
if (klass.getAnnotation(IocBean.class) == null) {
obj = Mirror.me(klass).born();
} else {
continue;
}
aware(obj);
if (obj instanceof IocLoaderProvider) {
IocLoader loader = ((IocLoaderProvider) obj).getIocLoader();
ctx.getComboIocLoader().addLoader(loader);
}
ctx.addStarter(obj);
}
for (Class klass : starterClasses) {
Object obj;
if (klass.getAnnotation(IocBean.class) == null) {
continue;
} else {
obj = ctx.getIoc().get(klass);
}
aware(obj);
if (obj instanceof IocLoaderProvider) {
IocLoader loader = ((IocLoaderProvider) obj).getIocLoader();
ctx.getComboIocLoader().addLoader(loader);
}
ctx.addStarter(obj);
}
}
protected void aware(Object obj) {
// 需要注入AppContext
if (obj instanceof AppContextAware) {
((AppContextAware) obj).setAppContext(ctx);
}
// 需要注入ClassLoader
if (obj instanceof ClassLoaderAware) {
((ClassLoaderAware) obj).setClassLoader(ctx.getClassLoader());
}
// 需要注入EnvHolder
if (obj instanceof EnvHolderAware) {
((EnvHolderAware) obj).setEnvHolder(ctx.getEnvHolder());
}
// 需要注入ResourceLoader
if (obj instanceof ResourceLoaderAware) {
((ResourceLoaderAware) obj).setResourceLoader(ctx.getResourceLoader());
}
// 需要注入Ioc
if (obj instanceof IocAware) {
((IocAware) obj).setIoc(ctx.getIoc());
}
}
public void setListener(NbAppEventListener listener) {
this.listeners.clear();
addListener(listener);
}
public void addListener(NbAppEventListener listener) {
this.listeners.add(listener);
}
public boolean isStarted() {
return started;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy