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
|
package main
import (
"fmt"
"log"
"net/http"
"strconv"
)
const addr = ":9090"
var rooms = []RoomID {
",4AL[+V*:*k*n{7vL{}/d=K#Mo*y*^.@",
"Jq!+<p3g-iu%-vU]FZp2H,AKZWp@!4![",
}
type Humidity float32
type RoomID string
type HumidityHandler struct {
rooms map[RoomID]Record[Humidity]
}
func newHumidityHandler(rooms []RoomID) HumidityHandler {
h := HumidityHandler{make(map[RoomID]Record[Humidity])}
for _, id := range rooms {
h.rooms[id] = newRecord[Humidity]()
}
return h
}
func (h HumidityHandler) Close() {
for _, record := range h.rooms {
record.Close()
}
}
func (h HumidityHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
log.Println(r.Method, r.URL)
switch r.Method {
case http.MethodGet:
h.get(w, r)
case http.MethodPost:
h.post(w, r)
default:
w.WriteHeader(http.StatusMethodNotAllowed)
fmt.Fprintf(w, "invalid method: '%s'", r.Method)
}
}
func (h HumidityHandler) get(w http.ResponseWriter, r *http.Request) {
if humidity, ok := h.average(); ok {
fmt.Fprintf(w, "%.2f", humidity)
} else {
w.WriteHeader(http.StatusGone)
fmt.Fprintf(w, "no humidity data stored on server")
}
}
func (h HumidityHandler) post(w http.ResponseWriter, r *http.Request) {
query := r.URL.RawQuery
humidity, err := strconv.ParseFloat(query, 32)
if err != nil {
w.WriteHeader(http.StatusBadRequest)
fmt.Fprintf(w, "invalid query string: '%s'", query)
return
}
h.humidity.put <- float32(humidity)
}
// Calculate the average humidity in the building. Returns false if there is not enough data available.
func (h HumidityHandler) average() (Humidity, bool) {
sum := 0
nRooms := 0
for room, record := range h.rooms {
c := make(chan Humidity)
record.getRecent() <- c
if humidity, ok := <-c; ok {
sum += humidity
nRooms++
} else {
log.Printf("Warning: no humidity for room '%s'\n", room)
}
}
if nRooms == 0 {
log.Println("Warning: not enough data to calculate average humidity")
return -1.0, false
}
return sum/nRooms, true
}
func main() {
humidityHandler := newHumidityHandler()
defer humidityHandler.Close()
http.Handle("/humidity", humidityHandler)
fmt.Printf("Listening on %s...\n", addr)
log.Fatal(http.ListenAndServe(addr, nil))
}
|