litestream 0.7.2 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6bec723f525384051201a0e4e664537089399390d497214acb38a8a192cab3a6
-  data.tar.gz: 32ea621d3eecb12350d4c41099e7ad0a015c76a775f5bac941a00b77c0c7b078
+  metadata.gz: 62ef078cc395bd64c09f4c1d6168f7c93b2e793da798bd2a33f8c6b049dca0cc
+  data.tar.gz: bbde0e9b3f6cbe2bed3e2283338aa2628253dcdc183bb53e142c850b673ea0d3
 SHA512:
-  metadata.gz: 25e84db2832b5c217cd11457b12a257084ef00cfe757dfa6a9e08c0888d3bfc574de2fe6dc092b9714060486c7aeb52fee30fae27cb24478f0675f58c305f4b6
-  data.tar.gz: 456834bf7110b15204de85c89b8bb8085b24cd8453e04e7ac82e04744e48b9f6d36a15ac73b6221390c258f8fdcd52b8f8647f933b2eeca385fe28d67fe4de49
+  metadata.gz: a18f350d6cc794e7d328fdbcb2bcdac975d1c7064f0912838abc1e01ebfdf6dd83ec26cb31b35b28b1bda54945cce29c368f8fb7af6c206e416a57e989db5b6d
+  data.tar.gz: 9002f0e3a2b3cd5fe64f64baab6ac60f592f12d37cb10f831b5f91e76c83b6b5e9272dd619c09012fa36f1d56d56a529d5276d8d3dae356070b75e98b40d4afa

data/lib/litestream/commands.rb

@@ -102,16 +102,29 @@ module Litestream
         }.merge(argv)
         restore(database, async: false, **args)
 
-        restored_tables_count = `sqlite3 #{backup} "select count(*) from sqlite_schema where type='table';"`.chomp.to_i
-        restored_size = File.size(backup)
-        original_tables_count = `sqlite3 #{database} "select count(*) from sqlite_schema where type='table';"`.chomp.to_i
-        original_size = File.size(database)
+        restored_schema = `sqlite3 #{backup} "select name, type from sqlite_schema;"`.chomp.split("\n")
+        restored_data = restored_schema.map { _1.split("|") }.group_by(&:last)
+        restored_rows_count = restored_data["table"]&.sum { |tbl, _| `sqlite3 #{backup} "select count(*) from #{tbl};"`.chomp.to_i }
+
+        original_schema = `sqlite3 #{database} "select name, type from sqlite_schema;"`.chomp.split("\n")
+        original_data = original_schema.map { _1.split("|") }.group_by(&:last)
+        original_rows_count = original_data["table"]&.sum { |tbl, _| `sqlite3 #{database} "select count(*) from #{tbl};"`.chomp.to_i }
 
         Dir.glob(backup + "*").each { |file| File.delete(file) }
 
         {
-          "size" => {"original" => original_size, "restored" => restored_size},
-          "tables" => {"original" => original_tables_count, "restored" => restored_tables_count}
+          "original" => {
+            "path" => database,
+            "tables" => original_data["table"]&.size,
+            "indexes" => original_data["index"]&.size,
+            "rows" => original_rows_count
+          },
+          "restored" => {
+            "path" => backup,
+            "tables" => restored_data["table"]&.size,
+            "indexes" => restored_data["index"]&.size,
+            "rows" => restored_rows_count
+          }
         }
       end
 

data/lib/litestream/version.rb

@@ -1,3 +1,3 @@
 module Litestream
-  VERSION = "0.7.2"
+  VERSION = "0.8.0"
 end

data/lib/tasks/litestream_tasks.rake

@@ -96,15 +96,20 @@ namespace :litestream do
 
     puts <<~TXT if result
 
-      size
-        original          #{result["size"]["original"]}
-        restored          #{result["size"]["restored"]}
-        delta             #{result["size"]["original"] - result["size"]["restored"]}
-
       tables
-        original          #{result["tables"]["original"]}
-        restored          #{result["tables"]["restored"]}
-        delta             #{result["tables"]["original"] - result["tables"]["restored"]}
+        original          #{result["original"]["tables"]}
+        restored          #{result["restored"]["tables"]}
+        delta             #{result["original"]["tables"] - result["restored"]["tables"]}
+
+      indexes
+        original          #{result["original"]["indexes"]}
+        restored          #{result["restored"]["indexes"]}
+        delta             #{result["original"]["indexes"] - result["restored"]["indexes"]}
+
+      rows
+        original          #{result["original"]["rows"]}
+        restored          #{result["restored"]["rows"]}
+        delta             #{result["original"]["rows"] - result["restored"]["rows"]}
     TXT
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: litestream
 version: !ruby/object:Gem::Version
-  version: 0.7.2
+  version: 0.8.0
 platform: ruby
 authors:
 - Stephen Margheim