mirror of
				https://github.com/ClusterCockpit/cc-metric-collector.git
				synced 2025-11-04 02:35:07 +01:00 
			
		
		
		
	Add samples for collectors, sinks and receivers
This commit is contained in:
		
							
								
								
									
										81
									
								
								collectors/sampleMetric.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										81
									
								
								collectors/sampleMetric.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,81 @@
 | 
			
		||||
package collectors
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	cclog "github.com/ClusterCockpit/cc-metric-collector/internal/ccLogger"
 | 
			
		||||
	lp "github.com/ClusterCockpit/cc-metric-collector/internal/ccMetric"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// These are the fields we read from the JSON configuration
 | 
			
		||||
type SampleCollectorConfig struct {
 | 
			
		||||
	Interval string `json:"interval"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// This contains all variables we need during execution and the variables
 | 
			
		||||
// defined by metricCollector (name, init, ...)
 | 
			
		||||
type SampleCollector struct {
 | 
			
		||||
	metricCollector
 | 
			
		||||
	config SampleTimerCollectorConfig // the configuration structure
 | 
			
		||||
	meta   map[string]string          // default meta information
 | 
			
		||||
	tags   map[string]string          // default tags
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *SampleCollector) Init(config json.RawMessage) error {
 | 
			
		||||
	var err error = nil
 | 
			
		||||
	// Always set the name early in Init() to use it in cclog.Component* functions
 | 
			
		||||
	m.name = "InternalCollector"
 | 
			
		||||
	// This is for later use, also call it early
 | 
			
		||||
	m.setup()
 | 
			
		||||
	// Define meta information sent with each metric
 | 
			
		||||
	// (Can also be dynamic or this is the basic set with extension through AddMeta())
 | 
			
		||||
	m.meta = map[string]string{"source": m.name, "group": "SAMPLE"}
 | 
			
		||||
	// Define tags sent with each metric
 | 
			
		||||
	// The 'type' tag is always needed, it defines the granulatity of the metric
 | 
			
		||||
	// node -> whole system
 | 
			
		||||
	// socket -> CPU socket (requires socket ID as 'type-id' tag)
 | 
			
		||||
	// cpu -> single CPU hardware thread (requires cpu ID as 'type-id' tag)
 | 
			
		||||
	m.tags = map[string]string{"type": "node"}
 | 
			
		||||
	// Read in the JSON configuration
 | 
			
		||||
	if len(config) > 0 {
 | 
			
		||||
		err = json.Unmarshal(config, &m.config)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			cclog.ComponentError(m.name, "Error reading config:", err.Error())
 | 
			
		||||
			return err
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Set up everything that the collector requires during the Read() execution
 | 
			
		||||
	// Check files required, test execution of some commands, create data structure
 | 
			
		||||
	// for all topological entities (sockets, NUMA domains, ...)
 | 
			
		||||
	// Return some useful error message in case of any failures
 | 
			
		||||
 | 
			
		||||
	// Set this flag only if everything is initialized properly, all required files exist, ...
 | 
			
		||||
	m.init = true
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *SampleCollector) Read(interval time.Duration, output chan lp.CCMetric) {
 | 
			
		||||
	// Create a sample metric
 | 
			
		||||
	timestamp := time.Now()
 | 
			
		||||
 | 
			
		||||
	value := 1.0
 | 
			
		||||
	// If you want to measure something for a specific amout of time, use interval
 | 
			
		||||
	// start := readState()
 | 
			
		||||
	// time.Sleep(interval)
 | 
			
		||||
	// stop := readState()
 | 
			
		||||
	// value = (stop - start) / interval.Seconds()
 | 
			
		||||
 | 
			
		||||
	y, err := lp.New("sample_metric", m.tags, m.meta, map[string]interface{}{"value": value}, timestamp)
 | 
			
		||||
	if err == nil {
 | 
			
		||||
		// Send it to output channel
 | 
			
		||||
		output <- y
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (m *SampleCollector) Close() {
 | 
			
		||||
	// Unset flag
 | 
			
		||||
	m.init = false
 | 
			
		||||
}
 | 
			
		||||
		Reference in New Issue
	
	Block a user