(server) Impelements export for db range command

project-violet · Apr 6, 2021 · a85e31e · a85e31e
1 parent 0b31077
commit a85e31e
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/hsync/Command.cs b/hsync/Command.cs
@@ -74,6 +74,10 @@ public class Options : IConsoleOption
             Info = "Export database bulk datas for elastic-search to json using id range", Help = "--export-for-es-range")]
         public string[] ExportForESRange;
 
+        [CommandLine("--export-for-db-range", CommandType.ARGUMENTS, ArgumentsCount = 2,
+            Info = "Upload data to server database by user range", Help = "--export-for-db-range")]
+        public string[] ExportForDBRange;
+
         /// <summary>
         /// User Option
         /// </summary>
@@ -178,6 +182,10 @@ public static void Start(string[] arguments)
             {
                 ProcessInitServerPages(null);
             }
+            else if (option.ExportForDBRange != null)
+            {
+                ProcessExportForDBRange(option.ExportForDBRange);
+            }
             else if (option.ExportForES)
             {
                 ProcessExportToES(null);
@@ -558,14 +566,19 @@ static void ProcessCreateDateTimeEstimator(bool low_perf)
 
         static void ProcessInitServer(string[] args)
         {
-            _initServerArticles();
+            _uploadToServerArticlesData();
         }
 
         static void ProcessInitServerPages(string[] args)
         {
             _initServerArticlePages();
         }
 
+        static void ProcessExportForDBRange(string[] args)
+        {
+            _uploadToServerArticlesData(Convert.ToInt32(args[0]), Convert.ToInt32(args[1]));
+        }
+
         static void _initServerArticlePages()
         {
             var db = new SQLiteConnection("data.db");
@@ -611,7 +624,7 @@ static string _ggg(string nu)
             return nu.Replace("\\", "\\\\").Replace("\"", "\\\"").Replace("＼", "\\\\").Replace("＂", "\\\"");
         }
 
-        static void _initServerArticles()
+        static void _uploadToServerArticlesData(int range1 = 0, int range2 = int.MaxValue)
         {
             var db = new SQLiteConnection("data.db");
             var count = db.ExecuteScalar<int>("SELECT COUNT(*) FROM HitomiColumnModel");
@@ -634,6 +647,10 @@ static void _initServerArticles()
 
                     Console.WriteLine($"{i}/{count}");
 
+                    query = query.Where(x => range1 <= x.Id && x.Id <= range2).ToList();
+
+                    if (query.Count == 0) continue;
+
                     var myCommand = conn.CreateCommand();
                     var transaction = conn.BeginTransaction();