litestream 0.7.1 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 52282949824c8413d029df186f9e88b77bdfe68f771bf58988750601751fd8db
-  data.tar.gz: 634a7f0fa817086938880c33d68191944d9c2bb3606c655b2995f99640849461
+  metadata.gz: 62ef078cc395bd64c09f4c1d6168f7c93b2e793da798bd2a33f8c6b049dca0cc
+  data.tar.gz: bbde0e9b3f6cbe2bed3e2283338aa2628253dcdc183bb53e142c850b673ea0d3
 SHA512:
-  metadata.gz: c612148ced5d1e3464f38955d377c5fc53942404547b48a34da462fc0f88854c6d39411c84502fdda17a3101cca95303b1db7b82cf10bbf12ccf237c5d5e9aeb
-  data.tar.gz: d6376138a401421a7949ce3ed957131fc3cb7e25ecda269508156c6bab39b1199b0238ad0cc2595d79dd562cab30112eb520022e5a63117f8e95d38b53606a0b
+  metadata.gz: a18f350d6cc794e7d328fdbcb2bcdac975d1c7064f0912838abc1e01ebfdf6dd83ec26cb31b35b28b1bda54945cce29c368f8fb7af6c206e416a57e989db5b6d
+  data.tar.gz: 9002f0e3a2b3cd5fe64f64baab6ac60f592f12d37cb10f831b5f91e76c83b6b5e9272dd619c09012fa36f1d56d56a529d5276d8d3dae356070b75e98b40d4afa

data/lib/litestream/commands.rb

@@ -1,4 +1,5 @@
 require_relative "upstream"
+require "logfmt"
 
 module Litestream
   module Commands
@@ -101,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.1"
+  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.1
+  version: 0.8.0
 platform: ruby
 authors:
 - Stephen Margheim