Separate sauna api into single module

This commit is contained in:
Thomas Klaehn
2023-03-01 08:11:55 +01:00
parent 84d84b4313
commit 4eae02cd0c
4 changed files with 92 additions and 85 deletions

85
main.go
View File

@@ -1,82 +1,21 @@
package main
import (
"encoding/json"
"flag"
"log"
"net/http"
"sync"
"time"
mqtt "github.com/eclipse/paho.mqtt.golang"
sauna "git.blackfinn.de/apiservice/sauna"
)
type temperature struct {
Value float64 `json:"value"`
Unit string `json:"unit"`
Time time.Time `json:"time"`
Valid bool `json:"valid"`
}
var (
logger log.Logger = *log.Default()
sauna_mutex sync.Mutex
sauna_temperature = temperature{
Value: 0.0,
Unit: "°C",
}
)
var messagePubHandler mqtt.MessageHandler = func(client mqtt.Client, msg mqtt.Message) {
logger.Printf("Received message: %s from topic: %s\n", msg.Payload(), msg.Topic())
}
var saunaHandler mqtt.MessageHandler = func(client mqtt.Client, msg mqtt.Message) {
log.Printf("Received message: %s from topic: %s\n", msg.Payload(), msg.Topic())
sauna_mutex.Lock()
err := json.Unmarshal(msg.Payload(), &sauna_temperature)
sauna_temperature.Time = time.Now()
if err != nil {
sauna_temperature.Valid = false
sauna_mutex.Unlock()
logger.Print(err)
return
}
sauna_temperature.Valid = true
sauna_mutex.Unlock()
}
var connectHandler mqtt.OnConnectHandler = func(client mqtt.Client) {
logger.Println("Connected")
}
var connectLostHandler mqtt.ConnectionLostHandler = func(client mqtt.Client, err error) {
logger.Printf("Connect lost: %v", err)
}
func init() {
logger.SetPrefix("Homeservice: ")
}
func http_endpoint_sauna(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-type", "application/json; charset=utf-8;")
if r.Method == http.MethodGet {
sauna_mutex.Lock()
data, err := json.Marshal(sauna_temperature)
sauna_mutex.Unlock()
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
w.Write(json.RawMessage(`{"error": "cannot marshal object to json"}`))
} else {
w.WriteHeader(http.StatusOK)
w.Write(data)
}
} else {
w.WriteHeader(http.StatusMethodNotAllowed)
}
}
func main() {
logger.Println("starting")
@@ -84,30 +23,12 @@ func main() {
flag.StringVar(&webui_path, "d", "./build/webui", "Specify path to serve the web ui. Default is ./static")
flag.Parse()
// MQTT connection
opts := mqtt.NewClientOptions()
opts.AddBroker("tcp://nuc:1883")
opts.SetClientID("homeservice")
opts.SetDefaultPublishHandler(messagePubHandler)
opts.OnConnect = connectHandler
opts.OnConnectionLost = connectLostHandler
client := mqtt.NewClient(opts)
if token := client.Connect(); token.Wait() && token.Error() != nil {
panic(token.Error())
}
// Start apis
sauna.Start()
// MQTT subscribtion
topic := "sauna/temperature"
token := client.Subscribe(topic, 1, saunaHandler)
token.Wait()
logger.Printf("Subscribed to topic %s", topic)
// API routes
// Serve files from static folder
http.Handle("/", http.FileServer(http.Dir(webui_path)))
http.HandleFunc("/api/sauna", http_endpoint_sauna)
port := ":5000"
logger.Println("Server is running on port" + port)