ceph/src/pybind/mgr/dashboard/controllers/crush_rule.py

   1 # -*- coding: utf-8 -*-
   2 from __future__ import absolute_import
   3
   4 from cherrypy import NotFound
   5
   6 from .. import mgr
   7 from ..security import Scope
   8 from ..services.ceph_service import CephService
   9 from . import ApiController, ControllerDoc, Endpoint, EndpointDoc, \
  10     ReadPermission, RESTController, UiApiController
  11
  12 LIST_SCHEMA = {
  13     "rule_id": (int, 'Rule ID'),
  14     "rule_name": (str, 'Rule Name'),
  15     "ruleset": (int, 'RuleSet related to the rule'),
  16     "type": (int, 'Type of Rule'),
  17     "min_size": (int, 'Minimum size of Rule'),
  18     "max_size": (int, 'Maximum size of Rule'),
  19     'steps': ([{str}], 'Steps included in the rule')
  20 }
  21
  22
  23 @ApiController('/crush_rule', Scope.POOL)
  24 @ControllerDoc("Crush Rule Management API", "CrushRule")
  25 class CrushRule(RESTController):
  26     @EndpointDoc("List Crush Rule Configuration",
  27                  responses={200: LIST_SCHEMA})
  28     def list(self):
  29         return mgr.get('osd_map_crush')['rules']
  30
  31     def get(self, name):
  32         rules = mgr.get('osd_map_crush')['rules']
  33         for r in rules:
  34             if r['rule_name'] == name:
  35                 return r
  36         raise NotFound('No such crush rule')
  37
  38     def create(self, name, root, failure_domain, device_class=None):
  39         rule = {
  40             'name': name,
  41             'root': root,
  42             'type': failure_domain,
  43             'class': device_class
  44         }
  45         CephService.send_command('mon', 'osd crush rule create-replicated', **rule)
  46
  47     def delete(self, name):
  48         CephService.send_command('mon', 'osd crush rule rm', name=name)
  49
  50
  51 @UiApiController('/crush_rule', Scope.POOL)
  52 @ControllerDoc("Dashboard UI helper function; not part of the public API", "CrushRuleUi")
  53 class CrushRuleUi(CrushRule):
  54     @Endpoint()
  55     @ReadPermission
  56     def info(self):
  57         '''Used for crush rule creation modal'''
  58         return {
  59             'names': [r['rule_name'] for r in mgr.get('osd_map_crush')['rules']],
  60             'nodes': mgr.get('osd_map_tree')['nodes']
  61         }