共计 4048 个字符,预计需要花费 11 分钟才能阅读完成。
作者:洪斌
爱可生南区负责人兼技术服务总监,MySQL ACE,擅长数据库架构规划、故障诊断、性能优化分析,实践经验丰富,帮助各行业客户解决 MySQL 技术问题,为金融、运营商、互联网等行业客户提供 MySQL 整体解决方案。
本文来源:转载自公众号 - 玩转 MySQL
* 爱可生开源社区出品,原创内容未经授权不得随意使用,转载请联系小编并注明来源。
MySQL Shell 现在不只是用来部署 InnoDB cluster 和 Replica set 了,在 8.0.17 之后有了自定义扩展能力,DBA 可以用 javascript 和 python 编写扩展了,把一些常用的 SQL 脚本封装成自己的工具箱,甚至可以围绕它构建 DBA 的 DevOps 作业平台。
为 MySQL Shell 添加扩展并不复杂,现有两种扩展方式,一种是 report,这种主要用来做各种查询,可以使用内建的两个命令调用自定义的 report,另一种是 plugin,可以定义任意功能函数。
- show 就是普通一次性输出,已经内建了好多常用指标,参见脑图。
- watch 类似 top 方式,持续输出信息,写些简单的监控脚本方便多了。
自定义扩展
将自定义的 js 或 py 脚本放在 ~/.mysqlsh/plugin 和~/.mysqlsh/init.d 目录下,建议是放在 plugin 目录下,目录可按功能类别命名,目录中必须有 init.py 或 init.js 文件用来初始化扩展,代码中可约定顶级目录名作为全局对象,二级目录作为成员对象。
hongbin@MBP ~/.m/plugins> tree
.
└── ext
└── table
└── init.py
2 directories, 1 file
下面是一段示例代码
# init.py
# -------
# 演示注册 report 和 plugin 两种方式
# 定义一个查询函数,获取没有主键或唯一索引表
def report_table_without_pk(session):
query = '''SELECT tables.table_schema , tables.table_name
FROM information_schema.tables
LEFT JOIN (
SELECT table_schema , table_name
FROM information_schema.statistics
GROUP BY table_schema, table_name, index_name HAVING
SUM(case when non_unique = 0 and nullable != 'YES' then 1 else 0 end) = count(*) ) puks
ON tables.table_schema = puks.table_schema and tables.table_name = puks.table_name
WHERE puks.table_name is null
AND tables.table_type = 'BASE TABLE' AND Engine="InnoDB";'''
result = session.run_sql(query)
report = []
if (result.has_data()):
report = [result.get_column_names()]
for row in result.fetch_all():
report.append(list(row))
# 注册为 report,需要返回字典类型
return {"report": report}
# 功能同上,这里为演示以 Pluginf 方式重新定义函数,两者 report 和 plugin 差异主要在于输出方式
def plugin_table_without_pk(session):
query = '''SELECT tables.table_schema , tables.table_name
FROM information_schema.tables
LEFT JOIN (
SELECT table_schema , table_name
FROM information_schema.statistics
GROUP BY table_schema, table_name, index_name HAVING
SUM(case when non_unique = 0 and nullable != 'YES' then 1 else 0 end) = count(*) ) puks
ON tables.table_schema = puks.table_schema and tables.table_name = puks.table_name
WHERE puks.table_name is null
AND tables.table_type = 'BASE TABLE' AND Engine="InnoDB";'''
result = session.run_sql(query)
shell.dump_rows(result)
return
# 注册一个 function,用来给表添加主键字段,变更类操作通常以 function
def _add_pk(table, columns, session=None):
query = 'ALTER TABLE %s ADD PRIMARY KEY (%s)' % (table, columns)
if session is None:
session = shell.get_session()
if session is None:
print("No session specified. Either pass a session object to this"
"function or connect the shell to a database")
return
# session = shell.get_session()
result = session.run_sql(query)
# 这里注册上面定义的 report 函数,report 名称,返回格式类型,函数名,描述
shell.register_report("table_without_pk", "list", report_table_without_pk, {"brief": "Lists the table without primary key."})
# 这里定义全局对象,可以看做一个命名空间,ext 是默认社区扩展插件的对象名,或者其他自定义对象名称
if 'ext' in globals():
global_obj = ext
else:
# Otherwise register new global object named 'ext'
global_obj = shell.create_extension_object()
shell.register_global("ext", global_obj,
{"brief": "MySQL Shell extension plugins."})
# 可以按类别在全局对象下添加子对象
try:
plugin_obj = global_obj.table
except IndexError:
plugin_obj = shell.create_extension_object()
shell.add_extension_object_member(global_obj,
"table",
plugin_obj,
{"brief": "Utility object for table operations."})
# 添加功能函数到自定义插件对象中,父级对象名,调用函数名,定义的函数,描述,函数接受的参数名 / 类型 / 是否必选 / 描述
try:
shell.add_extension_object_member(plugin_obj,
"add_pk",
_add_pk,
{"brief":
"Add a primary key to the table",
"parameters": [
{
"name": "table",
"type": "string",
"required": True,
"brief": "table name."
},
{
"name": "columns",
"type": "string",
"required": True,
"brief": "column name."
},
{
"name": "session",
"type": "object",
"class": "Session",
"required": False,# 若不想单独传 session 参数,可以在函数中获取当前会话对象
"brief": "The session to be used on the operation."
}
]
})
except Exception as e:
shell.log("ERROR", "Failed to register ext.table.add_pk ({0}).".format(str(e).rstrip()))
# 添加 plugin_table_without_pk
try:
shell.add_extension_object_member(plugin_obj,
"get_without_pk",
plugin_table_without_pk,
{"brief":
"Lists the table without primary key.",
})
except Exception as e:
shell.log("ERROR", "Failed to register ext.table.get_without_pk ({0}).".format(str(e).rstrip()))
使用方法
登录 mysqlsh 后自动搜索并初始化插件,指定 –log-level 参数时可记录详细调试信息到 ~/.mysqlsh/mysqlsh.log,如果加载失败,可以查看日志分析原因。
对于 report,使用 show 或 watch 命令指定 report 名称输出结果。
对于 plugin,直接调用函数。
这里抛砖引玉简单介绍了一下 MySQL Shell 插件扩展功能,更多有趣的用法等你来发现。
参考:
https://dev.mysql.com/doc/mys… https://github.com/lefred/mysqlshell-plugins