io.datarouter.client.mysql.util.MysqlTool Maven / Gradle / Ivy
/*
* Copyright © 2009 HotPads ([email protected])
*
* 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 io.datarouter.client.mysql.util;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Supplier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.mysql.cj.log.Slf4JLogger;
import io.datarouter.client.mysql.connection.MysqlConnectionPoolHolder.MysqlConnectionPool;
import io.datarouter.client.mysql.field.codec.factory.MysqlFieldCodecFactory;
import io.datarouter.instrumentation.trace.TraceSpanGroupType;
import io.datarouter.instrumentation.trace.TracerTool;
import io.datarouter.model.databean.Databean;
import io.datarouter.model.exception.DataAccessException;
import io.datarouter.model.field.Field;
import io.datarouter.model.index.IndexEntry;
import io.datarouter.model.key.primary.PrimaryKey;
import io.datarouter.model.serialize.fielder.DatabeanFielder;
import io.datarouter.storage.serialize.fieldcache.DatabeanFieldInfo;
import io.datarouter.storage.serialize.fieldcache.IndexEntryFieldInfo;
public class MysqlTool{
private static final Logger logger = LoggerFactory.getLogger(MysqlTool.class);
public static final String PRIMARY_KEY_INDEX_NAME = "PRIMARY";
private static final String TABLE_CATALOG = "TABLE_CAT";
public static Connection openConnection(String hostname, int port, String user, String password){
logger.warn("hostname={}, port={}, user={}", hostname, port, user);
try{
String url = String.format("jdbc:mysql://%s:%s?user=%s&password=%s&logger=%s&enabledTLSProtocols=TLSv1.2",
hostname,
port,
user,
password,
Slf4JLogger.class.getName());
return DriverManager.getConnection(url);
}catch(Exception e){
String message = String.format("failed to connect hostname=%s port=%s user=%s",
hostname,
port,
user);
throw new RuntimeException(message, e);
}
}
public static List showTables(Connection connection, String schemaName){
try{
List tableNames = new ArrayList<>();
ResultSet rs = connection.getMetaData().getTables(schemaName, schemaName, "%", null);
while(rs.next()){
tableNames.add(rs.getString(3));
}
return tableNames;
}catch(SQLException e){
throw new RuntimeException(e);
}
}
public static List showDatabases(Connection connection){
try{
ResultSet rs = connection.getMetaData().getCatalogs();
List catalogs = new ArrayList<>();
while(rs.next()){
catalogs.add(rs.getString(TABLE_CATALOG));
}
return catalogs;
}catch(SQLException e){
throw new RuntimeException(e);
}
}
public static <
PK extends PrimaryKey,
D extends Databean,
F extends DatabeanFielder>
List selectPrimaryKeys(
MysqlFieldCodecFactory fieldCodecFactory,
DatabeanFieldInfo fieldInfo,
PreparedStatement ps){
try{
String spanName = fieldInfo.getPrimaryKeySupplier().get().getClass().getSimpleName()
+ " selectPrimaryKeys PreparedStatement.execute";
try(var $ = TracerTool.startSpan(spanName, TraceSpanGroupType.DATABASE)){
ps.execute();
}
ResultSet rs = ps.getResultSet();
List primaryKeys = new ArrayList<>();
while(rs.next()){
PK primaryKey = fieldSetFromMysqlResultSetUsingReflection(
fieldCodecFactory,
fieldInfo.getPrimaryKeySupplier(),
fieldInfo.getPrimaryKeyFields(),
rs);
primaryKeys.add(primaryKey);
}
return primaryKeys;
}catch(Exception e){
throw new DataAccessException(ps.toString(), e);
}
}
public static ,D extends Databean>
List selectDatabeans(
MysqlFieldCodecFactory fieldCodecFactory,
Supplier databeanSupplier,
List> fields,
PreparedStatement ps){
try{
String spanName = databeanSupplier.get().getDatabeanName() + " selectDatabeans PreparedStatement.execute";
try(var $ = TracerTool.startSpan(spanName, TraceSpanGroupType.DATABASE)){
ps.execute();
}
return getDatabeansFromSelectResult(fieldCodecFactory, databeanSupplier, fields, ps);
}catch(Exception e){
String message = "error executing sql:" + ps;
throw new DataAccessException(message, e);
}
}
public static ,D extends Databean>
List getDatabeansFromSelectResult(
MysqlFieldCodecFactory fieldCodecFactory,
Supplier databeanSupplier,
List> fields,
PreparedStatement ps){
String spanName = databeanSupplier.get().getDatabeanName() + " getDatabeansFromSelectResult";
try(var $ = TracerTool.startSpan(spanName, TraceSpanGroupType.SERIALIZATION)){
ResultSet rs = ps.getResultSet();
List databeans = new ArrayList<>();
while(rs.next()){
D databean = fieldSetFromMysqlResultSetUsingReflection(fieldCodecFactory, databeanSupplier, fields, rs);
databeans.add(databean);
}
return databeans;
}catch(Exception e){
throw new RuntimeException(e);
}
}
public static ,
D extends Databean,
IK extends PrimaryKey,
IE extends IndexEntry,
IF extends DatabeanFielder>
List selectIndexEntryKeys(
MysqlFieldCodecFactory fieldCodecFactory,
IndexEntryFieldInfo fieldInfo,
PreparedStatement ps){
try{
String spanName = fieldInfo.getPrimaryKeySupplier().get().getClass().getSimpleName()
+ " selectIndexEntryKeys PreparedStatement.execute";
try(var $ = TracerTool.startSpan(spanName, TraceSpanGroupType.DATABASE)){
ps.execute();
}
ResultSet rs = ps.getResultSet();
List keys = new ArrayList<>();
while(rs.next()){
IK key = fieldSetFromMysqlResultSetUsingReflection(
fieldCodecFactory,
fieldInfo.getPrimaryKeySupplier(),
fieldInfo.getPrimaryKeyFields(),
rs);
keys.add(key);
}
return keys;
}catch(Exception e){
String message = "error executing sql:" + ps;
throw new DataAccessException(message, e);
}
}
public static int update(PreparedStatement statement){
String spanName = "update PreparedStatement.execute";
try(var $ = TracerTool.startSpan(spanName, TraceSpanGroupType.DATABASE)){
return statement.executeUpdate();
}catch(SQLException e){
String message = "error executing sql:" + statement;
throw new DataAccessException(message, e);
}
}
public static F fieldSetFromMysqlResultSetUsingReflection(
MysqlFieldCodecFactory fieldCodecFactory,
Supplier supplier,
List> fields,
ResultSet rs){
F targetFieldSet = supplier.get();
for(Field field : fields){
fromMysqlResultSetUsingReflection(fieldCodecFactory, field, rs, targetFieldSet);
}
return targetFieldSet;
}
public static ,FS> void fromMysqlResultSetUsingReflection(
MysqlFieldCodecFactory fieldCodecFactory,
F field,
ResultSet rs,
FS targetFieldSet){
var codec = fieldCodecFactory.createCodec(field);
Object value = codec.fromMysqlResultSetButDoNotSet(rs, field);
field.setUsingReflection(targetFieldSet, value);
}
public static PreparedStatement prepareStatement(Connection connection, String sql){
try{
return connection.prepareStatement(sql);
}catch(SQLException e){
throw new RuntimeException(sql, e);
}
}
public static void execute(MysqlConnectionPool connectionPool, String sql){
try(Connection connection = connectionPool.checkOut()){
connection.createStatement().execute(sql);
}catch(SQLException e){
throw new RuntimeException(sql, e);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy