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
|
package main
import (
"fmt"
"log"
"net/http"
"net/url"
"strconv"
)
type Humidity float32
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) {
queryVals, err := parseQuery(r.URL.RawQuery, []string{"room", "humidity"})
if err != nil {
badRequest(w, "invalid query: %v", err)
return
}
room := RoomID(queryVals["room"])
humidityStr := queryVals["humidity"]
humidity, err := strconv.ParseFloat(humidityStr, 32)
if err != nil {
badRequest(w, "invalid humidity: '%s'", humidityStr)
return
}
record, ok := h.rooms[room]
if !ok {
badRequest(w, "invalid room ID: '%s'", room)
return
}
record.put <- Humidity(humidity)
}
// Calculate the average humidity in the building. Returns false if there is not enough data available.
func (h HumidityHandler) average() (Humidity, bool) {
var sum Humidity = 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 / Humidity(nRooms), true
}
// Parse the value associated with each key in the query string. Returns a map of
// keys and values, or error if one of the keys is missing or if there is no value
// associated with one of the keys.
func parseQuery(query string, keys []string) (map[string]string, error) {
queryVals, err := url.ParseQuery(query)
if err != nil {
return nil, err
}
vals := make(map[string]string)
for _, key := range keys {
val := queryVals.Get(key)
if val == "" {
return nil, fmt.Errorf("missing key '%s'", key)
}
vals[key] = val
}
return vals, nil
}
|