Author: reschke Date: Thu Jan 11 10:36:37 2018 New Revision: 1820861 URL: http://svn.apache.org/viewvc?rev=1820861&view=rev Log: OAK-7029: RDBDocumentStore.getStats() implementation
Modified: jackrabbit/oak/trunk/oak-store-document/src/main/java/org/apache/jackrabbit/oak/plugins/document/rdb/RDBDocumentStoreDB.java Modified: jackrabbit/oak/trunk/oak-store-document/src/main/java/org/apache/jackrabbit/oak/plugins/document/rdb/RDBDocumentStoreDB.java URL: http://svn.apache.org/viewvc/jackrabbit/oak/trunk/oak-store-document/src/main/java/org/apache/jackrabbit/oak/plugins/document/rdb/RDBDocumentStoreDB.java?rev=1820861&r1=1820860&r2=1820861&view=diff ============================================================================== --- jackrabbit/oak/trunk/oak-store-document/src/main/java/org/apache/jackrabbit/oak/plugins/document/rdb/RDBDocumentStoreDB.java (original) +++ jackrabbit/oak/trunk/oak-store-document/src/main/java/org/apache/jackrabbit/oak/plugins/document/rdb/RDBDocumentStoreDB.java Thu Jan 11 10:36:37 2018 @@ -674,6 +674,80 @@ public enum RDBDocumentStoreDB { } return result.toString(); } + + private long parseSize(String readable) { + try { + if (readable != null && readable.endsWith(" KB")) { + return 1024 * Long.parseLong(readable.substring(0, readable.length() - 3)); + } else { + return -1; + } + } catch (NumberFormatException ex) { + return -1; + } + } + + @Override + public Map<String, String> getAdditionalStatistics(RDBConnectionHandler ch, String catalog, String tableName) { + Map<String, String> result = new HashMap<String, String>(); + Connection con = null; + + // table data + try { + con = ch.getROConnection(); + try (PreparedStatement stmt = con.prepareStatement("exec sp_spaceused ?")) { + stmt.setString(1, tableName.toLowerCase(Locale.ENGLISH)); + try (ResultSet rs = stmt.executeQuery()) { + while (rs.next()) { + long treserved = parseSize(rs.getString("reserved")); + long tdata = parseSize(rs.getString("data")); + long tindexSize = parseSize(rs.getString("index_size")); + long tunused = parseSize(rs.getString("unused")); + if (treserved >= 0 && tdata >= 0 && tindexSize >= 0 && tunused >= 0) { + result.put("storageSize", Long.toString(treserved + tdata + tindexSize + tunused)); + result.put("size", Long.toString(treserved + tdata + tunused)); + result.put("totalIndexSize", Long.toString(tindexSize)); + } + String data = extractFields(rs, "rows reserved data index_size unused"); + result.put("_data", data); + } + } + con.commit(); + } + } catch (SQLException ex) { + LOG.debug("while getting diagnostics", ex); + } finally { + ch.closeConnection(con); + } + + // index data + try { + con = ch.getROConnection(); + try (PreparedStatement stmt = con.prepareStatement( + "SELECT i.[name] AS name, SUM(s.[row_count]) as rows, SUM(s.[used_page_count] * 8) as usedKB, SUM(s.[reserved_page_count] * 8) as reservedKB " + + "FROM sys.dm_db_partition_stats AS s " + + "INNER JOIN sys.indexes AS i ON s.[object_id] = i.[object_id] " + + " AND s.[index_id] = i.[index_id] WHERE i.[object_id]=OBJECT_ID(?) " + "GROUP BY i.[name]")) { + stmt.setString(1, tableName.toLowerCase(Locale.ENGLISH)); + try (ResultSet rs = stmt.executeQuery()) { + while (rs.next()) { + String index = rs.getString("name"); + String data = extractFields(rs, "rows usedKB reservedKB"); + result.put("index." + index + "._data", data); + } + } + con.commit(); + } + con.commit(); + } catch (SQLException ex) { + ex.printStackTrace(); + LOG.debug("while getting diagnostics", ex); + } finally { + ch.closeConnection(con); + } + + return result; + } }; private static final Logger LOG = LoggerFactory.getLogger(RDBDocumentStoreDB.class); @@ -836,6 +910,11 @@ public enum RDBDocumentStoreDB { * SHOW TABLE STATUS Syntax</a>, <a href= * "https://dev.mysql.com/doc/refman/5.7/en/show-index.html">13.7.5.22 SHOW * INDEX Syntax</a> + * <li>{@link #MSSQL} - Developer Reference for SQL Server: <a href= + * "https://docs.microsoft.com/en-us/sql/relational-databases/system-dynamic-management-views/sys-dm-db-partition-stats-transact-sql">sys.dm_db_partition_stats + * (Transact-SQL)</a>, <a href= + * "https://docs.microsoft.com/en-us/sql/relational-databases/system-stored-procedures/sp-spaceused-transact-sql">sp_spaceused + * (Transact-SQL)</a> * </ul> */ public String getAdditionalDiagnostics(RDBConnectionHandler ch, String tableName) {