如何检查程序是在终端运行还是在后台运行？

Question

嗨，我正在编写一个 go 程序，它根据执行情况以两种方式提供一些输出：

• 如果在终端中执行，则输出到 stdout
• 它与键盘快捷键（例如 CTRL+M）配对，如果它 通过键盘快捷键调用，它会发送 输出作为通知。

现在我想确定程序是否在终端中运行来决定输出模式...linux中有没有办法确定这种情况？

当前程序在标准输出和通知中输出，我只想在通过键盘快捷键调用时通过通知发送输出。因为当它不在终端中运行时打印到标准输出根本没有用。

Answer 1

我假设当通过键盘快捷键调用它时，它是由您的桌面环境以某种方式运行的，并且输出不会是终端，因此您可以检查stdout是否是tty。根据 Arkadiusz Drabczyk 的回答，您可以使用以下代码：

package main

import (
        "os"
        "golang.org/x/sys/unix"
)

func main() {
        _, err := unix.IoctlGetWinsize(int(os.Stdout.Fd()), unix.TIOCGWINSZ)
        if err != nil {
                // Show notification here
        } else {
                // Print to stdout here
        }
}

请注意，当使用

|

或

>

重定向时，这也会导致它使用通知而不是标准输出。如果这不好，您将不得不使用显式命令行参数 - 无论如何，我个人更喜欢这个，因为它清楚地传达了意图并且不会产生意外。您可以接受添加到键盘快捷键的参数

--notification

，但在命令行中调用该工具时不使用该参数。

如果由于某种原因不方便添加参数，您也可以使用具有不同名称的符号链接并检查

os.Args[0]

。例如，如果您的程序是

~/bin/hello

，那么您将创建一个名为

~/bin/hello-notify

的“第二个版本”，它实际上只是

~/bin/hello

的符号链接，但在您的代码中，您检查

os.Args[0]

是否以以下结尾

-notify

。然后您可以将键盘快捷键连接到

~/bin/hello-notify

，但在命令行中使用

~/bin/hello

。