cluster.go

// 维护集群配置信息
package handler

import (
	"github.com/andy-zhangtao/qcloud_api/v1/cvm"
	"github.com/andy-zhangtao/qcloud_api/v1/public"
	"net/http"
	"encoding/json"
	"strconv"
	"github.com/andy-zhangtao/DDog/const"
	"github.com/andy-zhangtao/DDog/server/metadata"
	"errors"
	"github.com/andy-zhangtao/DDog/server/mongo"
	"github.com/andy-zhangtao/DDog/server/tool"
)

type Cluster struct {
	SecretId  string `json:"secret_id"`
	SecretKey string `json:"secret_key"`
	Region    string `json:"region"`
}

// saveClusterInfo 查询集群信息并持久化到etcd
// 如果选择持久化，则会覆盖旧值
func (this Cluster) SaveClusterInfo(save bool) (*cvm.ClusterInfo, error) {

	c := cvm.Cluster{
		Pub: public.Public{
			Region:   this.Region,
			SecretId: this.SecretId,
		},
		SecretKey: this.SecretKey,
	}

	c.SetDebug(_const.DEBUG)

	cinfo, err := c.QueryClusters()
	if err != nil {
		return nil, err
	}

	if save {
		//data, err := json.Marshal(cinfo.Data.Clusters)
		//if err != nil {
		//	return nil, err
		//}

		for _, c := range cinfo.Data.Clusters {
			mongo.DeleteCluster(c.ClusterId)

			err = mongo.SaveCluster(c)
			if err != nil {
				return nil, err
			}
		}
		//err = etcd.Put(_const.CloudEtcdRootPath+"/"+c.Pub.Region+_const.CloudEtcdClusterInfo, string(data))

	}

	return cinfo, nil
}

// QueryClusterInfo 更新集群列表信息
// 列出此账户下所有的集群信息
func QueryClusterInfo(w http.ResponseWriter, r *http.Request) {
	//data, err := ioutil.ReadAll(r.Body)
	//if err != nil {
	//	tool.ReturnError(w, err)
	//	return
	//}

	region := r.URL.Query().Get("region")
	if region == "" {
		tool.ReturnError(w, errors.New("Region Can not be empty!"))
		return
	}

	isSave := r.URL.Query().Get("save")
	if isSave == "" || isSave != "true" {
		isSave = "false"
	}

	save, err := strconv.ParseBool(isSave)
	if err != nil {
		save = false
	}

	md, err := metadata.GetMetaData(region)
	if err != nil {
		tool.ReturnError(w, err)
		return
	}

	ch := Cluster{
		SecretId:  md.Sid,
		SecretKey: md.Skey,
		Region:    region,
	}

	cinfo, err := ch.SaveClusterInfo(save)
	if err != nil {
		tool.ReturnError(w, err)
		return
	}

	data, err := json.Marshal(cinfo)
	if err != nil {
		tool.ReturnError(w, err)
		return
	}

	w.Header().Set("Content-Type", "application/json")
	w.Write(data)
}

func GetClusterInfo(w http.ResponseWriter, r *http.Request) {
	region := r.URL.Query().Get("region")
	if region == "" {
		tool.ReturnError(w, errors.New(_const.RegionNotFound))
		return
	}

	cs, err := mongo.GetClusterByRegion(_const.ReverseRegionMap[region])
	if err != nil {
		tool.ReturnError(w, err)
		return
	}

	w.Header().Set("Content-Type", "application/json")

	data, err := json.Marshal(cs)
	w.Write(data)
}