Add MariadbDBRecord implementations

Author: psainics

mariadb-plugin/pom.xml

@@ -83,6 +83,11 @@
             <version>RELEASE</version>
             <scope>compile</scope>
         </dependency>
+        <dependency>
+            <groupId>io.cdap.plugin</groupId>
+            <artifactId>mysql-plugin</artifactId>
+            <version>${project.version}</version>
+        </dependency>
     </dependencies>
     <build>
         <plugins>

mariadb-plugin/src/main/java/io/cdap/plugin/mariadb/MariadbDBRecord.java

@@ -0,0 +1,40 @@
+/*
+ * Copyright © 2025 Cask Data, Inc.
+ *
+ * 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.cdap.plugin.mariadb;
+
+import io.cdap.cdap.api.data.format.StructuredRecord;
+import io.cdap.plugin.db.ColumnType;
+import io.cdap.plugin.mysql.MysqlDBRecord;
+import java.util.List;
+
+/**
+ * Writable class for MariaDB Source/Sink.
+ */
+public class MariadbDBRecord extends MysqlDBRecord {
+
+  /**
+   * Used in map-reduce. Do not remove.
+   */
+  @SuppressWarnings("unused")
+  public MariadbDBRecord() {
+    // Required by Hadoop DBRecordReader to create an instance
+  }
+
+  public MariadbDBRecord(StructuredRecord record, List<ColumnType> columnTypes) {
+    super(record, columnTypes);
+  }
+}

mariadb-plugin/src/main/java/io/cdap/plugin/mariadb/MariadbSink.java

@@ -19,10 +19,13 @@
 import io.cdap.cdap.api.annotation.Description;
 import io.cdap.cdap.api.annotation.Name;
 import io.cdap.cdap.api.annotation.Plugin;
+import io.cdap.cdap.api.data.format.StructuredRecord;
 import io.cdap.cdap.etl.api.batch.BatchSink;
+import io.cdap.plugin.db.DBRecord;
 import io.cdap.plugin.db.config.DBSpecificSinkConfig;
 import io.cdap.plugin.db.sink.AbstractDBSink;
 
+import io.cdap.plugin.mysql.MysqlDBRecord;
 import java.util.Map;
 import javax.annotation.Nullable;
 
@@ -45,6 +48,11 @@ public MariadbSink(MariadbSinkConfig mariadbSinkConfig) {
     this.mariadbSinkConfig = mariadbSinkConfig;
   }
 
+  @Override
+  protected DBRecord getDBRecord(StructuredRecord output) {
+    return new MariadbDBRecord(output, columnTypes);
+  }
+
   /**
    * MariaDB Sink Config.
    */

mariadb-plugin/src/main/java/io/cdap/plugin/mariadb/MariadbSource.java

@@ -22,6 +22,7 @@
 import io.cdap.cdap.etl.api.batch.BatchSource;
 import io.cdap.plugin.db.config.DBSpecificSourceConfig;
 import io.cdap.plugin.db.source.AbstractDBSource;
+import org.apache.hadoop.mapreduce.lib.db.DBWritable;
 
 import java.util.List;
 import java.util.Map;
@@ -53,6 +54,11 @@ protected String createConnectionString() {
                          mariadbSourceConfig.host, mariadbSourceConfig.port, mariadbSourceConfig.database);
   }
 
+  @Override
+  protected Class<? extends DBWritable> getDBRecordType() {
+    return MariadbDBRecord.class;
+  }
+
   /**
    * MaraiDB source mariadbSourceConfig.
    */