mirror of
https://github.com/tencentmusic/supersonic.git
synced 2025-12-10 11:07:06 +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}")
|
||||
protected List<String> demoList;
|
||||
|
||||
@Value("${spring.datasource.driver-class-name}")
|
||||
protected String driverClassName;
|
||||
|
||||
public void run(String... args) {
|
||||
demoDatabase = addDatabaseIfNotExist();
|
||||
demoChatModel = addChatModelIfNotExist();
|
||||
@@ -122,10 +125,10 @@ public abstract class S2BaseDemo implements CommandLineRunner {
|
||||
databaseReq.setName("S2数据库DEMO");
|
||||
databaseReq.setDescription("样例数据库实例仅用于体验");
|
||||
databaseReq.setType(DataType.H2.getFeature());
|
||||
String profile = System.getProperty("spring.profiles.active");
|
||||
if ("postgres".equalsIgnoreCase(profile)) {
|
||||
if ("org.postgresql.Driver".equals(driverClassName)) {
|
||||
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.setUrl(url);
|
||||
|
||||
Reference in New Issue
Block a user