src/main/java/org/embulk/output/jdbc/setter/ColumnSetterVisitor.java in embulk-output-jdbc-0.8.2 vs src/main/java/org/embulk/output/jdbc/setter/ColumnSetterVisitor.java in embulk-output-jdbc-0.8.3

@@ -1,33 +1,34 @@
 package org.embulk.output.jdbc.setter;
 
 import java.io.IOException;
 import java.sql.SQLException;
+
+import org.embulk.output.jdbc.Record;
 import org.embulk.spi.Column;
 import org.embulk.spi.ColumnVisitor;
-import org.embulk.spi.PageReader;
 
 public class ColumnSetterVisitor
         implements ColumnVisitor
 {
-    private final PageReader pageReader;
+    private final Record record;
     private final ColumnSetter setter;
 
-    public ColumnSetterVisitor(PageReader pageReader, ColumnSetter setter)
+    public ColumnSetterVisitor(Record record, ColumnSetter setter)
     {
-        this.pageReader = pageReader;
+        this.record = record;
         this.setter = setter;
     }
 
     @Override
     public void booleanColumn(Column column)
     {
         try {
-            if (pageReader.isNull(column)) {
+            if (record.isNull(column)) {
                 setter.nullValue();
             } else {
-                setter.booleanValue(pageReader.getBoolean(column));
+                setter.booleanValue(record.getBoolean(column));
             }
         } catch (IOException | SQLException ex) {
             // TODO exception class
             throw new RuntimeException(ex);
         }
@@ -35,14 +36,14 @@
 
     @Override
     public void longColumn(Column column)
     {
         try {
-            if (pageReader.isNull(column)) {
+            if (record.isNull(column)) {
                 setter.nullValue();
             } else {
-                setter.longValue(pageReader.getLong(column));
+                setter.longValue(record.getLong(column));
             }
         } catch (IOException | SQLException ex) {
             // TODO exception class
             throw new RuntimeException(ex);
         }
@@ -50,14 +51,14 @@
 
     @Override
     public void doubleColumn(Column column)
     {
         try {
-            if (pageReader.isNull(column)) {
+            if (record.isNull(column)) {
                 setter.nullValue();
             } else {
-                setter.doubleValue(pageReader.getDouble(column));
+                setter.doubleValue(record.getDouble(column));
             }
         } catch (IOException | SQLException ex) {
             // TODO exception class
             throw new RuntimeException(ex);
         }
@@ -65,14 +66,14 @@
 
     @Override
     public void stringColumn(Column column)
     {
         try {
-            if (pageReader.isNull(column)) {
+            if (record.isNull(column)) {
                 setter.nullValue();
             } else {
-                setter.stringValue(pageReader.getString(column));
+                setter.stringValue(record.getString(column));
             }
         } catch (IOException | SQLException ex) {
             // TODO exception class
             throw new RuntimeException(ex);
         }
@@ -80,14 +81,14 @@
 
     @Override
     public void jsonColumn(Column column)
     {
         try {
-            if (pageReader.isNull(column)) {
+            if (record.isNull(column)) {
                 setter.nullValue();
             } else {
-                setter.jsonValue(pageReader.getJson(column));
+                setter.jsonValue(record.getJson(column));
             }
         } catch (IOException | SQLException ex) {
             // TODO exception class
             throw new RuntimeException(ex);
         }
@@ -95,13 +96,13 @@
 
     @Override
     public void timestampColumn(Column column)
     {
         try {
-            if (pageReader.isNull(column)) {
+            if (record.isNull(column)) {
                 setter.nullValue();
             } else {
-                setter.timestampValue(pageReader.getTimestamp(column));
+                setter.timestampValue(record.getTimestamp(column));
             }
         } catch (IOException | SQLException ex) {
             // TODO exception class
             throw new RuntimeException(ex);
         }