netdata/netdata

// SPDX-License-Identifier: GPL-3.0-or-later

package hddtemp

import (
    "github.com/netdata/netdata/go/go.d.plugin/pkg/socket"
)

func newHddTempConn(conf Config) hddtempConn {
    return &hddtempClient{conn: socket.New(socket.Config{
        Address:        conf.Address,
        ConnectTimeout: conf.Timeout.Duration(),
        ReadTimeout:    conf.Timeout.Duration(),
        WriteTimeout:   conf.Timeout.Duration(),
    })}
}

type hddtempClient struct {
    conn socket.Client
}

func (c *hddtempClient) connect() error {
    return c.conn.Connect()
}

func (c *hddtempClient) disconnect() {
    _ = c.conn.Disconnect()
}

func (c *hddtempClient) queryHddTemp() (string, error) {
    var i int
    var s string
    err := c.conn.Command("", func(bytes []byte) bool {
        if i++; i > 1 {
            return false
        }
        s = string(bytes)
        return true
    })
    if err != nil {
        return "", err
    }
    return s, nil
}