about summary refs log tree commit diff
path: root/main.go
blob: 2e527f65c6e0df28343d2202f2bfd71abd2f4e1f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package main

import (
	"fmt"
	"io/ioutil"
	"log"
	"net/http"

	"github.com/gorilla/mux"

	"git.darknebu.la/GalaxySimulator/structs"
)

var (
	treeArray  []*structs.Node
	starCount  []int
	errorCount []int
)

// indexHandler
func indexHandler(w http.ResponseWriter, r *http.Request) {
	infostring := `Galaxy Simulator Database

API:
	- / ("GET")
	- /new ("POST")
	- /printall ("GET")
	- /insert/{treeindex} ("POST")
	- /starlist/{treeindex} ("GET")
	- /dumptree/{treeindex} ("GET")

	- /updatetotalmass/{treeindex} ("GET")
	- /updatecenterofmass/{treeindex} ("GET")

	- /metrics ("GET")
	- /export/{treeindex} ("POST")

	- /fastinsertjson/{filename} ("GET")
	- /fastinsertlist/{filename} ("GET")

	- /readdir ("GET")
`
	_, _ = fmt.Fprintf(w, infostring)
}

func readdirHandler(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	dirname := vars["dirname"]
	log.Printf("Reading from %s", dirname)

	files, err := ioutil.ReadDir(fmt.Sprintf("./%s", dirname))
	log.Println(files)
	log.Println(err)
	if err != nil {
		fmt.Println(err)
	}

	for _, f := range files {
		fmt.Println(f.Name())
		_, _ = fmt.Fprintf(w, "%v", f.Name())
	}
}

// return the amount of galaxies currently present
func nrofgalaxiesHandler(w http.ResponseWriter, r *http.Request) {
	_, _ = fmt.Fprintf(w, "%d", len(treeArray))
}

func main() {
	router := mux.NewRouter()

	router.HandleFunc("/", indexHandler).Methods("GET")
	router.HandleFunc("/new", newTreeHandler).Methods("POST")
	router.HandleFunc("/printall", printAllHandler).Methods("GET")
	router.HandleFunc("/insert/{treeindex}", insertStarHandler).Methods("POST")
	router.HandleFunc("/starlist/{treeindex}", starlistHandler).Methods("GET")
	router.HandleFunc("/dumptree/{treeindex}", dumptreeHandler).Methods("GET")
	router.HandleFunc("/updatetotalmass/{treeindex}", updateTotalMassHandler).Methods("GET")
	router.HandleFunc("/updatecenterofmass/{treeindex}", updateCenterOfMassHandler).Methods("GET")
	router.HandleFunc("/metrics", metricHandler).Methods("GET")
	router.HandleFunc("/export/{treeindex}", exportHandler).Methods("POST")
	router.HandleFunc("/nrofgalaxies", nrofgalaxiesHandler).Methods("GET")

	router.HandleFunc("/fastinsertjson/{filename}", fastInsertJSONHandler).Methods("GET")
	router.HandleFunc("/fastinsertlist/{filename}/{treeindex}", fastInsertListHandler).Methods("GET")

	// router.HandleFunc("/readdir/{dirname}", readdirHandler).Methods("GET")

	fmt.Println("Database Container up")
	log.Fatal(http.ListenAndServe(":80", router))
}