From c0c4c32eb8b66892a2acd1b86fde7ac824d16a0e Mon Sep 17 00:00:00 2001
From: ioito <qu_xuan@icloud.com>
Date: Mon, 27 Mar 2023 22:28:45 +0800
Subject: [PATCH] fix(cloudmon): support h3c metric pull

---
 pkg/cloudmon/providerdriver/h3c.go | 106 +++++++++++++++++++++++++++++
 1 file changed, 106 insertions(+)
 create mode 100644 pkg/cloudmon/providerdriver/h3c.go

diff --git a/pkg/cloudmon/providerdriver/h3c.go b/pkg/cloudmon/providerdriver/h3c.go
new file mode 100644
index 0000000000..6525cb043e
--- /dev/null
+++ b/pkg/cloudmon/providerdriver/h3c.go
@@ -0,0 +1,106 @@
+// Copyright 2019 Yunion
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package providerdriver
+
+import (
+	"context"
+	"strconv"
+	"sync"
+	"time"
+
+	"yunion.io/x/cloudmux/pkg/cloudprovider"
+	api "yunion.io/x/onecloud/pkg/apis/compute"
+	"yunion.io/x/onecloud/pkg/util/influxdb"
+)
+
+type H3CCollect struct {
+	SBaseCollectDriver
+}
+
+func (self *H3CCollect) GetProvider() string {
+	return api.CLOUD_PROVIDER_H3C
+}
+
+func init() {
+	Register(&H3CCollect{})
+}
+
+func (self *H3CCollect) IsSupportMetrics() bool {
+	return true
+}
+
+func (self *H3CCollect) CollectServerMetrics(ctx context.Context, manager api.CloudproviderDetails, provider cloudprovider.ICloudProvider, res map[string]api.ServerDetails, start, end time.Time) error {
+	metrics := []influxdb.SMetricData{}
+	var mu sync.Mutex
+	opts := &cloudprovider.MetricListOptions{
+		ResourceType: cloudprovider.METRIC_RESOURCE_TYPE_SERVER,
+	}
+	data := []cloudprovider.MetricValues{}
+	for opts.StartTime.Before(opts.EndTime) {
+		part, err := provider.GetMetrics(opts)
+		if err != nil {
+			continue
+		}
+		opts.StartTime = opts.StartTime.Add(time.Minute)
+		time.Sleep(time.Minute)
+		data = append(data, part...)
+	}
+	for _, value := range data {
+		vm, ok := res[value.Id]
+		if !ok {
+			continue
+		}
+		tags := []influxdb.SKeyValue{}
+		for k, v := range vm.GetMetricTags() {
+			tags = append(tags, influxdb.SKeyValue{
+				Key:   k,
+				Value: v,
+			})
+		}
+		pairs := []influxdb.SKeyValue{}
+		for k, v := range vm.GetMetricPairs() {
+			pairs = append(pairs, influxdb.SKeyValue{
+				Key:   k,
+				Value: v,
+			})
+		}
+		for _, v := range value.Values {
+			metric := influxdb.SMetricData{
+				Name:      value.MetricType.Name(),
+				Timestamp: v.Timestamp,
+				Tags:      []influxdb.SKeyValue{},
+				Metrics: []influxdb.SKeyValue{
+					{
+						Key:   value.MetricType.Key(),
+						Value: strconv.FormatFloat(v.Value, 'E', -1, 64),
+					},
+				},
+			}
+			for k, v := range v.Tags {
+				metric.Tags = append(metric.Tags, influxdb.SKeyValue{
+					Key:   k,
+					Value: v,
+				})
+			}
+			metric.Metrics = append(metric.Metrics, pairs...)
+			metric.Tags = append(metric.Tags, tags...)
+			mu.Lock()
+			metrics = append(metrics, metric)
+			mu.Unlock()
+		}
+	}
+
+	return self.sendMetrics(ctx, manager, "server", len(res), metrics)
+}