pcm-coordinator/internal/logic/monitoring/schedulesituationlogic.go

package monitoring

import (
	"context"
	"gitlink.org.cn/JointCloud/pcm-coordinator/internal/svc"
	"gitlink.org.cn/JointCloud/pcm-coordinator/internal/types"
	"strings"

	"github.com/zeromicro/go-zero/core/logx"
)

type ScheduleSituationLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewScheduleSituationLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ScheduleSituationLogic {
	return &ScheduleSituationLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *ScheduleSituationLogic) ScheduleSituation() (resp *types.ScheduleSituationResp, err error) {
	resp = &types.ScheduleSituationResp{}
	// node region
	tx := l.svcCtx.DbEngin.Raw("SELECT c.id, c.name, tdi.id AS category, count(DISTINCT ta.id)+count(DISTINCT tc.id)+COUNT(DISTINCT th.id)+COUNT(tv.id) as value FROM t_cluster c LEFT JOIN t_dict_item tdi ON c.region_dict = tdi.id left JOIN task_ai ta ON ta.cluster_id = c.id left JOIN task_ai_asynchronous tas ON tas.cluster_id = c.id  left JOIN task_cloud tc ON tc.cluster_id = c.id left JOIN task_hpc th ON th.cluster_id = c.id left JOIN task_vm tv ON tv.cluster_id = c.id WHERE tc.deleted_at IS NULL GROUP BY c.id").Scan(&resp.Nodes)
	if tx.Error != nil {
		return nil, tx.Error
	}

	// hpc
	var hpcLinks []string
	tx = l.svcCtx.DbEngin.Raw("SELECT distinct GROUP_CONCAT( distinct cluster_id SEPARATOR ',') as cluster_ids FROM task_hpc WHERE deleted_at IS NULL GROUP BY task_id HAVING COUNT(*) > 1;").Scan(&hpcLinks)
	if tx.Error != nil {
		return nil, tx.Error
	}
	LinksHandler(hpcLinks, resp)
	// cloud
	var cloudLinks []string
	tx = l.svcCtx.DbEngin.Raw("SELECT distinct  GROUP_CONCAT(distinct cluster_id SEPARATOR ',') as cluster_ids FROM task_cloud  WHERE deleted_at IS NULL GROUP BY task_id HAVING COUNT(*) > 1;").Scan(&cloudLinks)
	if tx.Error != nil {
		return nil, tx.Error
	}
	LinksHandler(cloudLinks, resp)
	// ai
	var aiLinks []string
	tx = l.svcCtx.DbEngin.Raw("SELECT distinct GROUP_CONCAT(distinct cluster_id SEPARATOR ',') as cluster_ids FROM task_ai WHERE deleted_at IS NULL GROUP BY task_id HAVING COUNT(*) > 1;").Scan(&aiLinks)
	if tx.Error != nil {
		return nil, tx.Error
	}
	LinksHandler(aiLinks, resp)
	var aiAsynchronousLinks []string
	tx = l.svcCtx.DbEngin.Raw("SELECT distinct GROUP_CONCAT(distinct cluster_id SEPARATOR ',') as cluster_ids FROM task_ai_asynchronous WHERE deleted_at IS NULL GROUP BY task_id HAVING COUNT(*) > 1;").Scan(&aiLinks)
	if tx.Error != nil {
		return nil, tx.Error
	}
	LinksHandler(aiAsynchronousLinks, resp)
	// vm
	var vmLinks []string
	tx = l.svcCtx.DbEngin.Raw("SELECT distinct GROUP_CONCAT(distinct cluster_id SEPARATOR ',') as cluster_ids FROM task_vm WHERE deleted_at IS NULL GROUP BY task_id HAVING COUNT(*) > 1;").Scan(&vmLinks)
	if tx.Error != nil {
		return nil, tx.Error
	}
	LinksHandler(vmLinks, resp)

	// categories
	tx = l.svcCtx.DbEngin.Raw("select tdi.item_text as name from t_dict_item tdi,t_dict td where td.dict_code = 'cluster_region_dict' and tdi.dict_id = td.id").Scan(&resp.Categories)
	if tx.Error != nil {
		return nil, tx.Error
	}
	return resp, nil
}

func LinksHandler(sources []string, resp *types.ScheduleSituationResp) {
	for _, source := range sources {
		links := strings.Split(source, ",")

		for i := 1; i < len(links); i++ {
			if links[i] != links[i-1] {
				resp.Links = append(resp.Links, types.Link{Source: links[i], Target: links[i-1]})
			}
		}
	}

}