116 lines
2.9 KiB
Go
116 lines
2.9 KiB
Go
// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
|
|
// Use of this source code is governed by the GPL 3.0
|
|
|
|
package main
|
|
|
|
// An app to submit patches for the 30 GO GUI repos
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
"go.wit.com/lib/protobuf/virtpb"
|
|
"go.wit.com/log"
|
|
)
|
|
|
|
func doDroplet() (string, error) {
|
|
err := me.clusters.ConfigLoad()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
msg := []byte(`{"message": "Hello"}`)
|
|
|
|
// Initialize a persistent client with a custom Transport
|
|
client = &http.Client{
|
|
Transport: &http.Transport{
|
|
DisableKeepAlives: false, // Ensure Keep-Alive is enabled
|
|
},
|
|
Timeout: 10 * time.Second, // Set a reasonable timeout
|
|
}
|
|
|
|
me.cmap = make(map[*virtpb.Cluster]*adminT)
|
|
for c := range me.clusters.IterAll() {
|
|
var err error
|
|
admin := new(adminT)
|
|
if admin.cluster == nil {
|
|
admin.cluster = new(virtpb.Cluster)
|
|
}
|
|
me.cmap[c] = admin
|
|
log.Info("found in the config file", c.URL[0])
|
|
// a.makeClusterGroup(c)
|
|
admin.url, err = url.Parse(c.URL[0])
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// update the droplet list
|
|
if data, err := postData(admin.url.String()+"/DropletsPB", msg); err != nil {
|
|
log.Info("/DropletsPB Error:", err)
|
|
continue
|
|
} else {
|
|
admin.cluster.Droplets = new(virtpb.Droplets)
|
|
if err := admin.cluster.Droplets.Unmarshal(data); err != nil {
|
|
log.Printf("DropletsPB Response len:%d\n", len(data))
|
|
log.Println("droplets marshal failed", err)
|
|
continue
|
|
}
|
|
}
|
|
log.Printf("Cluster Name: %s\n", c.Name)
|
|
log.Printf("Number of Droplets: %d\n", admin.cluster.Droplets.Len())
|
|
|
|
if argv.Droplet.Name == "" {
|
|
return "", fmt.Errorf("--name droplet name was empty")
|
|
}
|
|
|
|
var found *virtpb.Droplets
|
|
found = virtpb.NewDroplets()
|
|
all := admin.cluster.Droplets.All()
|
|
for all.Scan() {
|
|
vm := all.Next()
|
|
if argv.Droplet.Name == vm.Hostname {
|
|
if argv.Droplet.Show != nil {
|
|
log.Info(vm.SprintHeader())
|
|
txt := vm.FormatTEXT()
|
|
log.Info(txt)
|
|
return "droplet status", nil
|
|
}
|
|
if argv.Droplet.Start != nil {
|
|
log.Info("should start droplet here")
|
|
log.Info(vm.SprintHeader())
|
|
e := new(virtpb.Event)
|
|
e.Etype = virtpb.EventType_POWERON
|
|
e.DropletUuid = vm.Uuid
|
|
|
|
if err := admin.postEvent(e); err != nil {
|
|
return "droplet start err", err
|
|
}
|
|
return "droplet start", nil
|
|
}
|
|
return "droplet found", fmt.Errorf("do what to the droplet?")
|
|
}
|
|
found.Append(vm)
|
|
}
|
|
log.Println("On Droplet count=", found.Len())
|
|
}
|
|
return "", fmt.Errorf("droplet %s not found", argv.Droplet.Name)
|
|
}
|
|
|
|
func doEvent(e *virtpb.Event) *virtpb.Event {
|
|
result := new(virtpb.Event)
|
|
if e.Etype == virtpb.EventType_POWERON {
|
|
log.Println("power on droplet on local cluster here", e.DropletUuid)
|
|
result.State = virtpb.Event_DONE
|
|
rs, err := Start(e.DropletUuid)
|
|
log.Println("Start() returned", rs)
|
|
log.Println("Start() returned err", err)
|
|
if err != nil {
|
|
result.Error = fmt.Sprintf("%v", err)
|
|
}
|
|
}
|
|
|
|
return result
|
|
}
|