pcm-coordinator/api/internal/mqs/kq/ScheduleCloud.go

package kq

import (
	"bytes"
	"context"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/pkg/scheduler"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/svc"
	"gitlink.org.cn/jcce-pcm/pcm-coordinator/model"
	"io"
	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
	"k8s.io/apimachinery/pkg/runtime"
	syaml "k8s.io/apimachinery/pkg/runtime/serializer/yaml"
	kyaml "k8s.io/apimachinery/pkg/util/yaml"
)

/*
*
Listening to the payment flow status change notification message queue
*/
type ScheduleCloudMq struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewScheduleCloudMq(ctx context.Context, svcCtx *svc.ServiceContext) *ScheduleCloudMq {
	return &ScheduleCloudMq{
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func UnMarshalK8sStruct(yamlString string, taskId int64) model.Cloud {
	var cloud model.Cloud
	d := kyaml.NewYAMLOrJSONDecoder(bytes.NewBufferString(yamlString), 4096)
	var err error
	for {
		var rawObj runtime.RawExtension
		err = d.Decode(&rawObj)
		if err == io.EOF {
			break
		}
		if err != nil {
		}
		obj := &unstructured.Unstructured{}
		syaml.NewDecodingSerializer(unstructured.UnstructuredJSONScheme).Decode(rawObj.Raw, nil, obj)
		if err != nil {
		}

		unstructuredMap, err := runtime.DefaultUnstructuredConverter.ToUnstructured(obj)
		if err != nil {
		}

		unstructureObj := &unstructured.Unstructured{Object: unstructuredMap}
		cloud = model.Cloud{
			TaskId:     taskId,
			ApiVersion: unstructureObj.GetAPIVersion(),
			Name:       unstructureObj.GetName(),
			Kind:       unstructureObj.GetKind(),
			Namespace:  unstructureObj.GetNamespace(),
			Status:     "Saved",
		}
	}
	return cloud
}

func (l *ScheduleCloudMq) Consume(_, val string) error {
	// 接受消息, 根据标签筛选过滤
	cloudScheduler := scheduler.NewCloudScheduler()
	scheduler, err := scheduler.NewScheduler(cloudScheduler, val)
	if err != nil {
		return err
	}
	scheduler.MatchLabels(l.svcCtx.DbEngin)

	// 存储数据
	err = scheduler.SaveToDb(l.svcCtx.DbEngin)
	if err != nil {
		return err
	}
	return nil
}