330 lines
12 KiB
Java
Executable File
330 lines
12 KiB
Java
Executable File
/**
|
||
* Copyright 2018-2028 WindChat Group
|
||
*
|
||
* 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 com.windchat.im.boot.main;
|
||
|
||
import java.io.File;
|
||
import java.io.IOException;
|
||
import java.io.PrintWriter;
|
||
import java.util.Base64;
|
||
import java.util.Map;
|
||
|
||
import com.windchat.im.boot.spring.OpenzalySpringBoot;
|
||
import com.windchat.im.boot.utils.BootLog;
|
||
import com.windchat.im.boot.utils.Helper;
|
||
import org.apache.commons.lang3.StringUtils;
|
||
import org.apache.log4j.Level;
|
||
|
||
import com.windchat.common.command.Command;
|
||
import com.windchat.common.command.CommandResponse;
|
||
import com.windchat.common.constant.HttpUriAction;
|
||
import com.windchat.common.constant.RequestAction;
|
||
import com.windchat.common.executor.AbstracteExecutor;
|
||
import com.windchat.common.logs.AkxLog4jManager;
|
||
import com.windchat.common.utils.StringHelper;
|
||
import com.akaxin.proto.core.FileProto.FileType;
|
||
import com.windchat.im.boot.config.WindProject;
|
||
import com.windchat.im.boot.config.ConfigHelper;
|
||
import com.windchat.im.boot.config.ConfigKey;
|
||
import com.windchat.im.boot.config.ConfigListener;
|
||
import com.windchat.im.boot.config.SiteDefaultIcon;
|
||
import com.windchat.im.business.utils.FilePathUtils;
|
||
import com.windchat.im.business.utils.FileServerUtils;
|
||
import com.windchat.im.connector.exception.HttpServerException;
|
||
import com.windchat.im.connector.exception.TcpServerException;
|
||
import com.windchat.im.connector.handler.ApiRequestHandler;
|
||
import com.windchat.im.connector.handler.HttpRequestHandler;
|
||
import com.windchat.im.connector.handler.ImMessageHandler;
|
||
import com.windchat.im.connector.handler.ImSiteAuthHandler;
|
||
import com.windchat.im.connector.handler.WSRequestHandler;
|
||
import com.windchat.im.connector.http.HttpServer;
|
||
import com.windchat.im.connector.netty.NettyServer;
|
||
import com.windchat.im.connector.ws.WsServer;
|
||
import com.windchat.im.storage.DataSourceManager;
|
||
import com.windchat.im.storage.dao.config.DBConfig;
|
||
import com.windchat.im.storage.dao.sqlite.manager.PluginArgs;
|
||
import com.windchat.im.storage.exception.InitDatabaseException;
|
||
import com.windchat.im.storage.exception.NeedInitMysqlException;
|
||
import com.windchat.im.storage.exception.UpgradeDatabaseException;
|
||
|
||
/**
|
||
* <pre>
|
||
* Openzaly是Akaxin聊天软件的服务端开源项目，当你第一次从github上下载源码至本地后，可以通过
|
||
* Bootstrap中的main方法启动Openzaly服务器代码。Openzaly-server配合Akaxin客户端协同使用，
|
||
* Akaxin客户端可以在苹果的Appstore以及<a href='www.akaxin.com'>Akaxin官方下载</a>
|
||
*
|
||
* Openzaly-boot是Openzaly项目中的启动模块，主要负责项目的初始化，事件监听，日志等级变更，帮
|
||
* 助文档，标准化输出，服务启动：
|
||
*
|
||
* 1.帮助文档
|
||
* Openzaly启动支持自定义参数，这些参数通过用户启动命令中增加[-h|-help]获取，具体执行如下：
|
||
* java -jar openzaly-server.jar -h
|
||
* java -jar openzaly-server.jar -help
|
||
*
|
||
* 2.初始化工作
|
||
* 项目启动前期，需要初始化服务端数据，当前需要初始化的数据包括：
|
||
* a.初始化数据库，自动创建SQLite中需要的table
|
||
* b.站点服务的默认配置或者用户自定义的配置信息
|
||
* c.默认后台管理与用户广场的ICON设置
|
||
*
|
||
* 3.日志等级变更
|
||
* Openzaly项目中使用的日志框架为Log4j+SLF4J，默认的日志等级为INFO级别，在后台管理中，支持
|
||
* 用户通过配置信息修改，来实时变更项目中的日志级别，从而达到在不停止服务情况下，修改日志级别。
|
||
*
|
||
* 4.服务启动
|
||
* Openzaly项目启动的主要部分，包含三个服务的启动分别如下：
|
||
* a.提供扩展使用的Netty-Http服务
|
||
* 使用Netty框架启动Http服务，当开发者开发站点的扩展功能，可以调用此Http接口实现与站点之
|
||
* 间的交互。
|
||
*
|
||
* b.提供客户端访问Netty-Tcp服务
|
||
* Akaxin的客户端【Andorid与IOS】通过tcp连接保持与站点之间的长连接，实现用户与Openzaly
|
||
* 之间的IM功能以及部分API访问请求。
|
||
*
|
||
* c.提供WEBIM使用的WebSocket服务
|
||
* 暂时此功能未上线
|
||
*
|
||
* 5.标准化输出
|
||
* 在标准输出界面输出Openzaly的启动情况【log日志信息中支持更详细的启动记录】
|
||
*
|
||
*
|
||
* Begin from here,start the Openzaly server for clients
|
||
* ......
|
||
*
|
||
* </pre>
|
||
*
|
||
* @author Sam{@link an.guoyue254@gmail.com}
|
||
* @since 2018.01.01 11:23:42
|
||
*/
|
||
public class Bootstrap {
|
||
|
||
public static void main(String[] args) {
|
||
|
||
// set root dir
|
||
try {
|
||
setBaseDir();
|
||
} catch (IOException ioe) {
|
||
BootLog.error("openzaly set base dir error");
|
||
System.exit(-100);
|
||
}
|
||
|
||
// 增加 -h|-help 启动参数 输出帮助文档
|
||
// use java -jar -h|-help ,get more help message
|
||
if (Helper.startHelper(args)) {
|
||
return;
|
||
}
|
||
|
||
PrintWriter pwriter = new PrintWriter(System.out);
|
||
Helper.showAkaxinBanner(pwriter);
|
||
Helper.buildEnvToSystemOut(pwriter);
|
||
|
||
String nettyTcpHost = "0.0.0.0";
|
||
int nettyTcpPort = 2021;
|
||
|
||
String pluginAPiAddress = "0.0.0.0";
|
||
int pluginAPiPort = 8280;
|
||
|
||
try {
|
||
setDefaultSystemLogLevel();
|
||
|
||
// client tcp address from openzaly.properties
|
||
nettyTcpHost = ConfigHelper.getStringConfig(ConfigKey.SITE_ADDRESS);
|
||
nettyTcpPort = ConfigHelper.getIntConfig(ConfigKey.SITE_PORT);
|
||
|
||
// plugin http address from openzaly.properties
|
||
pluginAPiAddress = ConfigHelper.getStringConfig(ConfigKey.PLUGIN_API_ADDRESS);
|
||
pluginAPiPort = ConfigHelper.getIntConfig(ConfigKey.PLUGIN_API_PORT);
|
||
|
||
// add site config to database
|
||
initDataSource();
|
||
// use thread to update site-config cached in memory
|
||
addConfigListener();
|
||
|
||
// start server
|
||
startNettyHttpServer(pluginAPiAddress, pluginAPiPort);// 0.0.0.0:8280
|
||
startNettyTcpServer(nettyTcpHost, nettyTcpPort);// 0.0.0.0:2021
|
||
|
||
// disable websocket server
|
||
// startWebSocketServer("0.0.0.0", 9090);// 0.0.0.0:9090
|
||
|
||
// start spring boot for openzaly-admin
|
||
initSpringBoot(args);
|
||
|
||
Helper.startSuccess(pwriter);
|
||
BootLog.info("start openzaly-server successfully");
|
||
} catch (Exception e) {
|
||
Helper.startFail(pwriter);
|
||
BootLog.error("start Openzaly-server error", e);
|
||
BootLog.error("Openzaly-server exit...");
|
||
System.exit(-1);// system exit
|
||
} catch (TcpServerException e) {
|
||
String errMessage = StringHelper.format("openzaly tcp-server {}:{} {}", nettyTcpHost, nettyTcpPort,
|
||
e.getCause().getMessage());
|
||
Helper.startFailWithError(pwriter, errMessage);
|
||
BootLog.error("start Openzaly with tcp server error", e);
|
||
BootLog.error("Openzaly-server exit...");
|
||
System.exit(-2);// system exit
|
||
} catch (HttpServerException e) {
|
||
String errMessage = StringHelper.format("openzaly http-server {}:{} {}", pluginAPiAddress, pluginAPiPort,
|
||
e.getCause().getMessage());
|
||
Helper.startFailWithError(pwriter, errMessage);
|
||
BootLog.error("start Openzaly with http server error", e);
|
||
BootLog.error("Openzaly-server exit...");
|
||
System.exit(-3);// system exit
|
||
} catch (InitDatabaseException e) {
|
||
String errMessage = StringHelper.format("openzaly init database error {}", e.getCause().getMessage());
|
||
Helper.startFailWithError(pwriter, errMessage);
|
||
BootLog.error("start Openzaly with init database error", e);
|
||
BootLog.error("Openzaly-server exit...");
|
||
System.exit(-4);// system exit
|
||
} catch (UpgradeDatabaseException e) {
|
||
Helper.printUpgradeWarn(pwriter);
|
||
BootLog.error("Openzaly-server current is an old version ,we need to upgrade.", e);
|
||
System.exit(-5);// system exit
|
||
} catch (NeedInitMysqlException e) {
|
||
Helper.printInitMysqlWarn(pwriter);
|
||
BootLog.error("Openzaly-server need to init mysql.", e);
|
||
System.exit(-6);// system exit
|
||
} finally {
|
||
if (pwriter != null) {
|
||
pwriter.close();
|
||
}
|
||
}
|
||
}
|
||
|
||
private static void setBaseDir() throws IOException {
|
||
String baseDir = ConfigHelper.getStringConfig(ConfigKey.SITE_BASE_DIR);
|
||
|
||
if (StringUtils.isNotBlank(baseDir)) {
|
||
File file = new File(baseDir);
|
||
if (!file.isDirectory()) {
|
||
file.mkdirs();
|
||
}
|
||
System.setProperty("user.dir", file.getCanonicalPath());
|
||
}
|
||
}
|
||
|
||
private static void setDefaultSystemLogLevel() {
|
||
// 更新日志级别
|
||
AkxLog4jManager.setLogLevel(Level.INFO);
|
||
BootLog.info("{} set system log level={}", WindProject.PLN, Level.INFO);
|
||
}
|
||
|
||
/**
|
||
* 初始化数据源
|
||
*
|
||
* @throws InitDatabaseException
|
||
* @throws UpgradeDatabaseException
|
||
* @throws NeedInitMysqlException
|
||
*/
|
||
private static void initDataSource()
|
||
throws InitDatabaseException, UpgradeDatabaseException, NeedInitMysqlException {
|
||
String adminHost = ConfigHelper.getStringConfig(ConfigKey.SITE_ADMIN_ADDRESS);
|
||
int adminPort = ConfigHelper.getIntConfig(ConfigKey.SITE_ADMIN_PORT);
|
||
|
||
String dbDir = System.getProperty("user.dir");
|
||
String adminUic = ConfigHelper.getStringConfig(ConfigKey.SITE_ADMIN_UIC);
|
||
Map<Integer, String> siteConfigMap = ConfigHelper.getConfigMap();
|
||
|
||
DBConfig config = new DBConfig();
|
||
config.setConfigMap(siteConfigMap);
|
||
config.setDbDir(dbDir);
|
||
config.setAdminAddress(adminHost);
|
||
config.setAdminPort(adminPort);
|
||
config.setAdminUic(adminUic);
|
||
config.setAdminServerName(PluginArgs.SITE_ADMIN_NAME);
|
||
config.setAdminIcon(getDefaultIcon(SiteDefaultIcon.DEFAULT_SITE_ADMIN_ICON));
|
||
config.setParam(PluginArgs.FRIEND_SQUARE, getDefaultIcon(SiteDefaultIcon.DEFAULT_FRIEND_SQUARE_ICON));
|
||
BootLog.info("{} init datasource config={}", WindProject.PLN, config.toString());
|
||
DataSourceManager.init(config);
|
||
}
|
||
|
||
/**
|
||
* 启动Http服务，提供与扩展服务之间的hai（http application interface）接口功能
|
||
*
|
||
* @throws HttpServerException
|
||
*/
|
||
private static void startNettyHttpServer(String address, int port) throws HttpServerException {
|
||
new HttpServer() {
|
||
|
||
@Override
|
||
public void loadExecutor(AbstracteExecutor<Command, CommandResponse> executor) {
|
||
executor.addChain(HttpUriAction.HTTP_ACTION.getRety(), new HttpRequestHandler());
|
||
}
|
||
|
||
}.start(address, port);
|
||
BootLog.info("{} start http server {}:{} ok.", WindProject.PLN, address, port);
|
||
}
|
||
|
||
/**
|
||
* 启动Netty服务器，提供用户与站点服务之间的长链接功能
|
||
*
|
||
* @param address
|
||
* @param port
|
||
* @throws TcpServerException
|
||
*/
|
||
private static void startNettyTcpServer(String address, int port) throws TcpServerException {
|
||
new NettyServer() {
|
||
|
||
@Override
|
||
public void loadExecutor(AbstracteExecutor<Command, CommandResponse> executor) {
|
||
executor.addChain(RequestAction.IM_SITE.getName(), new ImSiteAuthHandler());
|
||
executor.addChain(RequestAction.IM.getName(), new ImMessageHandler());
|
||
executor.addChain(RequestAction.API.getName(), new ApiRequestHandler());
|
||
}
|
||
|
||
}.start(address, port);
|
||
BootLog.info("{} start netty server {}:{} ok.", WindProject.PLN, address, port);
|
||
}
|
||
|
||
// websocket for web-im
|
||
private static void startWebSocketServer(String address, int port) throws Exception {
|
||
new WsServer() {
|
||
|
||
@Override
|
||
public void loadExecutor(AbstracteExecutor<Command, CommandResponse> executor) {
|
||
executor.addChain("WS-ACTION", new WSRequestHandler());
|
||
}
|
||
|
||
}.start(address, port);
|
||
}
|
||
|
||
// springboot for openzaly web
|
||
private static void initSpringBoot(String[] args) {
|
||
OpenzalySpringBoot.main(args);
|
||
}
|
||
|
||
// add config listener,timing to update cached config value
|
||
private static void addConfigListener() {
|
||
ConfigListener.startListenning();
|
||
BootLog.info("{} start listener to site-config", WindProject.PLN);
|
||
}
|
||
|
||
// get pic by base64
|
||
private static String getDefaultIcon(String base64Str) {
|
||
try {
|
||
String fileBasePath = System.getProperty("user.dir");
|
||
byte[] iconBytes = Base64.getDecoder().decode(base64Str);
|
||
String fileId = FileServerUtils.saveFile(iconBytes, FilePathUtils.getPicPath(fileBasePath),
|
||
FileType.SITE_PLUGIN, null);
|
||
return fileId;
|
||
} catch (Exception e) {
|
||
BootLog.error(StringHelper.format("{} set openzaly-admin default icon error", WindProject.PLN), e);
|
||
}
|
||
return "";
|
||
}
|
||
|
||
}
|