mirror of
https://github.com/tencentmusic/supersonic.git
synced 2025-12-12 12:37:55 +00:00
[fix][launcher]Fix database type issue in demo classes.
This commit is contained in:
@@ -98,6 +98,9 @@ public abstract class S2BaseDemo implements CommandLineRunner {
|
|||||||
@Value("${s2.demo.names:S2VisitsDemo}")
|
@Value("${s2.demo.names:S2VisitsDemo}")
|
||||||
protected List<String> demoList;
|
protected List<String> demoList;
|
||||||
|
|
||||||
|
@Value("${spring.datasource.driver-class-name}")
|
||||||
|
protected String driverClassName;
|
||||||
|
|
||||||
public void run(String... args) {
|
public void run(String... args) {
|
||||||
demoDatabase = addDatabaseIfNotExist();
|
demoDatabase = addDatabaseIfNotExist();
|
||||||
demoChatModel = addChatModelIfNotExist();
|
demoChatModel = addChatModelIfNotExist();
|
||||||
@@ -122,10 +125,10 @@ public abstract class S2BaseDemo implements CommandLineRunner {
|
|||||||
databaseReq.setName("S2数据库DEMO");
|
databaseReq.setName("S2数据库DEMO");
|
||||||
databaseReq.setDescription("样例数据库实例仅用于体验");
|
databaseReq.setDescription("样例数据库实例仅用于体验");
|
||||||
databaseReq.setType(DataType.H2.getFeature());
|
databaseReq.setType(DataType.H2.getFeature());
|
||||||
String profile = System.getProperty("spring.profiles.active");
|
if ("org.postgresql.Driver".equals(driverClassName)) {
|
||||||
if ("postgres".equalsIgnoreCase(profile)) {
|
|
||||||
databaseReq.setType(DataType.POSTGRESQL.getFeature());
|
databaseReq.setType(DataType.POSTGRESQL.getFeature());
|
||||||
} else if ("mysql".equalsIgnoreCase(profile)) {
|
} else if ("com.mysql.cj.jdbc.Driver".equals(driverClassName)
|
||||||
|
|| "com.mysql.jdbc.Driver".equals(driverClassName)) {
|
||||||
databaseReq.setType(DataType.MYSQL.getFeature());
|
databaseReq.setType(DataType.MYSQL.getFeature());
|
||||||
}
|
}
|
||||||
databaseReq.setUrl(url);
|
databaseReq.setUrl(url);
|
||||||
|
|||||||
Reference in New Issue
Block a user