mirror of https://github.com/longtai-cn/hippo4j
chen.ma
2 years ago
parent
bbe1e989c2
commit
7dc30efc53
@ -0,0 +1,42 @@
|
||||
/*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||
* contributor license agreements. See the NOTICE file distributed with
|
||||
* this work for additional information regarding copyright ownership.
|
||||
* The ASF licenses this file to You 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 cn.hippo4j.server.config;
|
||||
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
/**
|
||||
* Database configuration.
|
||||
*
|
||||
* @see org.apache.shenyu.admin.config.DataBaseConfiguration
|
||||
*/
|
||||
@Configuration
|
||||
public class DataBaseConfiguration {
|
||||
|
||||
@Bean
|
||||
@ConditionalOnMissingBean(value = DataBaseProperties.class)
|
||||
public DataBaseProperties dataBaseProperties(@Value("${hippo4j.database.init_script:sql-script/h2/schema.sql}") String initScript,
|
||||
@Value("${hippo4j.database.init_enable:true}") Boolean initEnable) {
|
||||
DataBaseProperties dataSourceProperties = new DataBaseProperties();
|
||||
dataSourceProperties.setInitScript(initScript);
|
||||
dataSourceProperties.setInitEnable(initEnable);
|
||||
return dataSourceProperties;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,41 @@
|
||||
/*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||
* contributor license agreements. See the NOTICE file distributed with
|
||||
* this work for additional information regarding copyright ownership.
|
||||
* The ASF licenses this file to You 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 cn.hippo4j.server.config;
|
||||
|
||||
import lombok.Data;
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
|
||||
/**
|
||||
* Database properties.
|
||||
*
|
||||
* @see org.apache.shenyu.admin.config.properties.DataBaseProperties
|
||||
*/
|
||||
@Data
|
||||
@ConfigurationProperties(prefix = "hippo4j.database")
|
||||
public class DataBaseProperties {
|
||||
|
||||
/**
|
||||
* Init script
|
||||
*/
|
||||
private String initScript;
|
||||
|
||||
/**
|
||||
* Init enable
|
||||
*/
|
||||
private Boolean initEnable;
|
||||
}
|
||||
@ -0,0 +1,110 @@
|
||||
/*
|
||||
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||
* contributor license agreements. See the NOTICE file distributed with
|
||||
* this work for additional information regarding copyright ownership.
|
||||
* The ASF licenses this file to You 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 cn.hippo4j.server.init;
|
||||
|
||||
import cn.hippo4j.server.config.DataBaseProperties;
|
||||
import com.google.common.base.Splitter;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.apache.commons.lang3.StringUtils;
|
||||
import org.apache.ibatis.io.Resources;
|
||||
import org.apache.ibatis.jdbc.ScriptRunner;
|
||||
import org.springframework.beans.BeansException;
|
||||
import org.springframework.beans.factory.config.InstantiationAwareBeanPostProcessor;
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnExpression;
|
||||
import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;
|
||||
import org.springframework.lang.NonNull;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStreamReader;
|
||||
import java.io.Reader;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.sql.Connection;
|
||||
import java.sql.DriverManager;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* Local datasource loader.
|
||||
*
|
||||
* @see org.apache.shenyu.admin.spring.LocalDataSourceLoader
|
||||
*/
|
||||
@Slf4j
|
||||
@Component
|
||||
@ConditionalOnExpression("'${hippo4j.database.dialect}' == 'mysql' or '${hippo4j.database.dialect}' == 'h2'")
|
||||
public class LocalDataSourceLoader implements InstantiationAwareBeanPostProcessor {
|
||||
|
||||
private static final String PRE_FIX = "file:";
|
||||
|
||||
@Resource
|
||||
private DataBaseProperties dataBaseProperties;
|
||||
|
||||
@Override
|
||||
public Object postProcessAfterInitialization(@NonNull final Object bean, final String beanName) throws BeansException {
|
||||
if ((bean instanceof DataSourceProperties) && dataBaseProperties.getInitEnable()) {
|
||||
this.init((DataSourceProperties) bean);
|
||||
}
|
||||
return bean;
|
||||
}
|
||||
|
||||
protected void init(final DataSourceProperties properties) {
|
||||
try {
|
||||
// If jdbcUrl in the configuration file specifies the hippo4j database, it is removed,
|
||||
// because the hippo4j database does not need to be specified when executing the SQL file,
|
||||
// otherwise the hippo4j database will be disconnected when the hippo4j database does not exist
|
||||
String jdbcUrl = StringUtils.replace(properties.getUrl(), "/hippo4j_manager?", "?");
|
||||
Connection connection = DriverManager.getConnection(jdbcUrl, properties.getUsername(), properties.getPassword());
|
||||
this.execute(connection, dataBaseProperties.getInitScript());
|
||||
} catch (Exception ex) {
|
||||
log.error("Datasource init error.", ex);
|
||||
throw new RuntimeException(ex.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
protected void execute(final Connection conn, final String script) throws Exception {
|
||||
ScriptRunner runner = new ScriptRunner(conn);
|
||||
try {
|
||||
// Doesn't print logger
|
||||
runner.setLogWriter(null);
|
||||
runner.setAutoCommit(true);
|
||||
Resources.setCharset(StandardCharsets.UTF_8);
|
||||
List<String> initScripts = Splitter.on(";").splitToList(script);
|
||||
for (String sqlScript : initScripts) {
|
||||
if (sqlScript.startsWith(PRE_FIX)) {
|
||||
String sqlFile = sqlScript.substring(PRE_FIX.length());
|
||||
try (Reader fileReader = getResourceAsReader(sqlFile)) {
|
||||
log.info("Execute hippo4j schema sql: {}", sqlFile);
|
||||
runner.runScript(fileReader);
|
||||
}
|
||||
} else {
|
||||
try (Reader fileReader = Resources.getResourceAsReader(sqlScript)) {
|
||||
log.info("Execute hippo4j schema sql: {}", sqlScript);
|
||||
runner.runScript(fileReader);
|
||||
}
|
||||
}
|
||||
}
|
||||
} finally {
|
||||
conn.close();
|
||||
}
|
||||
}
|
||||
|
||||
private static Reader getResourceAsReader(final String resource) throws IOException {
|
||||
return new InputStreamReader(new FileInputStream(resource), StandardCharsets.UTF_8);
|
||||
}
|
||||
}
|
||||
@ -1,5 +0,0 @@
|
||||
### Data source customization section
|
||||
spring.datasource.driver-class-name=com.mysql.cj.jdbc.Driver
|
||||
spring.datasource.url=jdbc:mysql://localhost:3306/hippo4j_manager?characterEncoding=utf-8&zeroDateTimeBehavior=convertToNull&transformedBitIsBoolean=true&serverTimezone=GMT%2B8
|
||||
spring.datasource.username=root
|
||||
spring.datasource.password=root
|
||||
Loading…
Reference in new issue