前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >绘制Crushmap分布图-Python

绘制Crushmap分布图-Python

作者头像
用户1260683
发布2018-01-31 11:56:15
1.2K0
发布2018-01-31 11:56:15
举报
文章被收录于专栏:Ceph对象存储方案

原理

使用命令ceph report --format=json > crush.json导出json格式数据文件,之后使用pydot和graphviz实现绘图。

使用说明

安装依赖

代码语言:javascript
复制
pip install pydot
pip install graphviz

脚本代码

代码语言:javascript
复制
# -*- coding: utf-8 -*-
import pydot
from graphviz import Digraph
import json
import sys

class build_crushmap_graphviz():
    """
    1. 使用命令ceph report --format=json > crush.json导出数据文件
    2. 每种类型bucket一个颜色,不够自己去color_list里面添加,支持最多10级结构
    3. 生成的文件默认问png格式,文件保存在当前目录的crushmap.png
    """
    def __init__(self):
        self.graph = pydot.Dot('ceph_crushmap', graph_type='digraph')
        self.dot = Digraph(comment='CrushMap', node_attr={'shape': 'record', 'height': '.1'})
        self.dot.graph_attr['size'] = '4096,2160'
        self.dot.graph_attr['resolution'] = '100'
        self.dot.graph_attr['bb'] = '0,0,4,8'
        self.dot.format = 'png'
        self.color_list = ["maroon", "pink", "khaki", "orange", "purple", "yellow", "cyan", "beige", "red"]
        self.save_name = "crushmap"

    def build(self, crushmap_file):
        try:
            with open(crushmap_file) as data_file:
                data = json.load(data_file)
            for i in range(len(data['crushmap']['devices'])):
                self.dot.node(str(data['crushmap']['devices'][i]['id']),
                              'device: ' + data['crushmap']['devices'][i]['name'],
                              {'style': 'filled', 'fillcolor': 'green'})
            tmp_list = []
            color_dict = {}
            for i in range(len(data['crushmap']['buckets'])):
                if data['crushmap']['buckets'][i]['type_name'] in tmp_list:
                    color_ = color_dict[data['crushmap']['buckets'][i]['type_name']]
                else:
                    tmp_list.append(data['crushmap']['buckets'][i]['type_name'])
                    color_ = self.color_list.pop()
                    color_dict[data['crushmap']['buckets'][i]['type_name']] = color_
                self.dot.node(str(data['crushmap']['buckets'][i]['id']),
                              data['crushmap']['buckets'][i]['type_name'] + ': ' + data['crushmap']['buckets'][i]['name'],
                              {'style': 'filled', 'fillcolor': color_})
            edges_list = []
            for i in range(len(data['crushmap']['buckets'])):
                for j in range(len(data['crushmap']['buckets'][i]['items'])):
                    self.dot.edge(str(data['crushmap']['buckets'][i]['id']),
                                  str(data['crushmap']['buckets'][i]['items'][j]['id']))
                    edges_list.append(
                        str(data['crushmap']['buckets'][i]['id']) + str(data['crushmap']['buckets'][i]['items'][j]['id']))
            self.dot.render(self.save_name)
            print "Sucessful, File = {}.{}".format(self.save_name,self.dot.format)
        except:
            print "Faild!"

if __name__ == '__main__':
    file_path = sys.argv[1]
    crush_make = build_crushmap_graphviz()
    crush_make.build(file_path)

用例

保存上面的脚本为build_crushmap.py,执行以下命令,成功会生成crushmap.png

代码语言:javascript
复制
user@demo$ python build_crushmap.py crush2.json
Sucessful, File = crushmap.png
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2017-05-26,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 Ceph对象存储方案 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 原理
    • 使用说明
      • 安装依赖
      • 脚本代码
      • 用例
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档