mirror of
https://github.com/ClusterCockpit/cc-metric-collector.git
synced 2024-12-25 23:19:06 +01:00
Refactor: Embed Init() into New() function
This commit is contained in:
parent
3598aed090
commit
2f36375470
@ -14,7 +14,6 @@ The configuration file for the receivers is a list of configurations. The `type`
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
||||||
## Type `nats`
|
## Type `nats`
|
||||||
|
|
||||||
```json
|
```json
|
||||||
@ -30,7 +29,6 @@ The `nats` receiver subscribes to the topic `database` and listens on `address`
|
|||||||
|
|
||||||
# Contributing own receivers
|
# Contributing own receivers
|
||||||
A receiver contains a few functions and is derived from the type `Receiver` (in `metricReceiver.go`):
|
A receiver contains a few functions and is derived from the type `Receiver` (in `metricReceiver.go`):
|
||||||
* `Init(name string, config json.RawMessage) error`
|
|
||||||
* `Start() error`
|
* `Start() error`
|
||||||
* `Close()`
|
* `Close()`
|
||||||
* `Name() string`
|
* `Name() string`
|
||||||
|
@ -1,9 +1,6 @@
|
|||||||
package receivers
|
package receivers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
// "time"
|
|
||||||
"encoding/json"
|
|
||||||
|
|
||||||
lp "github.com/ClusterCockpit/cc-metric-collector/internal/ccMetric"
|
lp "github.com/ClusterCockpit/cc-metric-collector/internal/ccMetric"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -25,7 +22,6 @@ type receiver struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Receiver interface {
|
type Receiver interface {
|
||||||
Init(name string, config json.RawMessage) error
|
|
||||||
Start()
|
Start()
|
||||||
Close()
|
Close()
|
||||||
Name() string
|
Name() string
|
||||||
|
@ -32,38 +32,6 @@ var DefaultTime = func() time.Time {
|
|||||||
return time.Unix(42, 0)
|
return time.Unix(42, 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *NatsReceiver) Init(name string, config json.RawMessage) error {
|
|
||||||
r.name = fmt.Sprintf("NatsReceiver(%s)", name)
|
|
||||||
r.config.Addr = nats.DefaultURL
|
|
||||||
r.config.Port = "4222"
|
|
||||||
if len(config) > 0 {
|
|
||||||
err := json.Unmarshal(config, &r.config)
|
|
||||||
if err != nil {
|
|
||||||
cclog.ComponentError(r.name, "Error reading config:", err.Error())
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if len(r.config.Addr) == 0 ||
|
|
||||||
len(r.config.Port) == 0 ||
|
|
||||||
len(r.config.Subject) == 0 {
|
|
||||||
return errors.New("not all configuration variables set required by NatsReceiver")
|
|
||||||
}
|
|
||||||
r.meta = map[string]string{"source": r.name}
|
|
||||||
uri := fmt.Sprintf("%s:%s", r.config.Addr, r.config.Port)
|
|
||||||
cclog.ComponentDebug(r.name, "INIT", uri, "Subject", r.config.Subject)
|
|
||||||
nc, err := nats.Connect(uri)
|
|
||||||
if err == nil {
|
|
||||||
r.nc = nc
|
|
||||||
} else {
|
|
||||||
r.nc = nil
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
r.handler = influx.NewMetricHandler()
|
|
||||||
r.parser = influx.NewParser(r.handler)
|
|
||||||
r.parser.SetTimeFunc(DefaultTime)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *NatsReceiver) Start() {
|
func (r *NatsReceiver) Start() {
|
||||||
cclog.ComponentDebug(r.name, "START")
|
cclog.ComponentDebug(r.name, "START")
|
||||||
r.nc.Subscribe(r.config.Subject, r._NatsReceive)
|
r.nc.Subscribe(r.config.Subject, r._NatsReceive)
|
||||||
@ -93,6 +61,32 @@ func (r *NatsReceiver) Close() {
|
|||||||
|
|
||||||
func NewNatsReceiver(name string, config json.RawMessage) (Receiver, error) {
|
func NewNatsReceiver(name string, config json.RawMessage) (Receiver, error) {
|
||||||
r := new(NatsReceiver)
|
r := new(NatsReceiver)
|
||||||
err := r.Init(name, config)
|
r.name = fmt.Sprintf("NatsReceiver(%s)", name)
|
||||||
return r, err
|
r.config.Addr = nats.DefaultURL
|
||||||
|
r.config.Port = "4222"
|
||||||
|
if len(config) > 0 {
|
||||||
|
err := json.Unmarshal(config, &r.config)
|
||||||
|
if err != nil {
|
||||||
|
cclog.ComponentError(r.name, "Error reading config:", err.Error())
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(r.config.Addr) == 0 ||
|
||||||
|
len(r.config.Port) == 0 ||
|
||||||
|
len(r.config.Subject) == 0 {
|
||||||
|
return nil, errors.New("not all configuration variables set required by NatsReceiver")
|
||||||
|
}
|
||||||
|
r.meta = map[string]string{"source": r.name}
|
||||||
|
uri := fmt.Sprintf("%s:%s", r.config.Addr, r.config.Port)
|
||||||
|
cclog.ComponentDebug(r.name, "NewNatsReceiver", uri, "Subject", r.config.Subject)
|
||||||
|
if nc, err := nats.Connect(uri); err == nil {
|
||||||
|
r.nc = nc
|
||||||
|
} else {
|
||||||
|
r.nc = nil
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
r.handler = influx.NewMetricHandler()
|
||||||
|
r.parser = influx.NewParser(r.handler)
|
||||||
|
r.parser.SetTimeFunc(DefaultTime)
|
||||||
|
return r, nil
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user