Zabbix

/*

** Zabbix

** Copyright (C) 2001-2023 Zabbix SIA

**

** This program is free software; you can redistribute it and/or modify

** it under the terms of the GNU General Public License as published by

** the Free Software Foundation; either version 2 of the License, or

** (at your option) any later version.

**

** This program is distributed in the hope that it will be useful,

** but WITHOUT ANY WARRANTY; without even the implied warranty of

** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

** GNU General Public License for more details.

**

** You should have received a copy of the GNU General Public License

** along with this program; if not, write to the Free Software

** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

**/

​

package ceph

​

import (

    "encoding/json"

    "fmt"

    "strconv"

​

    "git.zabbix.com/ap/plugin-support/zbxerr"

)

​

type node struct {

    ID       int64   `json:"id"`

    Class    string  `json:"device_class"`

    Name     string  `json:"name"`

    Type     string  `json:"type"`

    Children []int64 `json:"children"`

}

​

type crushTree struct {

    Nodes []node `json:"nodes"`

}

​

type osdEntity struct {

    OSDName string `json:"{#OSDNAME}"`

    Class   string `json:"{#CLASS}"`

    Host    string `json:"{#HOST}"`

}

​

// osdDiscoveryHandler returns list of OSDs in LLD format.

func osdDiscoveryHandler(data map[command][]byte) (interface{}, error) {

    var (

        tree crushTree

        host *node

    )

​

    lld := make([]osdEntity, 0)

​

    err := json.Unmarshal(data[cmdOSDCrushTree], &tree)

    if err != nil {

        return nil, zbxerr.ErrorCannotUnmarshalJSON.Wrap(err)

    }

​

    for i, n := range tree.Nodes {

        switch n.Type {

        case "host":

            host = &tree.Nodes[i]

            continue

​

        case "osd":

            if host == nil {

                continue

            }

​

            for _, cid := range host.Children {

                if n.ID == cid {

                    lld = append(lld, osdEntity{

                        OSDName: strconv.FormatInt(n.ID, 10),

                        Class:   n.Class,

                        Host:    host.Name,

                    })

                }

            }

​

        default:

            continue

        }

    }

​

    jsonLLD, err := json.Marshal(lld)

    if err != nil {

        return nil, zbxerr.ErrorCannotMarshalJSON.Wrap(err)

    }