grafana.go

package prometheus_grafana

import (
	"fmt"
	"io/ioutil"
	"os/exec"
	"strings"
	"log"
)

type GrafanaSpec struct {
	NamespaceName  string `json:"namespaceName"`  //namespace 명 (defult: monitoring)
	ImgVersion     string `json:"imgVersion"`     //prometheus image version (defalt: latest)
	RequestsMemory string `json:"requestsMemory"` //request는 컨테이너가 생성될때 요청하는 리소스 양 (defalt: 1Gi)
	RequestsCpu   string `json:"requestsCpu"`    //default: 500m
	LimitsMemory   string `json:"limitsMemory"`   //리소스가 더 필요한 경우 추가로 더 사용할 수 있는 부분 (default: 2Gi)
	LimitsCpu      string `json:"limitsCpu"`      //default: 1000m
	NodePort	string `json:"nodePort"`	//default: 32000
}

func grafApplyYamlFileCmd(gitPath string, fileName string, spec *GrafanaSpec, option string) {
	replacerArg := strings.NewReplacer(
		"{{namespaceName}}", spec.NamespaceName,
		"{{imgVersion}}", spec.ImgVersion,
		"{{requestsMemory}}", spec.RequestsMemory,
		"{{requestsCpu}}", spec.RequestsCpu,
		"{{limitsMemory}}", spec.LimitsMemory,
		"{{limitsCpu}}", spec.LimitsCpu,
		"{{nodePort}}", spec.NodePort,
	)
	if yamlStr, err := getYaml(gitPath + fileName); err != nil {
			log.Printf("Failed to get yaml file: %v", err)
	} else {
		log.Println("Received YAML:")
		log.Println(yamlStr)

		// Replace all pairs.
		newFormatYmlString := replacerArg.Replace(yamlStr)
		//fmt.Println(newFormatYmlString)

		if err = ioutil.WriteFile("/tmp/custom_"+fileName, []byte(newFormatYmlString), 0666); err != nil {
			fmt.Println(err)
		} else{
			applyYamlFileCmd("/tmp/custom_"+fileName, option, spec.NamespaceName)
			//deleteFile("/tmp/custom_"+fileName)
		}
	}
}

func CreateGrafana(grafanaSpec GrafanaSpec, gitPath string) {
	// ////////test config
	// var grafana_spec = GrafanaSpec{
	// 	namespaceName:  "monitoring",
	// 	imgVersion:     "latest",
	// 	requestsMemory: "1Gi",
	// 	requests_cpu:    "500m",
	// 	limitsMemory:   "2Gi",
	// 	limitsCpu:      "1000m",
	//	nodePort:	"32000",
	// }
	// ///////////

	connectToClusterCmd()

	createNamespaceCmd(grafanaSpec.NamespaceName)
	///grafana + prometheus 라면
	grafApplyYamlFileCmd(gitPath, "graf_with_prom_config_map.yaml", &grafanaSpec, "")
	grafApplyYamlFileCmd(gitPath, "graf_deployment.yaml", &grafanaSpec, "")

	//////////////////Check deployment file
	cmd := exec.Command("kubectl", "get", "deployments", "--namespace="+grafanaSpec.NamespaceName)
	printCommand(cmd)
	output, err := cmd.CombinedOutput()
	printError(err)
	printOutput(output)
	//////////////////////////////////////////
	grafApplyYamlFileCmd(gitPath, "graf_service.yaml", &grafanaSpec, "--namespace=")

	grafApplyYamlFileCmd(gitPath, "graf_hpa.yaml", &grafanaSpec, "")

	///////////ingress and dns
	grafApplyYamlFileCmd(gitPath, "graf_dnsendpoint.yaml", &grafanaSpec, "")
	grafApplyYamlFileCmd(gitPath, "graf_ingressroute.yaml", &grafanaSpec, "")
}