4c33fc8b35
Automatically collects cpu & memory & network usage stats for each VM
every minute and appends it to `/vm/logs/$vmname.log`.
An example row in that log file looks like:
```
{"cpu_usage_usec":114769860,"memory_total_kb":8128676,
"memory_free_kb":6120912,"rx_bytes":313094089,
"tx_bytes":4107151,"timestamp":"2025-01-23T04:58:13Z"}
```
We will most likely revise where we store the stats. This is just a
prototype to try different pieces end-to-end. What & how we collect
stats is also subject to change.
108 lines
2.4 KiB
Ruby
108 lines
2.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "shellwords"
|
|
|
|
class VmPath
|
|
def initialize(vm_name)
|
|
@vm_name = vm_name
|
|
end
|
|
|
|
# Read from a path, removing a trailing newline if present.
|
|
def read(path)
|
|
File.read(path).chomp
|
|
end
|
|
|
|
# Write to a path, adding a trailing newline if not already present.
|
|
def write(path, s)
|
|
s += "\n" unless s.end_with?("\n")
|
|
File.write(path, s)
|
|
end
|
|
|
|
def dnsmasq_service
|
|
"/etc/systemd/system/#{@vm_name}-dnsmasq.service"
|
|
end
|
|
|
|
def write_dnsmasq_service(s)
|
|
write(dnsmasq_service, s)
|
|
end
|
|
|
|
def stats_collector_service
|
|
"/etc/systemd/system/#{@vm_name}-stats-collector.service"
|
|
end
|
|
|
|
def write_stats_collector_service(s)
|
|
write(stats_collector_service, s)
|
|
end
|
|
|
|
def stats_collector_timer
|
|
"/etc/systemd/system/#{@vm_name}-stats-collector.timer"
|
|
end
|
|
|
|
def write_stats_collector_timer(s)
|
|
write(stats_collector_timer, s)
|
|
end
|
|
|
|
def systemd_service
|
|
File.join("/etc/systemd/system",
|
|
IO.popen(["systemd-escape", @vm_name + ".service"]) { _1.read.chomp })
|
|
end
|
|
|
|
def write_systemd_service(s)
|
|
write(systemd_service, s)
|
|
end
|
|
|
|
def home(n)
|
|
File.join("", "vm", @vm_name, n)
|
|
end
|
|
|
|
# Define path, q_path, read, write methods for files in
|
|
# `/vm/#{vm_name}`
|
|
%w[
|
|
guest_ephemeral
|
|
clover_ephemeral
|
|
dnsmasq.conf
|
|
meta-data
|
|
network-config
|
|
user-data
|
|
cloudinit.img
|
|
ch-api.sock
|
|
serial.log
|
|
hugepages
|
|
public_ipv4
|
|
nftables_conf
|
|
prep.json
|
|
cert
|
|
].each do |file_name|
|
|
method_name = file_name.tr(".-", "_")
|
|
fail "BUG" if method_defined?(method_name)
|
|
|
|
# Method producing a path, e.g. #user_data
|
|
define_method method_name do
|
|
home(file_name)
|
|
end
|
|
|
|
# Method producing a shell-quoted path, e.g. #q_user_data.
|
|
quoted_method_name = "q_" + method_name
|
|
fail "BUG" if method_defined?(quoted_method_name)
|
|
define_method quoted_method_name do
|
|
home(file_name).shellescape
|
|
end
|
|
|
|
# Method reading the file's contents, e.g. #read_user_data
|
|
#
|
|
# Trailing newlines are removed.
|
|
read_method_name = "read_" + method_name
|
|
fail "BUG" if method_defined?(read_method_name)
|
|
define_method read_method_name do
|
|
read(home(file_name))
|
|
end
|
|
|
|
# Method overwriting the file's contents, e.g. #write_user_data
|
|
write_method_name = "write_" + method_name
|
|
fail "BUG" if method_defined?(write_method_name)
|
|
define_method write_method_name do |s|
|
|
write(home(file_name), s)
|
|
end
|
|
end
|
|
end
|