src/02_Data/Adapters/Data.Adapter.MySql/MySqlSchemaProvider.cs

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

using System.Collections.Generic;
using System.Data;
using System.Linq;
using Dapper;
using Mkh.Data.Abstractions.Schema;
using MySqlConnector;

namespace Mkh.Data.Adapter.MySql;

/// <summary>
/// MySql数据库架构提供器
/// </summary>
public class MySqlSchemaProvider : ISchemaProvider
{
    private readonly IDbConnection _con;
    public MySqlSchemaProvider(IDbConnection con)
    {
        _con = con;
    }

    public MySqlSchemaProvider(string connectionString)
    {
        _con = new MySqlConnection(connectionString);
    }

    public bool IsExistsDatabase(string database)
    {
        var sql = $"SELECT 1 FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = '{database}' LIMIT 1;";
        return _con.ExecuteScalar<int>(sql) > 0;
    }

    public bool IsExistsTable(string database, string table)
    {
        var sql = $"SELECT 1 FROM information_schema.TABLES WHERE TABLE_NAME ='{table}' and TABLE_SCHEMA='{database}';";
        return _con.ExecuteScalar<int>(sql) > 0;
    }

    public List<DatabaseSchema> GetDatabases()
    {
        var sql = "SELECT SCHEMA_NAME AS `Name` FROM INFORMATION_SCHEMA.SCHEMATA;";
        var databases = _con.Query<DatabaseSchema>(sql).ToList();
        foreach (var database in databases)
        {
            database.Tables = GetTables(database.Name);
        }
        return databases;
    }

    public DatabaseSchema GetDatabase(string database)
    {
        var sql = $"SELECT SCHEMA_NAME AS `Name` FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = '{database}';";

        var db = _con.QueryFirstOrDefault<DatabaseSchema>(sql);
        if (db != null)
        {
            db.Tables = GetTables(database);
        }

        return db;
    }