Better log message for connection failures

abock · Aug 14, 2020 · adc8ce2 · adc8ce2
1 parent 372c084
commit adc8ce2
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 5 deletions.
diff --git a/Goodbye.WordPress.CommandLineTool/Program.cs b/Goodbye.WordPress.CommandLineTool/Program.cs
@@ -86,7 +86,7 @@ static int ShowHelp(string? error = null, Exception? exception = null)
                 Console.ForegroundColor = ConsoleColor.DarkRed;
                 Console.Error.WriteLine($"error: {error ?? exception?.Message}");
                 Console.ResetColor();
-                if (verbosity > 0 && exception != null)
+                if (verbosity > 1 && exception != null)
                     Console.Error.WriteLine(exception);
                 Console.Error.WriteLine();
             }
@@ -151,9 +151,26 @@ static async Task<int> Main(string[] args)
                 return ShowHelp(exception: e);
             }
 
-            exporter = await exporter
-                .WithPostReader(postReader)
-                .ExportAsync();
+            try
+            {
+                exporter = await exporter
+                    .WithPostReader(postReader)
+                    .ExportAsync();
+            }
+            catch (ConnectionFailedException e)
+            {
+                Log.Fatal(
+                    verbosity > 1 ? e : null,
+                    "Cannot connect to WordPress: {Message}",
+                    e.Message);
+            }
+            catch (Exception e)
+            {
+                Log.Fatal(
+                    e,
+                    "Failed to perform export process for an unknown reason: {Message}",
+                    e.Message);
+            }
 
             return 0;
         }

diff --git a/Goodbye.WordPress/ConnectionFailedException.cs b/Goodbye.WordPress/ConnectionFailedException.cs
@@ -0,0 +1,18 @@
+// https://github.com/abock/goodbye-wordpress
+// Copyright 2020 Aaron Bockover.
+// Licensed under the MIT License.
+
+using System;
+
+namespace Goodbye.WordPress
+{
+    public sealed class ConnectionFailedException : Exception
+    {
+        internal ConnectionFailedException(Exception innerException)
+            : base(
+                innerException.Message,
+                innerException)
+        {
+        }
+    }
+}
diff --git a/Goodbye.WordPress/MysqlPostReader.cs b/Goodbye.WordPress/MysqlPostReader.cs
@@ -45,7 +45,15 @@ public async IAsyncEnumerable<Post> ReadPostsAsync(
             string? originalPermalinkStructure = null;
 
             using var connection = new MySqlConnection(ConnectionString);
-            await connection.OpenAsync(cancellationToken);
+
+            try
+            {
+                await connection.OpenAsync(cancellationToken);
+            }
+            catch (Exception e)
+            {
+                throw new ConnectionFailedException(e);
+            }
 
             bool dbSupported = false;
             int dbVersion = 0;