-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #164 from kmwtechnology/LC-536-add-JDBC-utils-for-…
…type-handling LC-536: added JDBCUtils for JDBC type handling
- Loading branch information
Showing
15 changed files
with
589 additions
and
56 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
132 changes: 132 additions & 0 deletions
132
lucille-core/src/main/java/com/kmwllc/lucille/util/JDBCUtils.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,132 @@ | ||
| package com.kmwllc.lucille.util; | ||
|
|
||
| import com.kmwllc.lucille.core.Document; | ||
| import java.sql.Date; | ||
| import java.sql.JDBCType; | ||
| import java.sql.ResultSet; | ||
| import java.sql.ResultSetMetaData; | ||
| import java.sql.SQLException; | ||
| import java.sql.Timestamp; | ||
| import java.sql.Types; | ||
| import java.util.Map; | ||
| import org.apache.logging.log4j.LogManager; | ||
| import org.apache.logging.log4j.Logger; | ||
|
|
||
| public class JDBCUtils { | ||
|
|
||
| private static final Logger log = LogManager.getLogger(JDBCUtils.class); | ||
|
|
||
| public static void parseResultToDoc(Document doc, ResultSet rs, Map<String, Object> fieldMapping) throws SQLException { | ||
| ResultSetMetaData metadata = rs.getMetaData(); | ||
| for (String key : fieldMapping.keySet()) { | ||
| String field = (String) fieldMapping.get(key); | ||
| int columnIndex = rs.findColumn(key); | ||
| int columnType = metadata.getColumnType(columnIndex); | ||
| typeHandling(doc, rs, field, columnIndex, columnType); | ||
| } | ||
| } | ||
|
|
||
| public static void parseResultToDoc(Document doc, ResultSet rs, String fieldName) | ||
| throws SQLException { | ||
| int columnIndex = rs.findColumn(fieldName); | ||
| parseResultToDoc(doc, rs, fieldName, columnIndex); | ||
| } | ||
|
|
||
| public static void parseResultToDoc(Document doc, ResultSet rs, String fieldName, int columnIndex) | ||
| throws SQLException { | ||
| ResultSetMetaData metadata = rs.getMetaData(); | ||
| typeHandling(doc, rs, fieldName, columnIndex, metadata.getColumnType(columnIndex)); | ||
| } | ||
|
|
||
| private static void typeHandling(Document doc, ResultSet rs, String fieldName, int columnIndex, int columnType) | ||
| throws SQLException { | ||
| switch (columnType) { | ||
| case Types.CLOB: | ||
| case Types.NCLOB: | ||
| case Types.VARCHAR: | ||
| case Types.CHAR: | ||
| case Types.LONGVARCHAR: | ||
| case Types.NCHAR: | ||
| case Types.NVARCHAR: | ||
| case Types.LONGNVARCHAR: | ||
| String str = rs.getString(columnIndex); | ||
| if (str != null) { | ||
| doc.setOrAdd(fieldName, str); | ||
| } | ||
| break; | ||
| case Types.TINYINT: | ||
| case Types.SMALLINT: | ||
| case Types.INTEGER: | ||
| int integer = rs.getInt(columnIndex); | ||
| // rs.getInt would return 0 if SQL value is null. This would to check that the field value is not null | ||
| if (!rs.wasNull()) { | ||
| doc.setOrAdd(fieldName, integer); | ||
| } | ||
| break; | ||
| case Types.BIGINT: | ||
| long longVal = rs.getLong(columnIndex); | ||
| if (!rs.wasNull()) { | ||
| doc.setOrAdd(fieldName,longVal); | ||
| } | ||
| break; | ||
| case Types.DECIMAL: | ||
| case Types.NUMERIC: | ||
| case Types.DOUBLE: | ||
| double doubleVal = rs.getDouble(columnIndex); | ||
| // rs.getDouble would return 0 if SQL value is null. This would to check that the field value is not null | ||
| if (!rs.wasNull()) { | ||
| doc.setOrAdd(fieldName, doubleVal); | ||
| } | ||
| break; | ||
| case Types.FLOAT: | ||
| case Types.REAL: | ||
| float floatVal = rs.getFloat(columnIndex); | ||
| // rs.getFloat would return 0 if SQL value is null. This would to check that the field value is not null | ||
| if (!rs.wasNull()) { | ||
| doc.setOrAdd(fieldName, floatVal); | ||
| } | ||
| break; | ||
| case Types.BOOLEAN: | ||
| case Types.BIT: | ||
| boolean boolVal = rs.getBoolean(columnIndex); | ||
| // rs.getBoolean would return false if SQL value is null. This would to check that the field value is not null | ||
| if (!rs.wasNull()) { | ||
| doc.setOrAdd(fieldName, boolVal); | ||
| } | ||
| break; | ||
| case Types.DATE: | ||
| // Currently saving date types as Strings, as conversion to an Instant may change the value given | ||
| // Date to Timestamp: sets nanoseconds & time to 0, e.g. 2024-07-30 -> 2024-07-30 00:00:00.0 | ||
| // Timestamp to Instant: checks either DB server local time (which might be from JVM), and converts to UTC, | ||
| // e.g. 2024-07-30 00:00:00.0 UTC-4 -> 2024-07-30 04:00:00.0 | ||
| Date dateVal = rs.getDate(columnIndex); | ||
| if (dateVal != null) { | ||
| doc.setOrAdd(fieldName, dateVal.toString()); | ||
| } | ||
| break; | ||
| case Types.TIMESTAMP: | ||
| // saving timestamp as str to prevent conversion to UTC | ||
| // if TIMESTAMP = 2023-01-01T00:00:00Z when local time is UTC-4 in JVM or db, | ||
| // will be stored as 2023-01-01T04:00:00Z in Document if converted to Instant | ||
| Timestamp timestamp = rs.getTimestamp(columnIndex); | ||
| if (timestamp != null) { | ||
| doc.setOrAdd(fieldName, timestamp.toString()); | ||
| } | ||
| break; | ||
| // case Types.TIMESTAMP_WITH_TIMEZONE: | ||
| // if TIMESTAMP_WITH_TIMEZONE = 2023-01-01T00:00:00-4, will be retrieved as 2023-01-01T04:00:00Z in getTimestamp() | ||
| // better not save it if it would be converted | ||
| case Types.BLOB: | ||
| case Types.BINARY: | ||
| case Types.VARBINARY: | ||
| case Types.LONGVARBINARY: | ||
| byte[] bytes = rs.getBytes(columnIndex); | ||
| if (bytes != null) { | ||
| doc.setOrAdd(fieldName, bytes); | ||
| } | ||
| break; | ||
| default: | ||
| log.warn("SQL Type {} ({}) not supported", JDBCType.valueOf(columnType).getName(), columnType); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.