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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
package main
import (
"database/sql"
"flag"
"fmt"
"github.com/gorilla/mux"
"log"
"net/http"
"os"
"strconv"
"git.darknebu.la/GalaxySimulator/db-actions"
)
var (
db *sql.DB
currentStarBuffer int64 = 1
idBufferChannel = make(chan int64, 100000000)
// local http api
port string
// define the parameters needed to connect to the database
DBHOST = "postgres.docker.localhost"
DBPORT = 5432
DBUSER = "postgres"
DBPASSWD = ""
DBNAME = "postgres"
DBSSLMODE = "disable"
)
func listOfStarIDs(treeindex int64) []int64 {
log.Printf("getting a list of stars using the treeindex %d", treeindex)
// get a list of all stars inside of the treeindex
listofstars := db_actions.GetListOfStarIDsTimestep(db, treeindex)
return listofstars
}
func fillStarIdBufferChannel() {
log.Println("Filling the idBufferChannel")
// get a list of ids using the currentStarBuffer value
// the currentStarBuffer value is a counter keeping track of which galaxy is going to
// be inserted into the idBufferChannel next
listOfStarIDs := listOfStarIDs(currentStarBuffer)
log.Printf("len(listOfStarIDs: %d)", len(listOfStarIDs))
// insert all the ids from the list of ids into the idBufferChannel
for _, id := range listOfStarIDs {
idBufferChannel <- id
}
// increase the currentStarBuffer counter
currentStarBuffer += 1
}
func getFlags() {
// get the port on which the service should be hosted and the url of the database
flag.StringVar(&port, "port", "8080", "port used to host the service")
flag.Parse()
log.Println("[ ] Done loading the flags")
}
func pingDB() {
// ping the db
err := db.Ping()
if err != nil {
panic(err)
}
log.Println("[ ] Done Pinging the DB")
}
func getEnvironmentVariables() {
// get the data that should be used to connect to the database
DBHOST = os.Getenv("DBURL")
if DBHOST == "" {
DBHOST = "postgresql.docker.localhost"
}
DBUSER = os.Getenv("DBUSER")
if DBUSER == "" {
DBUSER = "postgres"
}
DBPASSWD = os.Getenv("DBPASSWD")
if DBPASSWD == "" {
DBPASSWD = ""
}
DBPORT, _ := strconv.ParseInt(os.Getenv("DBPORT"), 10, 64)
if DBPORT == 0 {
DBPORT = 5432
}
DBNAME = os.Getenv("DBNAME")
if DBNAME == "" {
DBNAME = "postgres"
}
log.Printf("DBURL: %s", DBHOST)
log.Printf("DBUSER: %s", DBUSER)
log.Printf("DBPASSWD: %s", DBPASSWD)
log.Printf("DBPORT: %d", DBPORT)
log.Printf("DBPROJECTNAME: %s", DBNAME)
log.Printf("frontend port: %s", port)
}
func main() {
getFlags()
getEnvironmentVariables()
db = connectToDB()
db.SetMaxOpenConns(75)
pingDB()
router := mux.NewRouter()
router.HandleFunc("/", indexHandler).Methods("GET")
router.HandleFunc("/distributor", distributorHandler).Methods("GET")
log.Printf("[ ] Distributor on localhost:%s", port)
log.Fatal(http.ListenAndServe(fmt.Sprintf(":%s", port), router))
}
|
