Ren*_* B. 5 python amazon-web-services aws-lambda
如何查看 AWS Lamdba 函数使用的存储的已用大小?
例如,可以这样检查文件是否存在:
import os
os.path.isfile('/tmp/' + filename)
Run Code Online (Sandbox Code Playgroud)
但是,我需要知道何时达到 512 MB 限制。
由于 Lambda 函数只是在 Linux 环境中的容器上运行,因此我们可以使用操作系统来告诉我们临时文件系统上还剩下多少空间。
from subprocess import check_output
out = str(check_output(["df", '-k']))
print(out)
result = re.search(r"(\d+)\s+(\d+)\s+(\d+)\s+(\d+)\%\s+\/tmp", out)
size, used, available, used_pct = (int(g) for g in result.groups())
print(size, used, available, used_pct) # 538424 440 526148 1
Run Code Online (Sandbox Code Playgroud)
这就是这里发生的事情。首先,我们使用该subprocess.check_output函数运行命令:df -k该命令显示有关磁盘空间使用情况信息的字符串(以 KB 为单位)。以下是 lambda 函数的输出示例:
Filesystem 1K-blocks Used Available Use% Mounted on\n/dev/root 6127168 4978732 1132052 82% /\n/dev/vdb 1965904 45296 1904224 3% /dev\n/dev/loop0 538424 440 526148 1% /tmp\n
Run Code Online (Sandbox Code Playgroud)
您可以看到分区位于字符串的末尾/tmp。因此,我们获取输出并使用正则表达式模式进行搜索:(\d+)\s+(\d+)\s+(\d+)\s+(\d+)\%\s+\/tmp它将相关字段分为 4 组,我们可以将它们设置为代表以下内容的变量size:总分区大小(以 KB 为单位)、:used当前使用的空间量(以 KB 为单位),:available分区上仍然可用的空间量(以 KB 为单位),以及used_pct:正在使用的分区空间的百分比。
这是获取可用空间的一种非常简单的方法,如果需要精确测量或猜测是否可以将新文件写入磁盘,我不会使用它。但如果您只是在执行时记录它以进行调试,那么它应该完全没问题。