merge into a single file to make it easier to work with
Signed-off-by: Jeff Carr <jcarr@wit.com>
This commit is contained in:
parent
e4031a68c0
commit
7274c5c3c3
|
|
@ -1,5 +1,6 @@
|
||||||
run:
|
|
||||||
go run *.go
|
|
||||||
|
|
||||||
build:
|
build:
|
||||||
go build
|
go build
|
||||||
|
./gorilla-server
|
||||||
|
|
||||||
|
run:
|
||||||
|
go run *.go
|
||||||
|
|
|
||||||
|
|
@ -4,11 +4,144 @@ import "flag"
|
||||||
import "io"
|
import "io"
|
||||||
import "log"
|
import "log"
|
||||||
import "net/http"
|
import "net/http"
|
||||||
|
import "encoding/json"
|
||||||
import "github.com/gorilla/websocket"
|
import "io/ioutil"
|
||||||
|
|
||||||
import "github.com/golang/protobuf/proto"
|
import "github.com/golang/protobuf/proto"
|
||||||
import pb "git.wit.com/wit/witProtobuf"
|
import pb "git.wit.com/wit/witProtobuf"
|
||||||
|
import "github.com/gorilla/websocket"
|
||||||
|
import "github.com/davecgh/go-spew/spew"
|
||||||
|
|
||||||
|
const (
|
||||||
|
API = "http://stackapi:4000/"
|
||||||
|
MAX_NETS = 10
|
||||||
|
MAX_VMS = 10
|
||||||
|
)
|
||||||
|
|
||||||
|
type network struct {
|
||||||
|
Subdomain string `json:"subdomain"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func fetchNetworks(token string) []network {
|
||||||
|
var networks []network
|
||||||
|
req, err := http.NewRequest("GET", API + "clusters", nil)
|
||||||
|
if err != nil { log.Fatal("can't make new req") }
|
||||||
|
req.Header.Set("X-Wit-Auth", token)
|
||||||
|
client := &http.Client{}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil { log.Printf("Can't exec the req to list networks: %s", err) }
|
||||||
|
body, err := ioutil.ReadAll(resp.Body)
|
||||||
|
if err != nil { log.Fatal("can't read resp") }
|
||||||
|
log.Printf("NETWORKS =", body)
|
||||||
|
if body == nil {
|
||||||
|
log.Println("networks is nil")
|
||||||
|
} else {
|
||||||
|
len := len(body)
|
||||||
|
err = json.Unmarshal(body, &networks)
|
||||||
|
if err != nil {
|
||||||
|
log.Println("body = ", body)
|
||||||
|
log.Println("body len = ", len)
|
||||||
|
log.Println("can't unmarshal resp for list networks: %s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
spew.Dump(networks)
|
||||||
|
return networks
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
{
|
||||||
|
"name": "vm10",
|
||||||
|
"cpu": 4,
|
||||||
|
"ram": 2048,
|
||||||
|
"created_on": "2019-03-18T19:22:10.755428Z",
|
||||||
|
"scope": "private",
|
||||||
|
"role": "k8s-node",
|
||||||
|
"stage": "Creating...",
|
||||||
|
"seconds_left": 300,
|
||||||
|
"base_image": "debcore_guest_latest",
|
||||||
|
"address_list": [
|
||||||
|
"192.168.0.1/32"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
type address struct {
|
||||||
|
Address string
|
||||||
|
}
|
||||||
|
|
||||||
|
type vm struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Role string `json:"role"`
|
||||||
|
BaseImage string `json:"base_image"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func fetchVMs(token string, network string) []vm {
|
||||||
|
var vms []vm
|
||||||
|
req, err := http.NewRequest("GET", API + "vms/" + network, nil)
|
||||||
|
if err != nil { log.Fatal("can't make new req") }
|
||||||
|
req.Header.Set("X-Wit-Auth", token)
|
||||||
|
client := &http.Client{}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil { log.Printf("Can't exec the req to fetch vms for %s: %s", network, err) }
|
||||||
|
body, err := ioutil.ReadAll(resp.Body)
|
||||||
|
if err != nil { log.Fatal("can't read resp") }
|
||||||
|
err = json.Unmarshal(body, &vms)
|
||||||
|
if err != nil { log.Fatal("can't unmarshal resp for fetch vms: ", err, "\n\n", network, string(body)) }
|
||||||
|
spew.Dump(vms)
|
||||||
|
return vms
|
||||||
|
}
|
||||||
|
|
||||||
|
func processGetEvent(ge *pb.Event) *pb.Event {
|
||||||
|
log.Println("processGetEvent() START")
|
||||||
|
log.Println("processGetEvent() ge.Token =", ge.Token)
|
||||||
|
|
||||||
|
e := pb.Event{}
|
||||||
|
|
||||||
|
networks := fetchNetworks(ge.Token)
|
||||||
|
for i, network := range(networks) {
|
||||||
|
if i >= MAX_NETS {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
log.Println("network.Subdomain =",network.Subdomain)
|
||||||
|
log.Println("network=",network)
|
||||||
|
enetwork := pb.Event_Network{}
|
||||||
|
enetwork.Name = network.Subdomain
|
||||||
|
|
||||||
|
e.Networks = append(e.Networks, &enetwork)
|
||||||
|
|
||||||
|
vms := fetchVMs(ge.Token, network.Subdomain)
|
||||||
|
for j, vm := range(vms) {
|
||||||
|
if j >= MAX_VMS {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
evm := pb.Event_VM{}
|
||||||
|
evm.Name = vm.Name
|
||||||
|
evm.Cpus = 1
|
||||||
|
evm.Memory = 512
|
||||||
|
evm.Disk = 20
|
||||||
|
evm.Hostname = vm.Name + "." + enetwork.Name + ".lab.wit.com"
|
||||||
|
e.Vms = append(e.Vms, &evm)
|
||||||
|
|
||||||
|
log.Println("\tvm.Name =",vm.Name)
|
||||||
|
log.Println("\tvm =",vm)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return &e
|
||||||
|
}
|
||||||
|
|
||||||
|
func processLoginEvent(ge *pb.Event) *pb.Event {
|
||||||
|
log.Println("processLoginEvent() START")
|
||||||
|
log.Println("processLoginEvent() ge.Token =", ge.Token)
|
||||||
|
|
||||||
|
e := pb.Event{}
|
||||||
|
|
||||||
|
log.Println("processLoginEvent() NEED TO LOGIN HERE OR CHECK TOKEN WORKS")
|
||||||
|
|
||||||
|
log.Println("processLoginEvent() END")
|
||||||
|
return &e
|
||||||
|
}
|
||||||
|
|
||||||
// Also potentially look at:
|
// Also potentially look at:
|
||||||
// https://www.reddit.com/r/golang/comments/boznnz/an_alternative_to_gorilla_sessions_jeff_manage/
|
// https://www.reddit.com/r/golang/comments/boznnz/an_alternative_to_gorilla_sessions_jeff_manage/
|
||||||
|
|
@ -75,6 +208,11 @@ func eventHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
e := processGetEvent(pdata)
|
e := processGetEvent(pdata)
|
||||||
sendProtobuf(conn, e)
|
sendProtobuf(conn, e)
|
||||||
}
|
}
|
||||||
|
if pdata.Type == pb.Event_LOGIN {
|
||||||
|
log.Printf("GOT LOGIN")
|
||||||
|
e := processLoginEvent(pdata)
|
||||||
|
sendProtobuf(conn, e)
|
||||||
|
}
|
||||||
if pdata.Type == pb.Event_GET {
|
if pdata.Type == pb.Event_GET {
|
||||||
log.Printf("GOT GET")
|
log.Printf("GOT GET")
|
||||||
e := processGetEvent(pdata)
|
e := processGetEvent(pdata)
|
||||||
|
|
|
||||||
|
|
@ -1,132 +0,0 @@
|
||||||
package main
|
|
||||||
|
|
||||||
import "net/http"
|
|
||||||
import "encoding/json"
|
|
||||||
import "io/ioutil"
|
|
||||||
import "log"
|
|
||||||
|
|
||||||
import "github.com/davecgh/go-spew/spew"
|
|
||||||
|
|
||||||
// import "github.com/golang/protobuf/proto"
|
|
||||||
import pb "git.wit.com/wit/witProtobuf"
|
|
||||||
|
|
||||||
const (
|
|
||||||
API = "http://stackapi:4000/"
|
|
||||||
MAX_NETS = 10
|
|
||||||
MAX_VMS = 10
|
|
||||||
)
|
|
||||||
|
|
||||||
type network struct {
|
|
||||||
Subdomain string `json:"subdomain"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func fetchNetworks(token string) []network {
|
|
||||||
var networks []network
|
|
||||||
req, err := http.NewRequest("GET", API + "clusters", nil)
|
|
||||||
if err != nil { log.Fatal("can't make new req") }
|
|
||||||
req.Header.Set("X-Wit-Auth", token)
|
|
||||||
client := &http.Client{}
|
|
||||||
resp, err := client.Do(req)
|
|
||||||
if err != nil { log.Printf("Can't exec the req to list networks: %s", err) }
|
|
||||||
body, err := ioutil.ReadAll(resp.Body)
|
|
||||||
if err != nil { log.Fatal("can't read resp") }
|
|
||||||
log.Printf("NETWORKS =", body)
|
|
||||||
if body == nil {
|
|
||||||
log.Println("networks is nil")
|
|
||||||
} else {
|
|
||||||
len := len(body)
|
|
||||||
err = json.Unmarshal(body, &networks)
|
|
||||||
if err != nil {
|
|
||||||
log.Println("body = ", body)
|
|
||||||
log.Println("body len = ", len)
|
|
||||||
log.Println("can't unmarshal resp for list networks: %s", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
spew.Dump(networks)
|
|
||||||
return networks
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
{
|
|
||||||
"name": "vm10",
|
|
||||||
"cpu": 4,
|
|
||||||
"ram": 2048,
|
|
||||||
"created_on": "2019-03-18T19:22:10.755428Z",
|
|
||||||
"scope": "private",
|
|
||||||
"role": "k8s-node",
|
|
||||||
"stage": "Creating...",
|
|
||||||
"seconds_left": 300,
|
|
||||||
"base_image": "debcore_guest_latest",
|
|
||||||
"address_list": [
|
|
||||||
"192.168.0.1/32"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
|
|
||||||
|
|
||||||
*/
|
|
||||||
|
|
||||||
type address struct {
|
|
||||||
Address string
|
|
||||||
}
|
|
||||||
|
|
||||||
type vm struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
Role string `json:"role"`
|
|
||||||
BaseImage string `json:"base_image"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func fetchVMs(token string, network string) []vm {
|
|
||||||
var vms []vm
|
|
||||||
req, err := http.NewRequest("GET", API + "vms/" + network, nil)
|
|
||||||
if err != nil { log.Fatal("can't make new req") }
|
|
||||||
req.Header.Set("X-Wit-Auth", token)
|
|
||||||
client := &http.Client{}
|
|
||||||
resp, err := client.Do(req)
|
|
||||||
if err != nil { log.Printf("Can't exec the req to fetch vms for %s: %s", network, err) }
|
|
||||||
body, err := ioutil.ReadAll(resp.Body)
|
|
||||||
if err != nil { log.Fatal("can't read resp") }
|
|
||||||
err = json.Unmarshal(body, &vms)
|
|
||||||
if err != nil { log.Fatal("can't unmarshal resp for fetch vms: ", err, "\n\n", network, string(body)) }
|
|
||||||
spew.Dump(vms)
|
|
||||||
return vms
|
|
||||||
}
|
|
||||||
|
|
||||||
func processGetEvent(ge *pb.Event) *pb.Event {
|
|
||||||
log.Println("processGetEvent() START")
|
|
||||||
log.Println("processGetEvent() ge.Token =", ge.Token)
|
|
||||||
|
|
||||||
e := pb.Event{}
|
|
||||||
|
|
||||||
e.Name = "stablesid.org"
|
|
||||||
|
|
||||||
networks := fetchNetworks(ge.Token)
|
|
||||||
for i, network := range(networks) {
|
|
||||||
if i >= MAX_NETS {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
log.Println("network.Subdomain =",network.Subdomain)
|
|
||||||
log.Println("network=",network)
|
|
||||||
enetwork := pb.Event_Network{}
|
|
||||||
enetwork.Name = network.Subdomain
|
|
||||||
|
|
||||||
e.Networks = append(e.Networks, &enetwork)
|
|
||||||
|
|
||||||
vms := fetchVMs(ge.Token, network.Subdomain)
|
|
||||||
for j, vm := range(vms) {
|
|
||||||
if j >= MAX_VMS {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
evm := pb.Event_VM{}
|
|
||||||
evm.Name = vm.Name
|
|
||||||
evm.Cpus = 1
|
|
||||||
evm.Memory = 512
|
|
||||||
evm.Disk = 20
|
|
||||||
evm.Hostname = vm.Name + "." + enetwork.Name + ".lab.wit.com"
|
|
||||||
e.Vms = append(e.Vms, &evm)
|
|
||||||
|
|
||||||
log.Println("\tvm.Name =",vm.Name)
|
|
||||||
log.Println("\tvm =",vm)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return &e
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue