mirror of
https://github.com/hanzoai/vm.git
synced 2026-10-30 14:55:23 +00:00
zooqueenandHanzo Dev
a121f8d0d0
- Rename Go module from github.com/casvisor/casvisor to github.com/hanzoai/visor - Update all copyright headers from casbin to Hanzo Industries Inc - Remove all hardcoded passwords and credentials from config files - All credentials now sourced from environment variables / Hanzo KMS - Switch default database from MySQL to PostgreSQL (matching production) - Rename binary output from server/casvisor to visor - Update Dockerfile, build.sh, docker-compose.yml, docker-entrypoint.sh - Replace casdoor* config key references with iam* equivalents - Update CI to use PostgreSQL service and mark CI-only test password - Add visor and casvisor binaries to .gitignore - Add envFrom secretRef for visor-env in K8s deployment - Add ExternalSecret template for Hanzo KMS credential provisioning Co-authored-by: Hanzo Dev <[email protected]>
128 lines
3.2 KiB
Go
128 lines
3.2 KiB
Go
// Copyright 2024 Hanzo Industries Inc. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE/2.0
|
|
//
|
|
// Unless required by law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package service
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
|
|
"github.com/digitalocean/go-libvirt"
|
|
)
|
|
|
|
type MachineKvmClient struct {
|
|
L *libvirt.Libvirt
|
|
}
|
|
|
|
// The URI format is "driver[+transport]://[username @][hostname][:post]/[path][?Extraparameters]", for example "kvm+ssh://root @192.168.1.100/system ".
|
|
// Use TCP connection method here: “kvm+tcp://192.168.1.100/system”.
|
|
func newMachineKvmClient(username string, hostname string) (MachineKvmClient, error) {
|
|
uri, err := url.Parse("kvm+tcp://" + hostname + "/system")
|
|
if err != nil {
|
|
return MachineKvmClient{}, err
|
|
}
|
|
l, err := libvirt.ConnectToURI(uri)
|
|
if err != nil {
|
|
return MachineKvmClient{}, err
|
|
}
|
|
|
|
return MachineKvmClient{L: l}, nil
|
|
}
|
|
|
|
func getMachineFromDom(l *libvirt.Libvirt, dom libvirt.Domain) (*Machine, error) {
|
|
machine := &Machine{}
|
|
|
|
name := dom.Name
|
|
ID := dom.ID
|
|
os, err := l.DomainGetOsType(dom)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cpuSize, err := l.DomainGetMaxVcpus(dom)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
memSize, err := l.DomainGetMaxMemory(dom)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
machine.Name = name
|
|
machine.Id = fmt.Sprintf("%d", ID)
|
|
machine.Os = os
|
|
machine.CpuSize = fmt.Sprintf("%d", cpuSize)
|
|
machine.MemSize = fmt.Sprintf("%d", memSize)
|
|
|
|
return machine, nil
|
|
}
|
|
|
|
func (client MachineKvmClient) GetMachines() ([]*Machine, error) {
|
|
flags := libvirt.ConnectListDomainsActive | libvirt.ConnectListDomainsInactive
|
|
doms, _, err := client.L.ConnectListAllDomains(1, flags)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
machines := []*Machine{}
|
|
for _, dom := range doms {
|
|
machine, err := getMachineFromDom(client.L, dom)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
machines = append(machines, machine)
|
|
}
|
|
|
|
return machines, nil
|
|
}
|
|
|
|
func (client MachineKvmClient) GetMachine(name string) (*Machine, error) {
|
|
dom, err := client.L.DomainLookupByName(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
machine, err := getMachineFromDom(client.L, dom)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return machine, nil
|
|
}
|
|
|
|
func (client MachineKvmClient) UpdateMachineState(name string, state string) (bool, string, error) {
|
|
dom, err := client.L.DomainLookupByName(name)
|
|
if err != nil {
|
|
return false, "", err
|
|
}
|
|
|
|
switch state {
|
|
case "Running":
|
|
err = client.L.DomainCreate(dom)
|
|
case "Stopped":
|
|
err = client.L.DomainShutdown(dom)
|
|
default:
|
|
return false, fmt.Sprintf("Unsupported state: %s", state), nil
|
|
}
|
|
|
|
if err != nil {
|
|
return false, "", err
|
|
}
|
|
|
|
return true, fmt.Sprintf("Domain: [%s]'s state has been successfully updated to: [%s]", name, state), nil
|
|
}
|
|
|
|
func (client MachineKvmClient) CreateMachine(spec *CreateMachineSpec) (*Machine, error) {
|
|
return nil, fmt.Errorf("CreateMachine not yet implemented for KVM")
|
|
}
|