watch 50 行代码实现 windows watch

驭鲸·环球

已于 2022-03-22 01:07:17 修改

阅读量709

点赞数 1

分类专栏： Golang 文章标签： golang 开发语言后端

于 2022-03-22 00:53:10 首次发布

知识因分享而具有价值，回忆昨天，把握今天，期待明天。 —— 驭鲸·环球

本文链接：https://blog.csdn.net/HighproX/article/details/123650431

版权

Golang 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

watch 50 行代码实现 windows watch

最近想监控 gpu 使用情况，无奈在 windows 上没有 watch 指令，nvidia-smi -lms 20 自带输出也不美观，那么为何不自己写个工具来监控呢。

如下图可以看到 gpu 温度和功耗的变动情况

watch-nvidia-smi

下面使用 50 行代码来实现 watch nvidia-smi 功能。

package main

import (
	"bufio"
	"bytes"
	"fmt"
	"log"
	"os/exec"
	"time"
)

func LineMoveUp(n int) {
	fmt.Printf("\033[%dA", n)
}

func HideCursor() {
	fmt.Printf("\033[?25l")
}

func ShowCursor() {
	fmt.Printf("\033[?25h")
}

// https://en.wikipedia.org/wiki/ANSI_escape_code
// CSI (Control Sequence Introducer) sequences
func main() {
	for true {
		out, err := exec.Command("nvidia-smi").Output()
		if err != nil {
			log.Fatal(err)
		}

		HideCursor()
		scanner := bufio.NewScanner(bytes.NewReader(out))
		line := 0
		for scanner.Scan() {
			fmt.Printf("%s\n", scanner.Text())
			line++
			if line > 12 {
				break
			}
		}
		LineMoveUp(line)
		time.Sleep(150 * time.Millisecond)
		ShowCursor()
	}
}