mysql查多张表的总数据量

原创

mob649e8161738c 2024-01-19 10:31:20 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e8161738c的原创作品，请联系作者获取转载授权，否则将追究法律责任

实现“mysql查多张表的总数据量”

流程图

flowchart TD
    start[开始]
    input[输入要查询的表名]
    connect[连接数据库]
    query[执行查询语句]
    output[输出结果]
    end[结束]
    
    start --> input --> connect --> query --> output --> end

整体流程

输入要查询的表名
连接数据库
执行查询语句
输出结果

详细步骤及代码

1. 输入要查询的表名

首先，我们需要输入要查询的表名。这个可以通过用户输入或者在代码中直接定义。假设我们要查询的表名为表A、表B和表C。

2. 连接数据库

连接数据库是查询数据的前提，我们需要使用数据库的连接信息来建立与数据库的连接。以下是连接数据库的代码。

import mysql.connector

# 连接数据库
def connect_db():
    conn = mysql.connector.connect(
        host="localhost",  # 数据库主机地址
        user="username",   # 数据库用户名
        passwd="password", # 数据库密码
        database="database" # 数据库名
    )

    return conn

其中，host是数据库主机地址，可以是IP地址或者域名；user和passwd是数据库的用户名和密码；database是要连接的数据库的名称。

3. 执行查询语句

接下来，我们需要编写查询语句来获取多张表的总数据量。在MySQL中，可以使用UNION ALL将多个SELECT COUNT(*)语句合并，然后使用SUM()函数计算总和。以下是查询语句的代码。

# 执行查询语句
def execute_query(conn, table_names):
    cursor = conn.cursor()

    query = "SELECT SUM(t.total_count) FROM ("

    for i, table_name in enumerate(table_names):
        if i > 0:
            query += " UNION ALL "

        query += f"SELECT COUNT(*) AS total_count FROM {table_name}"

    query += ") t"

    cursor.execute(query)
    result = cursor.fetchone()[0]

    cursor.close()
    conn.close()

    return result

在这段代码中，table_names是一个包含要查询的表名的列表。我们使用循环遍历表名，将每个表的COUNT(*)结果添加到查询语句中。最后，使用SUM()函数计算总和。cursor.execute(query)用于执行查询语句，cursor.fetchone()[0]用于获取查询结果的第一行第一列的值。

4. 输出结果

最后，我们可以将查询结果输出，以便用户查看。以下是输出结果的代码。

# 输出结果
def output_result(result):
    print(f"多张表的总数据量为：{result}")

完整代码示例

import mysql.connector

# 连接数据库
def connect_db():
    conn = mysql.connector.connect(
        host="localhost",  # 数据库主机地址
        user="username",   # 数据库用户名
        passwd="password", # 数据库密码
        database="database" # 数据库名
    )

    return conn

# 执行查询语句
def execute_query(conn, table_names):
    cursor = conn.cursor()

    query = "SELECT SUM(t.total_count) FROM ("

    for i, table_name in enumerate(table_names):
        if i > 0:
            query += " UNION ALL "

        query += f"SELECT COUNT(*) AS total_count FROM {table_name}"

    query += ") t"

    cursor.execute(query)
    result = cursor.fetchone()[0]

    cursor.close()
    conn.close()

    return result

# 输出结果
def output_result(result):
    print(f"多张表的总数据量为：{result}")

# 主函数
def main():
    table_names = ["tableA", "tableB", "tableC"]
    conn = connect_db()
    result = execute_query(conn, table_names)
    output_result(result)

if __name__ == "__main__":
    main()