Refactor: using sqlite with the C# parser

Author: matecsad

plugins/csharp/parser/src_csharp/Program.cs

@@ -86,28 +86,9 @@ static int Main(string[] args)
       //Converting the connectionstring into entiy framwork style connectionstring
       string csharpConnectionString = transformConnectionString();
 
-      var builder = new DbContextOptionsBuilder<CsharpDbContext>();
-      switch (_dbSystem)
-      {
-        case "sqlite":
-          builder.UseSqlite(csharpConnectionString);
-          break;
-        default:
-          builder.UseNpgsql(csharpConnectionString);
-          break;
-      }
-      var options = builder.Options;
-      CsharpDbContext _context;
-      switch (_dbSystem)
-      {
-        case "sqlite":
-          _context = new SqliteDbContext(options);
-          break;
-        default:
-          _context = new PostgresDbContext(options);
-          break;
-      }
-      _context.Database.Migrate();
+      var builderOptions = createDbContextOptions(csharpConnectionString);
+      CsharpDbContext context = createDbContext(builderOptions);
+      context.Database.Migrate();
 
       List<string> allFiles = new List<string>();
       foreach (var p in _rootDir)
@@ -156,41 +137,14 @@ static int Main(string[] args)
     private static async Task<int> ParalellRun(string csharpConnectionString, int threadNum,
         List<SyntaxTree> trees, CSharpCompilation compilation)
     {
-      var builder = new DbContextOptionsBuilder<CsharpDbContext>();
-      switch (_dbSystem)
-      {
-        case "sqlite":
-          builder.UseSqlite(csharpConnectionString);
-          break;
-        default:
-          builder.UseNpgsql(csharpConnectionString);
-          break;
-      }
-      var options = builder.Options;
-      CsharpDbContext dbContext;
-      switch (_dbSystem)
-      {
-        case "sqlite":
-          dbContext = new SqliteDbContext(options);
-          break;
-        default:
-          dbContext = new PostgresDbContext(options);
-          break;
-      }
+      
+      var builderOptions = createDbContextOptions(csharpConnectionString);
+      CsharpDbContext dbContext = createDbContext(builderOptions);
       var contextList = new List<CsharpDbContext>();
       contextList.Add(dbContext);
       for (int i = 1; i < threadNum; i++)
       {
-        CsharpDbContext dbContextInstance;
-        switch (_dbSystem)
-        {
-          case "sqlite":
-            dbContextInstance = new SqliteDbContext(options);
-            break;
-          default:
-            dbContextInstance = new PostgresDbContext(options);
-            break;
-        }
+        CsharpDbContext dbContextInstance = createDbContext(builderOptions);
         contextList.Add(dbContextInstance);
       }
 
@@ -340,5 +294,36 @@ private static string transformConnectionString()
 
       return csharpConnectionString;
     }
+
+    private static DbContextOptions<CsharpDbContext> createDbContextOptions(string connectionString)
+    {
+      var builder = new DbContextOptionsBuilder<CsharpDbContext>();
+      switch (_dbSystem)
+      {
+        case "sqlite":
+          builder.UseSqlite(connectionString);
+          break;
+        default:
+          builder.UseNpgsql(connectionString);
+          break;
+      } 
+
+      return builder.Options;
+    }
+    private static CsharpDbContext createDbContext(DbContextOptions<CsharpDbContext> options)
+    {
+      CsharpDbContext context;
+      switch (_dbSystem)
+      {
+        case "sqlite":
+          context = new SqliteDbContext(options);
+          break;
+        default:
+          context = new PostgresDbContext(options);
+          break;
+      } 
+
+      return context;
+    }
   }
 }