a G%@i&@sFddlZddlZddlmZmZmZddlmZGdddeeZdS)N)Plugin PluginOpt UbuntuPlugin) is_executablec @seZdZdZdZdZgdZdZdZdZ dZ d Z d d d d ddZ e ddedde ddedde ddedde ddeddgZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/S)0 CharmedMySQLa The Charmed MySQL plugin is used to collect MySQL configuration and logs from the Charmed MySQL snap package or K8s deployment. It also collects MySQL Router and MySQL Shell configuration and logs where available, journal logs for the snap, and snap info. If the `dumpdbs` option is set to `True`, the plugin will also try and collect the names of the databases that the user has access to. The `mysql` user is used by default, but that can be set using the `dbuser` option. When using the `dumpdbs` option, you must then provide the password for the user using the `dbpass` option or the `MYSQL_PWD` environment variable. z Charmed MySQLZ charmed_mysql)zshow databases;z*select * from mysql.juju_units_operations;zPselect * from performance_schema.replication_group_members order by MEMBER_HOST;z"show global status like "%conne%";z,select * from performance_schema.host_cache;zshow ENGINE InnoDB STATUS;z charmed-mysqlz/var/snap/charmed-mysql/commonz/var/snap/charmed-mysql/currentZkubectlz app.kubernetes.io/name=mysql-k8sz /etc/mysqlz/var/log/mysqlz/etc/mysqlrouterz/var/log/mysqlrouterz/var/log/mysqlsh) MYSQL_CONF MYSQL_LOGSMYSQL_ROUTER_CONFMYSQL_ROUTER_LOGSMYSQL_SHELL_LOGSdbusermysqlz%Username for database dump collection)defaultZval_typeZdescdbpassz%Password for database dump collectiondumpdbsFzGet name of all databases logs_sinceZ48hz9How far back to fetch logs with kubectl --since, K8s onlycCsV|d}|d}dtjvr:|s:|jdtjd}n|sN|jddS||fS)Nr r MYSQL_PWDzWMYSQL_PWD present: Using MYSQL_PWD environment variable, user did not provide password.zHdumpdbs_error: option is set, but username and password are not provided)NN) get_optionosenvironZsosloginfoZwarning)selfdb_userdb_passrD/usr/lib/python3.9/site-packages/sos/report/plugins/charmed_mysql.py_get_db_credentials[s   z CharmedMySQL._get_db_credentialscCsB||jd|jd}|ddkr>tt|dSgS)Nz get pods -A -l z- -o jsonpath='{.items[*].metadata.namespace}'statusroutput)exec_cmdkube_cmdselectorlistsetstripsplit)r namespacesrrr_determine_namespacesns  z"CharmedMySQL._determine_namespacescCs@||jd|d|jd}|ddkr<|dSgS)N -n z get pods -l z( -o jsonpath='{.items[*].metadata.name}'rrr)r r!r"r%r&)r namespacepodsrrr_get_pod_namesws  zCharmedMySQL._get_pod_namescCs^|jd|}||d|jd|d|jg||d|jd|dd}||}|}|jD]f\}}|D]X} tj|d | d |d d |d |d | d|d |d | d | } | | qqv| |d|dg| drZ| \} } | r| s dSd| } d| d}d dd|jD}t}d}|d |}t|ddd}|| Wdn1s0Y|D]} |d|d | d|}| ||d |d |d | d| } | | |d|d | d|d|d|d | }|j|| d!d"qWdn1sP0YdS)#Nr)z get all -l z -o widez describe pods -l z logs -l z --since=z* --all-containers=true --prefix --all-podsr /T)exist_okz cp -c  :z/*/MYSQL_CONF/*.pemz/*/MYSQL_CONF/*.keyr-h 127.0.0.1 -umysql  -ecss|]}|ddVqdS)"z\"N)replace.0queryrrr sz6CharmedMySQL._collect_per_namespace..Z mysql_pwdwutf8)encodingz exec -c z -- mkdir -p z -- sh -lc 'MYSQL_PWD=$(cat z) z "z "' && rm -rf z_dbs.txt)Zsuggest_filename)r!add_cmd_outputr"r,Zget_cmd_output_path conf_pathsitemsrmakedirsr add_forbidden_pathrrjoin mysql_queriestempfileZTemporaryDirectoryopenwrite)rr*rr!Z mysql_contr+Zdump_files_pathnamepathZpodZcopy_cmdrroptssql_cmdZqueriesZtmpdirZpwd_fileZpwd_pathfZ mkdir_cmdZ queries_cmdrrr_collect_per_namespaces         *   z#CharmedMySQL._collect_per_namespacecGs"dd|D}tjj|g|RS)NcSsg|]}|tjjqSr)lstriprrHsep)r7prrr z0CharmedMySQL._join_conf_path..)rrHrB)rbasepartsZstripped_partsrrr_join_conf_pathszCharmedMySQL._join_conf_pathcCs,|d}|}|D]}|||qdS)Nr)rr(rL)rrr'r*rrr _process_k8ss zCharmedMySQL._process_k8sc s|||j|jdd||j|jddg|||j|jd||j|jd||j|jd||j|jd||j|jdg|d|d |d r | \}}|r|sdSd |i}d |}d |d|jfdd|j D|ddS)Nrz*.pemz*.keyrr r r zsnap.charmed-mysql.*zsnap info charmed-mysqlrrr1r2r3csg|]}d|dqS)z ''rr6rJrrrPrQz.CharmedMySQL._process_snap..)env) rArTsnap_path_currentr>Z add_copy_specsnap_path_commonZ add_journalr=rrrC)rrrZ mysql_envrIrrWr _process_snaps^      zCharmedMySQL._process_snapcCs*||jrdSt|j|jo(t|S)NT) is_installed snap_packagerr!sysrootboolr(rrrr check_enableds   zCharmedMySQL.check_enabledcCs.||jr|t|j|jr*|dS)N)r\r]r[rr!r^rUr`rrrsetups zCharmedMySQL.setupN)__name__ __module__ __qualname____doc__Z short_descZ plugin_namerCr]rZrYr!r"r>rstrr_Z option_listrr(r,rLrTrUr[rarbrrrrrsR   N< r) rrDZsos.report.pluginsrrrZ sos.utilitiesrrrrrr s