sped up the soft by about 3x using basic caching and threads
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
parent
7ed109f70b
commit
e6dd3d81ce
@ -1,5 +1,4 @@
|
||||
import logging
|
||||
import time
|
||||
|
||||
from . import creation_utils, lxc_utils
|
||||
from ..utils.machine import LinuxMachine
|
||||
@ -177,7 +176,7 @@ class LXC(LinuxMachine):
|
||||
Get IPv4
|
||||
:return: ipv4
|
||||
"""
|
||||
if self.ipv4 == "dhcp":
|
||||
if self.ipv4 == "dhcp" or self.ipv4 == "auto":
|
||||
if self.is_running():
|
||||
if self.has_program("ip", use_ssh=use_ssh):
|
||||
if netmask:
|
||||
@ -304,6 +303,7 @@ class LXC(LinuxMachine):
|
||||
Start LXC
|
||||
"""
|
||||
self.pve.start_lxc(self.lxc_id)
|
||||
self.ipv4 = self.get_ipv4()
|
||||
|
||||
def stop(self):
|
||||
"""
|
||||
@ -389,7 +389,6 @@ class LXC(LinuxMachine):
|
||||
if working_directory:
|
||||
command = f"cd {working_directory} && {command}"
|
||||
|
||||
|
||||
# Using pct exec works every time but is 8x slower than using ssh
|
||||
if use_ssh:
|
||||
return self.pve.run_command(
|
||||
|
@ -6,6 +6,7 @@ from src.utils import utils
|
||||
class LinuxMachine():
|
||||
|
||||
def __init__(self):
|
||||
self.known_programs = []
|
||||
pass
|
||||
|
||||
def update(self):
|
||||
@ -60,10 +61,16 @@ class LinuxMachine():
|
||||
|
||||
:return: boolean
|
||||
"""
|
||||
if program in self.known_programs:
|
||||
return True
|
||||
|
||||
if type(program) == str:
|
||||
return self.run_command("which " + program, return_status_code=True, use_ssh=use_ssh) == 0
|
||||
result = self.run_command("which " + program, return_status_code=True, use_ssh=use_ssh) == 0
|
||||
if result:
|
||||
self.known_programs.append(program)
|
||||
return result
|
||||
elif type(program) == list:
|
||||
return all((self.run_command("which " + p, return_status_code=True, use_ssh=use_ssh) == 0) for p in program)
|
||||
return all(self.has_program(program=p, use_ssh=use_ssh) for p in program)
|
||||
|
||||
def has_file(self, file: str or Path):
|
||||
"""Check if file exists on LXC
|
||||
|
@ -2,6 +2,7 @@ from __future__ import annotations
|
||||
|
||||
import fnmatch
|
||||
import subprocess
|
||||
import threading
|
||||
import time
|
||||
from pathlib import Path
|
||||
from typing import TYPE_CHECKING
|
||||
@ -40,6 +41,31 @@ class ProxmoxHost(LinuxMachine):
|
||||
self.connection = None
|
||||
self.repo_path = path.as_posix()
|
||||
|
||||
self.running_lxcs = []
|
||||
|
||||
threading.Thread(target=self.update, daemon=True).start()
|
||||
|
||||
def update(self):
|
||||
update_connection = Connection(host=self.host, user=self.user, port=self.port)
|
||||
while True:
|
||||
pct_output = update_connection.run("pct list", hide=True, warn=True, encoding="utf-8").stdout.split("\n")[
|
||||
1:]
|
||||
for line in pct_output:
|
||||
if line.strip():
|
||||
if len(line.split()) == 3:
|
||||
vm_id, status, name = line.split()
|
||||
else:
|
||||
vm_id, status, lock, name = line.split()
|
||||
|
||||
if "running" in status:
|
||||
if vm_id not in self.running_lxcs:
|
||||
self.running_lxcs.append(int(vm_id))
|
||||
else:
|
||||
if vm_id in self.running_lxcs:
|
||||
self.running_lxcs.remove(int(vm_id))
|
||||
|
||||
time.sleep(0.5)
|
||||
|
||||
def __str__(self):
|
||||
return f"ProxmoxHost({self.host}, {self.user}, {self.port})"
|
||||
|
||||
@ -99,6 +125,8 @@ class ProxmoxHost(LinuxMachine):
|
||||
if the stdout is empty and exception_on_empty_stdout is True
|
||||
"""
|
||||
|
||||
start_time = time.time()
|
||||
|
||||
# Check if host is None, if it is, run the command locally, else run it on the host via SSH
|
||||
if self.host is None:
|
||||
command_result = subprocess.run(command, shell=True, capture_output=True, encoding="utf-8")
|
||||
@ -107,6 +135,9 @@ class ProxmoxHost(LinuxMachine):
|
||||
else:
|
||||
raise Exception("No host or connection provided")
|
||||
|
||||
end_time = time.time()
|
||||
print(f"Command {command} took {end_time - start_time} seconds to run")
|
||||
|
||||
# If return code is not 0 and that exception_on_exit is True and return_status_code is False, throw an exception
|
||||
if command_result.return_code != 0 and exception_on_exit and not return_status_code:
|
||||
raise Exception(f"Error while running command: \n{command_result.stderr}")
|
||||
@ -149,8 +180,14 @@ class ProxmoxHost(LinuxMachine):
|
||||
def is_lxc_running(self, lxc_id):
|
||||
"""Check if the given LXC is running on the Proxmox host."""
|
||||
|
||||
pct_status_output = self.run_command(f"pct status {lxc_id}", exception_on_exit=False)
|
||||
return "running" in pct_status_output
|
||||
if lxc_id in self.running_lxcs:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
# Not used anymore because it's too slow, now using an update thread to keep track of running LXCs
|
||||
# pct_status_output = self.run_command(f"pct status {lxc_id}", exception_on_exit=False)
|
||||
# return "running" in pct_status_output
|
||||
|
||||
def start_lxc(self, lxc_id):
|
||||
"""Start the given LXC on the Proxmox host."""
|
||||
|
Loading…
Reference in New Issue
Block a user