mirror of https://github.com/longtai-cn/hippo4j
chen.ma
3 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