Linux脚本:将.sh文件输出映射到日志文件并添加时间戳

文章目录

背景

  • 在linux系统中,当调用.sh脚本时,脚本中的标准输出原本是打印到控制台的,我们可以将其打印到某个文件,并添加上时间戳,方便问题溯源。
  • 有时我们运行的脚本是.py脚本,也需要将输出映射到日志文件。

解决办法

调用.sh脚本时

  • 假设我的目标脚本名为script.sh脚本,我需要写一个start.sh脚本,在该脚本中启动script.sh脚本,并将script.sh脚本的输出保存到out_log.txt文件中,过程如下:
  • 首先安装moreutils包:
bash 复制代码
sudo apt-get update
sudo apt-get install moreutils
  • 编写 start.sh 脚本:这个脚本将启动 script.sh 并处理日志重定向和时间戳添加。
bash 复制代码
# 运行 script.sh,并将其输出重定向到 out_log.txt,添加时间戳
./script.sh 2>&1 | ts '[%Y-%m-%d %H:%M:%S]' >> out_log.txt
  • 如果不安装moreutils包,将start.sh改为如下内容即可:
bash 复制代码
# 运行 script.sh,并将其输出重定向到 out_log.txt,添加时间戳,并在后台运行
(
  while IFS= read -r line
  do
    echo "$(date '+[%Y-%m-%d %H:%M:%S]') $line"
  done < <(./script.sh 2>&1)
) >> out_log.txt &
  • 假设script.sh脚本中内容如下:
bash 复制代码
#!/bin/sh

echo "Script started"
echo "Doing some work..."
sleep 2
echo "Work done"
echo "Script finished"
  • 最终结果我们将看到输出如下:
bash 复制代码
[2024-06-06 12:00:00] Script started
[2024-06-06 12:00:00] Doing some work...
[2024-06-06 12:00:02] Work done
[2024-06-06 12:00:02] Script finished

调用.py脚本时

  • 我们新建一个运行目标start.sh的启动脚本如下:
bash 复制代码
current_time=$(date +"%Y-%m-%d-%H-%M-%S")

echo "${current_time} start python3:\n" >>output.log

nohup python3 Test1.py &

echo $! > pid_file.txt
  • 在start.sh脚本中启动目标的.py文件。在.py脚本中,加入以下命令使能将print打印到日志文件:
py 复制代码
PrintToFile = True

if PrintToFile:
    # 配置日志记录器
    logging.basicConfig(filename='output.log', level=logging.INFO, format='%(asctime)s %(message)s', datefmt='%Y-%m-%d %H:%M:%S')

    # 定义一个函数用于重定向标准输出到日志文件
    def redirect_stdout_to_log():
        class StdOutLogger(object):
            def write(self, message):
                # 添加时间信息并写入日志文件
                logging.info('%s %s', datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'), message.rstrip())

            def flush(self):
                pass

        # 将标准输出重定向到 StdOutLogger 类的实例
        sys.stdout = StdOutLogger()

    # 调用函数将标准输出重定向到日志文件
    redirect_stdout_to_log()
  • 并将该进程号保存在一个pid_file.txt中。当我们后面想停止该进程时,直接运行以下stop.sh脚本即可:
bash 复制代码
pid=$(cat pid_file.txt)

kill $pid
相关推荐
淮北4945 分钟前
计算机网络学习(六、应用层)
linux·学习·计算机网络
红豆怪怪6 分钟前
[LeetCode 热题 100] 32. 最长有效括号
数据结构·python·算法·leetcode·动态规划·代理模式
lew-yu30 分钟前
【已解决】Linux中程序脚本可以手动执行成功,但加在rc.local中不能开机自启
linux·服务器
ajassi200032 分钟前
linux C 语言开发 (四) linux系统常用命令
linux·运维·服务器
小嵌同学35 分钟前
Linux:malloc背后的实现细节
大数据·linux·数据库
EveryPossible1 小时前
如何终止画图
linux·编辑器·vim
大嘴带你水论文1 小时前
震惊!仅用10张照片就能随意编辑3D人脸?韩国KAIST最新黑科技FFaceNeRF解析!
论文阅读·人工智能·python·科技·计算机视觉·3d·transformer
CodeCraft Studio1 小时前
国产化PDF处理控件Spire.PDF教程:如何在 Java 中通过模板生成 PDF
java·python·pdf·spire.pdf·java创建pdf·从html创建pdf
摆烂z2 小时前
Jupyter Notebook的交互式开发环境方便py开发
ide·python·jupyter
wangjialelele2 小时前
Linux匿名管道和命名管道以及共享内存
linux·运维·网络