go语言-k8s宿主信息采集运维脚本

背景

由于工作需要，需要采集k8s集群中的宿主相关信息，包括cpu,memory,lvm,标签等信息。通常作为SRE会主选shell或python脚本来实现。但最近我们团队主流开发语言已经切换到golang.所以本次尝试用go语言来写写运维脚本。

实现流程图

代码实现

package mainimport ("errors""fmt""github.com/gocarina/gocsv""log""os""os/exec""strings""time"
)// K8SNode 描述宿主节点
type K8SNode struct {IP           string `csv:"ip"`CPUSize      string `csv:"cpu_size"`MemorySize   string `csv:"memory_size"`LVMSize      string `csv:"lvm_size"`VMINum       int    `csv:"vmi_num"` // 虚拟机个数ProjectLabel string `csv:"project_label"`
}// NodeDescribeOutput 执行kubectl describe node 指定ip后面的输出内容
func NodeDescribeOutput(ip string) (string, error) {cmd := "kubectl  describe node " + iplog.Printf("execut command: %s\n", cmd)output, err := ExecCmd(cmd)if err != nil {log.Println("err:", err)return "", err}return output, nil
}// CpuSize 获取节点cpu
func CPUSize(describeInfo string) (string, error) {describeInfoList := strings.Split(describeInfo, "\n")for _, line := range describeInfoList {if strings.Contains(line, "cpu:") {cpusizelist := strings.Split(line, ":")//log.Printf("cpusize: %s", cpusizelist[1])size := strings.Replace(cpusizelist[1], " ", "", -1)return size, nil}}return "", errors.New("not found cpu size.")
}// MemorySize 获取节点内存大小
func MemorySize(describeInfo string) (string, error) {describeInfoList := strings.Split(describeInfo, "\n")for _, line := range describeInfoList {if strings.Contains(line, "memory:") {memorysizelist := strings.Split(line, ":")size := strings.Replace(memorysizelist[1], " ", "", -1)return size, nil}}return "", errors.New("not found memory size.")
}// LVMSize 获取节点本地磁盘大小
func LVMSize(describeInfo string) (string, error) {describeInfoList := strings.Split(describeInfo, "\n")for _, line := range describeInfoList {if strings.Contains(line, "lvm:") {sizelist := strings.Split(line, ":")size := strings.Replace(sizelist[1], " ", "", -1)return size, nil}}return "", errors.New("not found memory size.")
}// VMINum 统计节点已经允许vmi的数量(vmi表示在k8s+kubevirt中KVM对应的对象)
func VMINum(describeInfo string) (int, error) {describeInfoList := strings.Split(describeInfo, "\n")var count intfor _, line := range describeInfoList {if strings.Contains(line, "irt-launcher-") {count++}}return count, nil
}// ProjectLabel 获取节点项目标签[我们业务定义project表示宿主用于哪一个项目]
func ProjectLabel(describeInfo string) (string, error) {describeInfoList := strings.Split(describeInfo, "\n")for _, line := range describeInfoList {if strings.Contains(line, "project=") {//log.Printf("cpusize: %s", cpusizelist[1])label := strings.Replace(line, " ", "", -1)label = strings.Replace(label, "project=", "", -1)label = strings.Replace(label, "\n", "", -1)return label, nil}}return "", errors.New("not found project label.")
}// ExecCmd 对exec.Command()的封装
func ExecCmd(cmdstr string) (string, error) {cmd := exec.Command("bash", "-c", cmdstr)res, err := cmd.Output()if err != nil {return "", err}return string(res), nil}// GetIPList 获取集群中节点对应的IP列表
func GetIPList() []string {cmd := "kubectl get  node |grep -E \"[0-9]\" |awk '{print $1}'"log.Printf("execut command: %s\n", cmd)output, err := ExecCmd(cmd)if err != nil {log.Println("err:", err)var nilIPList []stringreturn nilIPList}return strings.Split(string(output), "\n")
}// 将结构体切片转换为csv文件
func WriteToCsv(filename string, nodes []*K8SNode) {// 创建csv文件os.Chdir("/tmp")nodesFile, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE, os.ModePerm)if err != nil {panic(err)}defer nodesFile.Close()// 将结构体切片转换为csv文件if err := gocsv.Marshal(&nodes, nodesFile); err != nil { // Load clients from filepanic(err)}
}// uploadToFilesever 将文件通过scp命令传到文件服务器
func uploadToFilesever(filename string) error {cmd := "scp " + filename + " 172.17.123.89:/home/segops/app/nginx/data/tmp/"log.Printf("execut command: %s\n", cmd)_, err := ExecCmd(cmd)if err != nil {log.Println("err:", err)return err}return nil
}func main() {// Nodes 保存节点的切片var Nodes = []*K8SNode{}// count 用于统计节点数量var count intipList := GetIPList()// 依次处理每一个宿主节点IPfor _, ip := range ipList {//fmt.Println(ip)if ip == "" {continue}count++var oneNode = &K8SNode{}oneNode.IP = ipdescribeNodeOutput, err := NodeDescribeOutput(ip)if err != nil {log.Printf("%s NodeDescribeOutput error.\n", ip)}cpusize, err := CPUSize(describeNodeOutput)if err == nil {oneNode.CPUSize = cpusize} else {oneNode.CPUSize = ""}memsize, err := MemorySize(describeNodeOutput)if err == nil {oneNode.MemorySize = memsize} else {oneNode.MemorySize = ""}lvmsize, err := LVMSize(describeNodeOutput)if err == nil {oneNode.LVMSize = lvmsize} else {oneNode.LVMSize = ""}vminum, err := VMINum(describeNodeOutput)if err == nil {oneNode.VMINum = vminum} else {oneNode.VMINum = -1}projectlabel, err := ProjectLabel(describeNodeOutput)if err == nil {oneNode.ProjectLabel = projectlabel} else {oneNode.ProjectLabel = ""}fmt.Printf("%v\n", oneNode)//log.Println(ProjectLabel(describeNodeOutput))// 将一个K8SNode对象加入切片Nodes = append(Nodes, oneNode)}//fmt.Println("nodemap=%v", nodeMap)fmt.Printf("===========================================================================================\n")fmt.Printf("=========================================统计结果===========================================\n")fmt.Printf("===========================================================================================\n")fmt.Printf("ip\t\tcpu\t\tmemory\t\tlvmsize\t\tvminum\t\tprojectlabel\n")for _, node := range Nodes {fmt.Printf("%s\t%s\t\t%s\t%s\t\t%d\t\t%s\n", node.IP, node.CPUSize, node.MemorySize, node.LVMSize, node.VMINum, node.ProjectLabel)}fmt.Println()fmt.Printf("node num is %d\n", count)fmt.Println()fmt.Println("write to csv file:")currentTime := time.Now().Format("20060102150405")hostname, _ := os.Hostname()csvfilename := hostname + "_" + "nodesinfo" + "_" + currentTime + ".csv"// 写入csv文件WriteToCsv(csvfilename, Nodes)// 上传到文件服务器err := uploadToFilesever(csvfilename)if err != nil {fmt.Println("upload csv file error.")return}// 定义下载文件的路径downloadPath := "http://172.17.123.89:8080/tmp/" + csvfilenamefmt.Println()fmt.Println("csv file can download from: ", downloadPath)fmt.Println()
}

实现结果

​
17:35:14  ===========================================================================================
17:35:14  =========================================统计结果===========================================
17:35:14  ===========================================================================================
17:35:14  ip		cpu		memory		lvmsize		vminum		projectlabel
17:35:14  172.24.52.11	96		394687572Ki	4862Gi		7		personal-dev
17:35:14  172.24.52.12	96		394687572Ki	4862Gi		8		personal-dev
17:35:14  172.24.52.13	96		394682812Ki	4862Gi		2		personal-dev
17:35:14  172.24.52.14	96		394687572Ki	4862Gi		5		personal-dev
17:35:14  172.24.52.141	96		394687572Ki	4862Gi		1		tools
17:35:14  172.24.52.142	96		394687572Ki	4862Gi		0		personal-dev-unuse
17:35:14  172.24.52.143	96		394687572Ki	4862Gi		0		personal-dev-unuse
17:35:14  172.24.53.79	80		394679084Ki	3225Gi		0		unuse
17:35:14  172.24.53.93	72		131267356Ki	11431Gi		0		unuse
17:35:14  172.24.53.94	72		131267356Ki	11431Gi		0		unuse
17:35:14  
17:35:14  node num is 122
17:35:14  
17:35:14  write to csv file:
17:35:14  2024/03/07 17:35:14 execut command: scp hyd01-seg-admin01_nodesinfo_20240307173514.csv 172.17.123.89:/home/segops/app/nginx/data/tmp/
17:35:15  
17:35:15  csv file can download from:  http://172.17.123.89:8080/tmp/hyd01-seg-admin01_nodesinfo_20240307173514.csv​

 

总结

用go脚本写了约300行，并不简洁。所以在工作实际中，如果写一些逻辑简单的脚本建议首选用shell或python。

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.hqwc.cn/news/522180.html

如若内容造成侵权/违法违规/事实不符，请联系编程知识网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

每日OJ题_牛客CM24 最近公共祖先

每日OJ题_牛客CM24 最近公共祖先

目录 牛客CM24 最近公共祖先 解析代码 牛客CM24 最近公共祖先 最近公共祖先_牛客题霸_牛客网 解析代码 class LCA { public:int getLCA(int a, int b) {// 左孩 父 * 2 1，右孩 父 * 2 2；// 父 （孩 - 1） / 2；/…
阅读更多...
妇女节特刊 | 上海交大吴梦玥：用言语智能技术，为精神疾病开出第一张诊疗单

妇女节特刊 | 上海交大吴梦玥：用言语智能技术，为精神疾病开出第一张诊疗单

从孩童的咿呀学语，到青年时期聊不完的少年心事，进而人到中年，耳边既有父母长辈的叮咛关切，也有生活压力一刻不停的催促……细细回想，人生每个阶段的经历与回忆都萦绕着不同的声音，夏日的蝉鸣与篮球场上的聒…
阅读更多...
探索数据可视化：Matplotlib 基础指南

探索数据可视化：Matplotlib 基础指南

图形绘制 import numpy as np import pandas as pd import matplotlib.pyplot as pltx np.linspace(0,2 * np.pi,100)# 说明：正弦波。x：NumPy数组 # 所有的数据，进行正弦计算 y np.sin(x)plt.plot(x,y)# 指定x轴范围 plt.xlim(-1,10) # 指…
阅读更多...
EXSI create datastore

EXSI create datastore

文章目录 1. 简介2. 清空磁盘3. 删除表4. 创建database 1. 简介 在 ESXi 环境中创建数据存储(Datastore)的步骤如下: 登录 vSphere Web Client 打开 Web 浏览器,输入 ESXi 主机或 vCenter Server 的 IP 地址,使用有权限的账户登录。 在 ESXi 环境中创建数据存储(Datastore)…
阅读更多...
使用 Docker 部署 File Browser 文件管理系统

使用 Docker 部署 File Browser 文件管理系统

1）File Browser 介绍 官网：https://filebrowser.org/ GitHub：https://github.com/filebrowser/filebrowser 今天为大家分享一款开源的私有云盘项目：File Browser，简单实用、轻量级、跨平台，安装部署简单快…
阅读更多...
图文并茂的讲清楚Linux零拷贝技术

图文并茂的讲清楚Linux零拷贝技术

今天我们来聊一聊Linux零拷贝技术，今天我们以一个比较有代表性的技术sendfile系统调用为切入点，详细介绍一下零拷贝技术的原理。 1.零拷贝技术简介 Linux零拷贝技术是一种优化数据传输的技术，它可以减少数据在内核态和用户态之间的拷贝次数&…
阅读更多...
基于java SSM校园兼职平台系统设计和实现

基于java SSM校园兼职平台系统设计和实现

基于java SSM校园兼职平台系统设计和实现 博主介绍：多年java开发经验，专注Java开发、定制、远程、文档编写指导等,csdn特邀作者、专注于Java技术领域 作者主页 央顺技术团队 Java毕设项目精品实战案例《500套》 欢迎点赞 收藏 ⭐留言 文末获取源码联系方…
阅读更多...
RabbitMQ的Windows版安装教程

RabbitMQ的Windows版安装教程

文章目录 前言一、Windows安装RabbitMQ总结 前言 曾经写过一篇关于RabbitMQ的Ubuntu安装教程（http://t.csdnimg.cn/5CYfC），当时使用的是Docker将RabbitMQ安装到虚拟机上，但是有很多小伙伴问Windows上如何进行安装RabbitMQ&#x…
阅读更多...
神经网络softmax算法与卷积层

神经网络softmax算法与卷积层

多类分类： 多类是分类算法中的一种，它区别于我们的0，1这样子的二进制分类，它会有多个分类的标签，让我们去取其中的一个。 softmax函数： softmax回归算法是我们的sigmoid回归的推广。 上图就是softmax运…
阅读更多...
java项目线上捉BUG经验记录

java项目线上捉BUG经验记录

一 线上问题 昨晚上突然接到公司紧急来电电桩设备大面积离线，意味着某市的车无法充电……”，细想这个平台很久都没有更新，最近出现问题是刚好在一个月前也是出现这种情况，再上一次就是几年前更新的。平台平时是稳定，开…
阅读更多...
获取C语言语句对应的汇编码和机器指令

获取C语言语句对应的汇编码和机器指令

借助IDE的调试功能 以CodeBlocks为例，先设置断点，然后点击红色三角形调试。 然后选择Debug➡ Debugging Windows➡Disassembly 就可以看到了 使用命令行 在工程文件中，一般可以找到一个.o文件。如果没有，可以先在program.c的目录下…
阅读更多...
c++提高部分

c++提高部分

c++提高部分 这部分主要涉及泛型编程和STL技术 1. 模版 1.1 模版的概念 模版就是通用的模具，大大提高复用性，但需要根据需求改动一些东西 1.2 函数模版 c++另一种编程思想为泛型编程，主要利用的技术就是模版c++提供两种模版机制：函数模板和类模板1.2.1 函数模板语法 …
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023