added webserver to serve json file. also running indefinitely now
This commit is contained in:
parent
94222942e6
commit
78b274c160
46
main.go
46
main.go
@ -23,18 +23,27 @@ const (
|
|||||||
// flags
|
// flags
|
||||||
var token = flag.String("token", "", "Defines the x-auth-token")
|
var token = flag.String("token", "", "Defines the x-auth-token")
|
||||||
var version = "development"
|
var version = "development"
|
||||||
|
var delay time.Duration = 60 * time.Second
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
log.Printf("starting version %s...\n", version)
|
||||||
log.Println("starting version", version, "...")
|
|
||||||
// parse all flags
|
// parse all flags
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
// check if token is set
|
// check if token is set
|
||||||
if *token == "" {
|
if *token == "" {
|
||||||
log.Fatalln("Please specify an API token via the flag '-token'")
|
log.Fatalln("Please specify an API token via the flag '-token'")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// start API processing (runs in a loop)
|
||||||
|
go processAPIs()
|
||||||
|
|
||||||
|
// start webserver
|
||||||
|
serveJSON()
|
||||||
|
}
|
||||||
|
|
||||||
|
func processAPIs() {
|
||||||
|
tick := time.Tick(delay)
|
||||||
|
for range tick {
|
||||||
// Variables for runtime
|
// Variables for runtime
|
||||||
var links []link
|
var links []link
|
||||||
var nodes []node
|
var nodes []node
|
||||||
@ -131,10 +140,14 @@ func main() {
|
|||||||
|
|
||||||
// create file output
|
// create file output
|
||||||
log.Println("writing json file")
|
log.Println("writing json file")
|
||||||
writeJSONtoFile(o)
|
|
||||||
|
if err := o.writeToFile(); err != nil {
|
||||||
|
log.Fatalln(err)
|
||||||
|
}
|
||||||
|
|
||||||
// we're done here
|
// we're done here
|
||||||
log.Println("...done")
|
log.Println("...done")
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func getDevices() (devices, error) {
|
func getDevices() (devices, error) {
|
||||||
@ -213,22 +226,19 @@ func addLink(dev unifiAPIResponse, airmaxes unifiAPIAirmax, links []link) []link
|
|||||||
return links
|
return links
|
||||||
}
|
}
|
||||||
|
|
||||||
func writeJSONtoFile(o output) error {
|
|
||||||
file, err := json.MarshalIndent(o, "", " ")
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("can't marshal to json: %+v", o)
|
|
||||||
}
|
|
||||||
|
|
||||||
// write to file
|
|
||||||
err = ioutil.WriteFile("example.json", file, 0644)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("can't write to json file example.json")
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func getAddresses(ip string) []string {
|
func getAddresses(ip string) []string {
|
||||||
var adresses []string
|
var adresses []string
|
||||||
adresses = append(adresses, strings.Split(ip, "/")[0])
|
adresses = append(adresses, strings.Split(ip, "/")[0])
|
||||||
return adresses
|
return adresses
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func serveJSON() {
|
||||||
|
fs := http.FileServer(http.Dir("./output"))
|
||||||
|
http.Handle("/", fs)
|
||||||
|
|
||||||
|
log.Println("Listening on :3000...")
|
||||||
|
err := http.ListenAndServe(":3000", nil)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalln(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user