DatabaseConnectionFactory.cs

Go to the documentation of this file.

using System;
using System.Data.Common;
using System.Data.SqlClient;
using Microsoft.Data.Sqlite;
using MySql.Data.MySqlClient;
using Tgstation.Server.Host.Configuration;

namespace Tgstation.Server.Host.Database
{
    sealed class DatabaseConnectionFactory : IDatabaseConnectionFactory
    {
        public DbConnection CreateConnection(string connectionString, DatabaseType databaseType)
        {
            if (connectionString == null)
                throw new ArgumentNullException(nameof(connectionString));

            switch (databaseType)
            {
                case DatabaseType.MariaDB:
                case DatabaseType.MySql:
                    return new MySqlConnection
                    {
                        ConnectionString = connectionString
                    };
                case DatabaseType.SqlServer:
                    return new SqlConnection
                    {
                        ConnectionString = connectionString
                    };
                case DatabaseType.Sqlite:
                    return new SqliteConnection
                    {
                        ConnectionString = connectionString
                    };
                default:
                    throw new ArgumentOutOfRangeException(nameof(databaseType), databaseType, "Invalid DatabaseType!");
            }
        }
    }
}