package com.qianwen.core.launch;
|
|
import java.nio.charset.StandardCharsets;
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.Collection;
|
import java.util.Comparator;
|
import java.util.List;
|
import java.util.Properties;
|
import java.util.ServiceLoader;
|
import java.util.function.Function;
|
import java.util.stream.Collectors;
|
import com.qianwen.core.launch.constant.AppConstant;
|
import com.qianwen.core.launch.constant.NacosConstant;
|
import com.qianwen.core.launch.service.LauncherService;
|
import org.springframework.boot.builder.SpringApplicationBuilder;
|
import org.springframework.context.ConfigurableApplicationContext;
|
import org.springframework.core.env.MapPropertySource;
|
import org.springframework.core.env.MutablePropertySources;
|
import org.springframework.core.env.SimpleCommandLinePropertySource;
|
import org.springframework.core.env.StandardEnvironment;
|
import org.springframework.core.env.SystemEnvironmentPropertySource;
|
import org.springframework.util.Assert;
|
import org.springframework.util.StringUtils;
|
|
public class BladeApplication {
|
public static ConfigurableApplicationContext run(String appName, Class source, String... args) {
|
SpringApplicationBuilder builder = createSpringApplicationBuilder(appName, source, args);
|
return builder.run(args);
|
}
|
|
public static SpringApplicationBuilder createSpringApplicationBuilder(String appName, Class source, String... args) {
|
String profile;
|
Assert.hasText(appName, "[appName]服务名不能为空");
|
StandardEnvironment standardEnvironment = new StandardEnvironment();
|
MutablePropertySources propertySources = standardEnvironment.getPropertySources();
|
|
propertySources.addFirst(new SimpleCommandLinePropertySource(args));
|
propertySources.addLast(new MapPropertySource("systemProperties", standardEnvironment.getSystemProperties()));
|
propertySources.addLast(new SystemEnvironmentPropertySource("systemEnvironment", standardEnvironment.getSystemEnvironment()));
|
|
String[] activeProfiles = standardEnvironment.getActiveProfiles();
|
List<String> profiles = Arrays.asList(activeProfiles);
|
List<String> presetProfiles = new ArrayList<>(Arrays.asList(new String[] { "dev", "test", "prod" }));
|
presetProfiles.retainAll(profiles);
|
List<String> activeProfileList = new ArrayList<>(profiles);
|
Function<Object[], String> joinFun = StringUtils::arrayToCommaDelimitedString;
|
SpringApplicationBuilder builder = new SpringApplicationBuilder(new Class[] { source });
|
if (activeProfileList.isEmpty()) {
|
profile = "dev";
|
activeProfileList.add(profile);
|
builder.profiles(new String[] { profile });
|
} else if (activeProfileList.size() == 1) {
|
profile = activeProfileList.get(0);
|
} else {
|
throw new RuntimeException("同时存在环境变量:[" + StringUtils.arrayToCommaDelimitedString(activeProfiles) + "]");
|
}
|
String startJarPath = BladeApplication.class.getResource("/").getPath().split("!")[0];
|
String activePros = joinFun.apply(activeProfileList.toArray());
|
System.out.printf("----启动中，读取到的环境变量:[%s]，jar地址:[%s]----%n", new Object[] { activePros, startJarPath });
|
Properties props = System.getProperties();
|
props.setProperty("spring.application.name", appName);
|
props.setProperty("spring.profiles.active", profile);
|
props.setProperty("info.version", "9.3.0.0-SNAPSHOT");
|
props.setProperty("info.desc", appName);
|
props.setProperty("file.encoding", StandardCharsets.UTF_8.name());
|
props.setProperty("blade.env", profile);
|
props.setProperty("blade.name", appName);
|
props.setProperty("blade.is-local", String.valueOf(isLocalDev()));
|
props.setProperty("blade.dev-mode", profile.equals("prod") ? "false" : "true");
|
props.setProperty("blade.service.version", "9.3.0.0-SNAPSHOT");
|
Properties defaultProperties = new Properties();
|
defaultProperties.setProperty("spring.main.allow-bean-definition-overriding", "true");
|
defaultProperties.setProperty("spring.sleuth.sampler.percentage", "1.0");
|
defaultProperties.setProperty("spring.cloud.alibaba.seata.tx-service-group", appName.concat("-group"));
|
builder.properties(defaultProperties);
|
|
List<LauncherService> launcherList = new ArrayList<>();
|
ServiceLoader.<LauncherService>load(LauncherService.class).forEach(launcherList::add);
|
|
launcherList.stream().sorted(Comparator.comparing(LauncherService::getOrder)).collect(Collectors.toList())
|
.forEach(launcherService -> launcherService.launcher(builder, appName, profile, isLocalDev()));
|
return builder;
|
|
}
|
|
public static boolean isLocalDev() {
|
String osName = System.getProperty("os.name");
|
return (StringUtils.hasText(osName) && !"LINUX".equalsIgnoreCase(osName));
|
}
|
}
|
