diff --git a/datasource/metadatamanager/mysql/service/pom.xml b/datasource/metadatamanager/mysql/service/pom.xml new file mode 100644 index 0000000000..bb3cba0677 --- /dev/null +++ b/datasource/metadatamanager/mysql/service/pom.xml @@ -0,0 +1,83 @@ + + + + + + + linkis + com.webank.wedatasphere.linkis + 0.9.3 + + 4.0.0 + + linkis-metadatamanager-service-mysql + + + UTF-8 + 5.1.34 + + + + + com.webank.wedatasphere.linkis + linkis-metadatamanager-common + ${linkis.version} + + + com.webank.wedatasphere.linkis + linkis-module + ${linkis.version} + + + asm + org.ow2.asm + + + mysql + mysql-connector-java + + + + + mysql + mysql-connector-java + ${mysql.version} + + + + + + + org.apache.maven.plugins + maven-deploy-plugin + + + + net.alchim31.maven + scala-maven-plugin + + + org.apache.maven.plugins + maven-jar-plugin + + + + + src/main/resources + + + ${project.artifactId}-${project.version} + + diff --git a/datasource/metadatamanager/mysql/service/src/main/java/com/webank/wedatasphere/linkis/metadatamanager/service/SqlConnection.java b/datasource/metadatamanager/mysql/service/src/main/java/com/webank/wedatasphere/linkis/metadatamanager/service/SqlConnection.java new file mode 100644 index 0000000000..383a0f1380 --- /dev/null +++ b/datasource/metadatamanager/mysql/service/src/main/java/com/webank/wedatasphere/linkis/metadatamanager/service/SqlConnection.java @@ -0,0 +1,210 @@ +/* + * Copyright 2019 WeBank + * Licensed 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 com.webank.wedatasphere.linkis.metadatamanager.service; + +import com.webank.wedatasphere.linkis.common.conf.CommonVars; +import com.webank.wedatasphere.linkis.metadatamanager.common.domain.MetaColumnInfo; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import java.io.Closeable; +import java.io.IOException; +import java.sql.*; +import java.util.ArrayList; +import java.util.List; +import java.util.Map; +import java.util.stream.Collectors; + +/** + * @author davidhua + * 2020/02/14 + */ +public class SqlConnection implements Closeable { + + private static final Logger LOG = LoggerFactory.getLogger(SqlConnection.class); + + private static final CommonVars SQL_DRIVER_CLASS = + CommonVars.apply("wds.linkis.server.mdm.service.sql.driver", "com.mysql.jdbc.Driver"); + + private static final CommonVars SQL_CONNECT_URL = + CommonVars.apply("wds.linkis.server.mdm.service.sql.url", "jdbc:mysql://%s:%s/%s"); + + private Connection conn; + + private ConnectMessage connectMessage; + + public SqlConnection(String host, Integer port, + String username, String password, + Map extraParams ) throws ClassNotFoundException, SQLException { + connectMessage = new ConnectMessage(host, port, username, password, extraParams); + conn = getDBConnection(connectMessage, ""); + //Try to create statement + Statement statement = conn.createStatement(); + statement.close(); + } + + public List getAllDatabases() throws SQLException { + java.util.List dataBaseName = new ArrayList<>(); + Statement stmt = null; + ResultSet rs = null; + try{ + stmt = conn.createStatement(); + rs = stmt.executeQuery("SHOW DATABASES"); + while (rs.next()){ + dataBaseName.add(rs.getString(1)); + } + } finally { + closeResource(null, stmt, rs); + } + return dataBaseName; + } + + public List getAllTables(String database) throws SQLException { + List tableNames = new ArrayList<>(); + Statement stmt = null; + ResultSet rs = null; + try { + stmt = conn.createStatement(); + rs = stmt.executeQuery("SHOW TABLES FROM `" + database + "`"); + while (rs.next()) { + tableNames.add(rs.getString(1)); + } + return tableNames; + } finally{ + closeResource(null, stmt, rs); + } + } + + public List getColumns(String database, String table) throws SQLException, ClassNotFoundException { + List columns = new ArrayList<>(); + String columnSql = "SELECT * FROM `" + database +"`.`" + table + "` WHERE 1 = 2"; + PreparedStatement ps = null; + ResultSet rs = null; + ResultSetMetaData meta = null; + try { + List primaryKeys = getPrimaryKeys(getDBConnection(connectMessage, database), table); + ps = conn.prepareStatement(columnSql); + rs = ps.executeQuery(); + meta = rs.getMetaData(); + int columnCount = meta.getColumnCount(); + for (int i = 1; i < columnCount + 1; i++) { + MetaColumnInfo info = new MetaColumnInfo(); + info.setIndex(i); + info.setName(meta.getColumnName(i)); + info.setType(meta.getColumnTypeName(i)); + if(primaryKeys.contains(meta.getColumnName(i))){ + info.setPrimaryKey(true); + } + columns.add(info); + } + }finally { + closeResource(null, ps, rs); + } + return columns; + } + + /** + * Get primary keys + * @param connection connection + * @param table table name + * @return + * @throws SQLException + */ + private List getPrimaryKeys(Connection connection, String table) throws SQLException { + ResultSet rs = null; + List primaryKeys = new ArrayList<>(); + try { + DatabaseMetaData dbMeta = connection.getMetaData(); + rs = dbMeta.getPrimaryKeys(null, null, table); + while(rs.next()){ + primaryKeys.add(rs.getString("column_name")); + } + return primaryKeys; + }finally{ + if(null != rs){ + closeResource(connection, null, rs); + } + } + } + + /** + * close database resource + * @param connection connection + * @param statement statement + * @param resultSet result set + */ + private void closeResource(Connection connection, Statement statement, ResultSet resultSet){ + try { + if(null != resultSet && !resultSet.isClosed()) { + resultSet.close(); + } + if(null != statement && !statement.isClosed()){ + statement.close(); + } + if(null != connection && !connection.isClosed()){ + connection.close(); + } + }catch (SQLException e){ + LOG.warn("Fail to release resource [" + e.getMessage() +"]", e); + } + } + + @Override + public void close() throws IOException { + closeResource(conn, null, null); + } + + /** + * @param connectMessage + * @param database + * @return + * @throws ClassNotFoundException + */ + private Connection getDBConnection(ConnectMessage connectMessage, String database) throws ClassNotFoundException, SQLException { + String extraParamString = connectMessage.extraParams.entrySet() + .stream().map(e -> String.join("=", e.getKey(), String.valueOf(e.getValue()))) + .collect(Collectors.joining("&")); + Class.forName(SQL_DRIVER_CLASS.getValue()); + String url = String.format(SQL_CONNECT_URL.getValue(), connectMessage.host, connectMessage.port, database); + if(!connectMessage.extraParams.isEmpty()) { + url += "?" + extraParamString; + } + return DriverManager.getConnection(url, connectMessage.username, connectMessage.password); + } + + /** + * Connect message + */ + private static class ConnectMessage{ + private String host; + + private Integer port; + + private String username; + + private String password; + + private Map extraParams; + + public ConnectMessage(String host, Integer port, + String username, String password, + Map extraParams){ + this.host = host; + this.port = port; + this.username = username; + this.password = password; + this.extraParams = extraParams; + } + } +} diff --git a/datasource/metadatamanager/mysql/service/src/main/java/com/webank/wedatasphere/linkis/metadatamanager/service/SqlMetaService.java b/datasource/metadatamanager/mysql/service/src/main/java/com/webank/wedatasphere/linkis/metadatamanager/service/SqlMetaService.java new file mode 100644 index 0000000000..744ee4a051 --- /dev/null +++ b/datasource/metadatamanager/mysql/service/src/main/java/com/webank/wedatasphere/linkis/metadatamanager/service/SqlMetaService.java @@ -0,0 +1,79 @@ +/* + * Copyright 2019 WeBank + * Licensed 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 com.webank.wedatasphere.linkis.metadatamanager.service; + +import com.webank.wedatasphere.linkis.metadatamanager.common.Json; +import com.webank.wedatasphere.linkis.metadatamanager.common.domain.MetaColumnInfo; +import com.webank.wedatasphere.linkis.metadatamanager.common.service.AbstractMetaService; +import com.webank.wedatasphere.linkis.metadatamanager.common.service.MetadataConnection; +import org.springframework.stereotype.Component; + +import java.sql.SQLException; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +/** + * @author davidhua + * 2020/02/14 + */ +@Component +public class SqlMetaService extends AbstractMetaService { + @Override + public MetadataConnection getConnection(String operator, Map params) throws Exception { + String host = String.valueOf(params.getOrDefault(SqlParamsMapper.PARAM_SQL_HOST.getValue(), "")); + //After deserialize, Integer will be Double, Why? + Integer port = (Double.valueOf(String.valueOf(params.getOrDefault(SqlParamsMapper.PARAM_SQL_PORT.getValue(), 0)))).intValue(); + String username = String.valueOf(params.getOrDefault(SqlParamsMapper.PARAM_SQL_USERNAME.getValue(), "")); + String password = String.valueOf(params.getOrDefault(SqlParamsMapper.PARAM_SQL_PASSWORD.getValue(), "")); + Map extraParams = new HashMap<>(); + Object sqlParamObj = params.get(SqlParamsMapper.PARAM_SQL_EXTRA_PARAMS.getValue()); + if(null != sqlParamObj){ + if(!(sqlParamObj instanceof Map)){ + extraParams = Json.fromJson(String.valueOf(sqlParamObj), Map.class, String.class, Object.class); + }else{ + extraParams = (Map)sqlParamObj; + } + } + assert extraParams != null; + return new MetadataConnection<>(new SqlConnection(host, port, username, password, extraParams)); + } + + @Override + public List queryDatabases(SqlConnection connection) { + try { + return connection.getAllDatabases(); + } catch (SQLException e) { + throw new RuntimeException("Fail to get Sql databases(获取数据库列表失败)", e); + } + } + + @Override + public List queryTables(SqlConnection connection, String database) { + try { + return connection.getAllTables(database); + } catch (SQLException e) { + throw new RuntimeException("Fail to get Sql tables(获取表列表失败)", e); + } + } + + @Override + public List queryColumns(SqlConnection connection, String database, String table) { + try { + return connection.getColumns(database, table); + } catch (SQLException | ClassNotFoundException e) { + throw new RuntimeException("Fail to get Sql columns(获取字段列表失败)", e); + } + } +} diff --git a/datasource/metadatamanager/mysql/service/src/main/java/com/webank/wedatasphere/linkis/metadatamanager/service/SqlParamsMapper.java b/datasource/metadatamanager/mysql/service/src/main/java/com/webank/wedatasphere/linkis/metadatamanager/service/SqlParamsMapper.java new file mode 100644 index 0000000000..c41285e4e0 --- /dev/null +++ b/datasource/metadatamanager/mysql/service/src/main/java/com/webank/wedatasphere/linkis/metadatamanager/service/SqlParamsMapper.java @@ -0,0 +1,38 @@ +/* + * Copyright 2019 WeBank + * Licensed 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 com.webank.wedatasphere.linkis.metadatamanager.service; + +import com.webank.wedatasphere.linkis.common.conf.CommonVars; + +/** + * @author davidhua + * 2020/02/14 + */ +public class SqlParamsMapper { + + public static final CommonVars PARAM_SQL_HOST = + CommonVars.apply("wds.linkis.server.mdm.service.sql.host", "host"); + + public static final CommonVars PARAM_SQL_PORT = + CommonVars.apply("wds.linkis.server.mdm.service.sql.port", "port"); + + public static final CommonVars PARAM_SQL_USERNAME = + CommonVars.apply("wds.linkis.server.mdm.service.sql.username", "username"); + + public static final CommonVars PARAM_SQL_PASSWORD = + CommonVars.apply("wds.linkis.server.mdm.service.sql.password", "password"); + + public static final CommonVars PARAM_SQL_EXTRA_PARAMS = + CommonVars.apply("wds.linkis.server.mdm.service.sql.params", "params"); +} diff --git a/datasource/metadatamanager/mysql/service/src/main/resources/application.yml b/datasource/metadatamanager/mysql/service/src/main/resources/application.yml new file mode 100644 index 0000000000..bb31831572 --- /dev/null +++ b/datasource/metadatamanager/mysql/service/src/main/resources/application.yml @@ -0,0 +1,23 @@ +server: + port: 8294 +spring: + application: + name: mdm-service-mysql + +eureka: + client: + serviceUrl: + defaultZone: http://${ip}:${port}/eureka/ + instance: + metadata-map: + test: wedatasphere + +management: + endpoints: + web: + exposure: + include: refresh,info +logging: + config: classpath:log4j2.xml + + diff --git a/datasource/metadatamanager/mysql/service/src/main/resources/linkis.properties b/datasource/metadatamanager/mysql/service/src/main/resources/linkis.properties new file mode 100644 index 0000000000..f83c10d281 --- /dev/null +++ b/datasource/metadatamanager/mysql/service/src/main/resources/linkis.properties @@ -0,0 +1,21 @@ +# +# Copyright 2019 WeBank +# Licensed 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. +# + +wds.linkis.server.mybatis.mapperLocations= +wds.linkis.server.mybatis.typeAliasesPackage= +wds.linkis.server.mybatis.BasePackage= +wds.linkis.server.restful.scan.packages= + +#sit +wds.linkis.server.version=v1 + diff --git a/datasource/metadatamanager/mysql/service/src/main/resources/log4j.properties b/datasource/metadatamanager/mysql/service/src/main/resources/log4j.properties new file mode 100644 index 0000000000..d5ee44b86b --- /dev/null +++ b/datasource/metadatamanager/mysql/service/src/main/resources/log4j.properties @@ -0,0 +1,33 @@ +# +# Copyright 2019 WeBank +# Licensed 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. +# + +### set log levels ### + +log4j.rootCategory=INFO,console + +log4j.appender.console=org.apache.log4j.ConsoleAppender +log4j.appender.console.Threshold=INFO +log4j.appender.console.layout=org.apache.log4j.PatternLayout +#log4j.appender.console.layout.ConversionPattern= %d{ISO8601} %-5p (%t) [%F:%M(%L)] - %m%n +log4j.appender.console.layout.ConversionPattern= %d{ISO8601} %-5p (%t) %p %c{1} - %m%n + + +log4j.appender.com.webank.bdp.ide.core=org.apache.log4j.DailyRollingFileAppender +log4j.appender.com.webank.bdp.ide.core.Threshold=INFO +log4j.additivity.com.webank.bdp.ide.core=false +log4j.appender.com.webank.bdp.ide.core.layout=org.apache.log4j.PatternLayout +log4j.appender.com.webank.bdp.ide.core.Append=true +log4j.appender.com.webank.bdp.ide.core.File=logs/linkis.log +log4j.appender.com.webank.bdp.ide.core.layout.ConversionPattern= %d{ISO8601} %-5p (%t) [%F:%M(%L)] - %m%n + +log4j.logger.org.springframework=INFO \ No newline at end of file diff --git a/datasource/metadatamanager/mysql/service/src/main/resources/log4j2.xml b/datasource/metadatamanager/mysql/service/src/main/resources/log4j2.xml new file mode 100644 index 0000000000..1c68190669 --- /dev/null +++ b/datasource/metadatamanager/mysql/service/src/main/resources/log4j2.xml @@ -0,0 +1,35 @@ + + + + + + + + + + + + + + + + + + + + + + + diff --git a/datasource/metadatamanager/mysql/service/src/main/scala/com/webank/wedatasphere/linkis/metadatamanager/service/receiver/SqlReceiver.scala b/datasource/metadatamanager/mysql/service/src/main/scala/com/webank/wedatasphere/linkis/metadatamanager/service/receiver/SqlReceiver.scala new file mode 100644 index 0000000000..11f24a9ee9 --- /dev/null +++ b/datasource/metadatamanager/mysql/service/src/main/scala/com/webank/wedatasphere/linkis/metadatamanager/service/receiver/SqlReceiver.scala @@ -0,0 +1,28 @@ +/* + * Copyright 2019 WeBank + * Licensed 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 com.webank.wedatasphere.linkis.metadatamanager.service.receiver + +import com.webank.wedatasphere.linkis.DataWorkCloudApplication +import com.webank.wedatasphere.linkis.metadatamanager.common.receiver.BaseMetaReceiver +import com.webank.wedatasphere.linkis.metadatamanager.common.service.MetadataService +import javax.annotation.PostConstruct +import org.springframework.stereotype.Component + +@Component +class SqlReceiver extends BaseMetaReceiver{ + @PostConstruct + def init(): Unit = { + metadataService = DataWorkCloudApplication.getApplicationContext.getBean(classOf[MetadataService]) + } +}