Add support for custom JdbcCustomization via Spring Boot

db-scheduler/src/main/java/com/github/kagkarlsson/scheduler/jdbc/AutodetectJdbcCustomization.java

@@ -43,8 +43,13 @@ public AutodetectJdbcCustomization(DataSource dataSource) {
             LOG.info("Detected database {}.", databaseProductName);
 
             if (databaseProductName.equals(MICROSOFT_SQL_SERVER)) {
-                LOG.info("Using MSSQL jdbc-overrides.");
-                detectedCustomization = new MssqlJdbcCustomization();
+                if (c.getMetaData().getDriverName().contains("jTDS")) {
+                    LOG.info("Using MSSQL jTDS jdbc-overrides.");
+                    detectedCustomization = new MssqlJtdsJdbcCustomization();
+                } else {
+                    LOG.info("Using MSSQL jdbc-overrides.");
+                    detectedCustomization = new MssqlJdbcCustomization();
+                }
             } else if (databaseProductName.equals(POSTGRESQL)) {
                 LOG.info("Using PostgreSQL jdbc-overrides.");
                 detectedCustomization = new PostgreSqlJdbcCustomization();
@@ -67,6 +72,11 @@ public Instant getInstant(ResultSet rs, String columnName) throws SQLException {
         return jdbcCustomization.getInstant(rs, columnName);
     }
 
+    @Override
+    public byte[] getBytes(ResultSet rs, String columnName) throws SQLException {
+        return jdbcCustomization.getBytes(rs, columnName);
+    }
+
     @Override
     public boolean supportsExplicitQueryLimitPart() {
         return jdbcCustomization.supportsExplicitQueryLimitPart();

db-scheduler/src/main/java/com/github/kagkarlsson/scheduler/jdbc/JdbcCustomization.java

@@ -29,6 +29,10 @@ public interface JdbcCustomization {
     void setInstant(PreparedStatement p, int index, Instant value) throws SQLException;
     Instant getInstant(ResultSet rs, String columnName) throws SQLException;
 
+    default byte[] getBytes(ResultSet rs, String columnName) throws SQLException {
+        return rs.getBytes(columnName);
+    }
+
     boolean supportsExplicitQueryLimitPart();
     String getQueryLimitPart(int limit);
 

db-scheduler/src/main/java/com/github/kagkarlsson/scheduler/jdbc/JdbcTaskRepository.java

@@ -470,7 +470,7 @@ public Void map(ResultSet rs) throws SQLException {
                 }
 
                 String instanceId = rs.getString("task_instance");
-                byte[] data = rs.getBytes("task_data");
+                byte[] data = jdbcCustomization.getBytes(rs,"task_data");
 
                 Instant executionTime = jdbcCustomization.getInstant(rs, "execution_time");
 

db-scheduler/src/main/java/com/github/kagkarlsson/scheduler/jdbc/MssqlJtdsJdbcCustomization.java

@@ -0,0 +1,48 @@
+/**
+ * Copyright (C) Gustav Karlsson
+ *
+ * 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.github.kagkarlsson.scheduler.jdbc;
+
+import java.io.IOException;
+import java.io.Reader;
+import java.nio.charset.StandardCharsets;
+import java.sql.Clob;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+public class MssqlJtdsJdbcCustomization extends MssqlJdbcCustomization {
+
+    @Override
+    public String getName() {
+        return "MSSQL-jTDS";
+    }
+
+    @Override
+    public byte[] getBytes(ResultSet rs, String columnName) throws SQLException {
+        byte[] result;
+        try (Reader reader = ((Clob) rs.getObject(columnName)).getCharacterStream()) {
+            char[] charArray = new char[8 * 1024];
+            StringBuilder builder = new StringBuilder();
+            int numCharsRead;
+            while ((numCharsRead = reader.read(charArray, 0, charArray.length)) != -1) {
+                builder.append(charArray, 0, numCharsRead);
+            }
+            result = builder.toString().getBytes(StandardCharsets.UTF_16LE);
+        } catch (IOException e) {
+            throw new SQLException(e);
+        }
+        return result;
+    }
+}

db-scheduler/src/main/java/com/github/kagkarlsson/scheduler/task/schedule/CronSchedule.java

@@ -39,6 +39,8 @@ public class CronSchedule implements Schedule, Serializable {
 
     private static final String DISABLED = "-";
     private static final Logger LOG = LoggerFactory.getLogger(CronSchedule.class);
+
+    private static final long serialVersionUID = 6577185347687546367L;
     private final String pattern;
     private final ZoneId zoneId;
     private transient ExecutionTime cronExecutionTime; // lazily initialized