#3025 Table truncate model

plugins/org.jkiss.dbeaver.model/src/org/jkiss/dbeaver/model/exec/DBCStatistics.java

@@ -25,6 +25,7 @@ import java.util.*;
  */
 public class DBCStatistics {
 
+    private final long startTime;
     private long rowsUpdated;
     private long rowsFetched;
     private long executeTime;
@@ -34,8 +35,8 @@ public class DBCStatistics {
     private Map<String, Object> infoMap;
     private List<String> messages;
 
-    public DBCStatistics()
-    {
+    public DBCStatistics() {
+        this.startTime = System.currentTimeMillis();
     }
 
     public long getRowsUpdated()
@@ -78,6 +79,11 @@ public class DBCStatistics {
         this.executeTime += executeTime;
     }
 
+    public void addExecuteTime()
+    {
+        this.executeTime += (System.currentTimeMillis() - startTime);
+    }
+
     public long getFetchTime()
     {
         return fetchTime;

plugins/org.jkiss.dbeaver.model/src/org/jkiss/dbeaver/model/impl/jdbc/struct/JDBCTable.java

@@ -47,10 +47,11 @@ import org.jkiss.utils.ArrayUtils;
  */
 public abstract class JDBCTable<DATASOURCE extends DBPDataSource, CONTAINER extends DBSObjectContainer>
     extends AbstractTable<DATASOURCE, CONTAINER>
-    implements DBSDataManipulator, DBSDataCleaner, DBPSaveableObject
+    implements DBSDataManipulator, DBPSaveableObject
 {
     private static final Log log = Log.getLog(JDBCTable.class);
-    public static final String DEFAULT_TABLE_ALIAS = "x";
+
+    private static final String DEFAULT_TABLE_ALIAS = "x";
     public static final int DEFAULT_READ_FETCH_SIZE = 10000;
 
     private boolean persisted;
@@ -507,28 +508,33 @@ public abstract class JDBCTable<DATASOURCE extends DBPDataSource, CONTAINER exte
     ////////////////////////////////////////////////////////////////////
     // Truncate
 
-    @Override
-    public boolean canTruncateData() {
-        return false;
-    }
-
     @Override
     public DBCStatistics truncateData(DBCSession session, DBCExecutionSource source) throws DBCException {
-        DBCStatistics statistics = new DBCStatistics();
-        DBRProgressMonitor monitor = session.getProgressMonitor();
-
-        monitor.subTask("Truncate data");
-        long startTime = System.currentTimeMillis();
-        try (DBCStatement dbStat = session.prepareStatement(
-            DBCStatementType.QUERY,
-            getTruncateTableQuery(),
-            false, false, false))
-        {
-            dbStat.setStatementSource(source);
-            dbStat.executeStatement();
+        if (!isTruncateSupported()) {
+            try (ExecuteBatch batch = deleteData(session, new DBSAttributeBase[0], source)) {
+                batch.add(new Object[0]);
+                return batch.execute(session);
+            }
+        } else {
+            DBCStatistics statistics = new DBCStatistics();
+            DBRProgressMonitor monitor = session.getProgressMonitor();
+
+            monitor.subTask("Truncate data");
+            try (DBCStatement dbStat = session.prepareStatement(
+                DBCStatementType.QUERY,
+                getTruncateTableQuery(),
+                false, false, false)) {
+                dbStat.setStatementSource(source);
+                dbStat.executeStatement();
+            }
+            statistics.addStatementsCount();
+            statistics.addExecuteTime();
+            return statistics;
         }
-        statistics.addExecuteTime(System.currentTimeMillis() - startTime);
-        return statistics;
+    }
+
+    protected boolean isTruncateSupported() {
+        return false;
     }
 
     protected String getTruncateTableQuery() {

plugins/org.jkiss.dbeaver.model/src/org/jkiss/dbeaver/model/struct/DBSDataCleaner.java

-/*
- * DBeaver - Universal Database Manager
- * Copyright (C) 2010-2017 Serge Rider (serge@jkiss.org)
- *
- * 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 org.jkiss.dbeaver.model.struct;
-
-import org.jkiss.code.NotNull;
-import org.jkiss.dbeaver.model.exec.DBCException;
-import org.jkiss.dbeaver.model.exec.DBCExecutionSource;
-import org.jkiss.dbeaver.model.exec.DBCSession;
-import org.jkiss.dbeaver.model.exec.DBCStatistics;
-
-/**
- * Data cleaner.
- * Supports data truncate.
- */
-public interface DBSDataCleaner extends DBSDataContainer {
-
-    /**
-     * Returns true if this data container can be truncated.
-     */
-    boolean canTruncateData();
-
-    /**
-     * Truncates data.
-     * Warning: all data will be deleted and this operation is not undoable.
-     * Use with attention.
-     */
-    @NotNull
-    DBCStatistics truncateData(
-        @NotNull DBCSession session,
-        @NotNull DBCExecutionSource source)
-        throws DBCException;
-
-}

plugins/org.jkiss.dbeaver.model/src/org/jkiss/dbeaver/model/struct/DBSDataManipulator.java

@@ -37,6 +37,7 @@ public interface DBSDataManipulator extends DBSDataContainer {
     int DATA_INSERT         = 1 << 16;
     int DATA_UPDATE         = 1 << 17;
     int DATA_DELETE         = 1 << 18;
+    int DATA_TRUNCATE       = 1 << 19;
 
     interface ExecuteBatch extends AutoCloseable {
         void add(@NotNull Object[] attributeValues) throws DBCException;
@@ -74,4 +75,10 @@ public interface DBSDataManipulator extends DBSDataContainer {
         @NotNull DBCExecutionSource source)
         throws DBCException;
 
+    @NotNull
+    DBCStatistics truncateData(
+        @NotNull DBCSession session,
+        @NotNull DBCExecutionSource source)
+        throws DBCException;
+
 }