Add the "getQuery" method to the "Command" interface

Author: nakamura-to

doma-core/src/main/java/org/seasar/doma/jdbc/command/BatchModifyCommand.java

@@ -27,6 +27,11 @@ protected BatchModifyCommand(QUERY query) {
     this.query = query;
   }
 
+  @Override
+  public QUERY getQuery() {
+    return query;
+  }
+
   @Override
   public int[] execute() {
     if (!query.isExecutable()) {

doma-core/src/main/java/org/seasar/doma/jdbc/command/Command.java

@@ -1,6 +1,10 @@
 package org.seasar.doma.jdbc.command;
 
+import org.seasar.doma.jdbc.query.Query;
+
 public interface Command<RESULT> {
 
+  Query getQuery();
+
   RESULT execute();
 }

doma-core/src/main/java/org/seasar/doma/jdbc/command/CreateCommand.java

@@ -15,6 +15,11 @@ public CreateCommand(CreateQuery<RESULT> query) {
     this.query = query;
   }
 
+  @Override
+  public CreateQuery<RESULT> getQuery() {
+    return query;
+  }
+
   @Override
   public RESULT execute() {
     Connection connection = JdbcUtil.getConnection(query.getConfig().getDataSource());

doma-core/src/main/java/org/seasar/doma/jdbc/command/ModifyCommand.java

@@ -28,6 +28,11 @@ protected ModifyCommand(QUERY query) {
     this.sql = query.getSql();
   }
 
+  @Override
+  public QUERY getQuery() {
+    return query;
+  }
+
   @Override
   public Integer execute() {
     if (!query.isExecutable()) {

doma-core/src/main/java/org/seasar/doma/jdbc/command/ModuleCommand.java

@@ -26,6 +26,11 @@ protected ModuleCommand(QUERY query) {
     this.sql = query.getSql();
   }
 
+  @Override
+  public QUERY getQuery() {
+    return query;
+  }
+
   @Override
   public RESULT execute() {
     Connection connection = JdbcUtil.getConnection(query.getConfig().getDataSource());

doma-core/src/main/java/org/seasar/doma/jdbc/command/ScriptCommand.java

@@ -29,6 +29,11 @@ public ScriptCommand(ScriptQuery query) {
     this.query = query;
   }
 
+  @Override
+  public ScriptQuery getQuery() {
+    return query;
+  }
+
   @Override
   public Void execute() {
     Config config = query.getConfig();

doma-core/src/main/java/org/seasar/doma/jdbc/command/SelectCommand.java

@@ -34,6 +34,11 @@ public SelectCommand(SelectQuery query, ResultSetHandler<RESULT> resultSetHandle
     this.resultSetHandler = resultSetHandler;
   }
 
+  @Override
+  public SelectQuery getQuery() {
+    return query;
+  }
+
   @Override
   public RESULT execute() {
     Supplier<RESULT> supplier = null;

doma-core/src/main/java/org/seasar/doma/jdbc/command/SqlProcessorCommand.java

@@ -20,6 +20,11 @@ public SqlProcessorCommand(
     this.handler = handler;
   }
 
+  @Override
+  public SqlProcessorQuery getQuery() {
+    return query;
+  }
+
   @Override
   public RESULT execute() {
     return handler.apply(query.getConfig(), query.getSql());

doma-core/src/test/java/org/seasar/doma/jdbc/command/SelectCommandTest.java

@@ -240,4 +240,12 @@ public void testExecute_resultStream() throws Exception {
     assertTrue(runtimeConfig.dataSource.connection.preparedStatement.isClosed());
     assertTrue(runtimeConfig.dataSource.connection.isClosed());
   }
+
+  @Test
+  public void getGetQuery() {
+    SqlFileSelectQuery query = new SqlFileSelectQuery();
+    SelectCommand command =
+        new SelectCommand<>(query, new EntityResultListHandler<>(_Emp.getSingletonInternal()));
+    assertEquals(query, command.getQuery());
+  }
 }