cloudfoundry.go

// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
// or more contributor license agreements. Licensed under the Elastic License;
// you may not use this file except in compliance with the Elastic License.

//go:build !aix
// +build !aix

package cloudfoundry

import (
	"fmt"

	"github.com/elastic/beats/v7/libbeat/logp"
	"github.com/elastic/beats/v7/metricbeat/mb"
	cfcommon "github.com/elastic/beats/v7/x-pack/libbeat/common/cloudfoundry"
)

// ModuleName is the name of this module.
const ModuleName = "cloudfoundry"

func init() {
	if err := mb.Registry.AddModule(ModuleName, newModule); err != nil {
		panic(err)
	}
}

type Module interface {
	mb.Module
	RunCounterReporter(mb.PushReporterV2)
	RunContainerReporter(mb.PushReporterV2)
	RunValueReporter(mb.PushReporterV2)
}

func newModule(base mb.BaseModule) (mb.Module, error) {
	factory := func(cfg *cfcommon.Config, name string, log *logp.Logger) CloudfoundryHub {
		return &HubAdapter{cfcommon.NewHub(cfg, name, log)}
	}
	return NewModuleWithHubFactory(base, factory)
}

type hubFactory func(cfg *cfcommon.Config, name string, log *logp.Logger) CloudfoundryHub

// NewModuleWithHubFactory initializes a module with a hub created with a hub factory
func NewModuleWithHubFactory(base mb.BaseModule, hubFactory hubFactory) (mb.Module, error) {
	var cfg cfcommon.Config
	if err := base.UnpackConfig(&cfg); err != nil {
		return nil, err
	}

	log := logp.NewLogger("cloudfoundry")
	hub := hubFactory(&cfg, "metricbeat", log)

	switch cfg.Version {
	case cfcommon.ConsumerVersionV1:
		return newModuleV1(base, hub, log)
	case cfcommon.ConsumerVersionV2:
		return newModuleV2(base, hub, log)
	default:
		return nil, fmt.Errorf("not supported consumer version: %s", cfg.Version)
	}
}