mirror of
https://github.com/UbiquitousIntelligentSystemBINUS/SleepMonitoringSystemUsingEdgeCache.git
synced 2026-07-11 06:58:12 +07:00
Add files via upload
This commit is contained in:
dije07
17
quantification/go.mod
Normal file
17
quantification/go.mod
Normal file
@@ -0,0 +1,17 @@
|
||||
module github.com/dije07/sqqs
|
||||
|
||||
go 1.21.3
|
||||
|
||||
require (
|
||||
github.com/eclipse/paho.mqtt.golang v1.4.3
|
||||
github.com/go-redis/redis/v8 v8.11.5
|
||||
github.com/lib/pq v1.10.9
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/cespare/xxhash/v2 v2.1.2 // indirect
|
||||
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
|
||||
github.com/gorilla/websocket v1.5.0 // indirect
|
||||
golang.org/x/net v0.8.0 // indirect
|
||||
golang.org/x/sync v0.1.0 // indirect
|
||||
)
|
||||
32
quantification/go.sum
Normal file
32
quantification/go.sum
Normal file
@@ -0,0 +1,32 @@
|
||||
github.com/cespare/xxhash/v2 v2.1.2 h1:YRXhKfTDauu4ajMg1TPgFO5jnlC2HCbmLXMcTG5cbYE=
|
||||
github.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78=
|
||||
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc=
|
||||
github.com/eclipse/paho.mqtt.golang v1.4.3 h1:2kwcUGn8seMUfWndX0hGbvH8r7crgcJguQNCyp70xik=
|
||||
github.com/eclipse/paho.mqtt.golang v1.4.3/go.mod h1:CSYvoAlsMkhYOXh/oKyxa8EcBci6dVkLCbo5tTC1RIE=
|
||||
github.com/fsnotify/fsnotify v1.4.9 h1:hsms1Qyu0jgnwNXIxa+/V/PDsU6CfLf6CNO8H7IWoS4=
|
||||
github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ=
|
||||
github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI=
|
||||
github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo=
|
||||
github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
|
||||
github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
|
||||
github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
|
||||
github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=
|
||||
github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE=
|
||||
github.com/nxadm/tail v1.4.8/go.mod h1:+ncqLTQzXmGhMZNUePPaPqPvBxHAIsmXswZKocGu+AU=
|
||||
github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE=
|
||||
github.com/onsi/ginkgo v1.16.5/go.mod h1:+E8gABHa3K6zRBolWtd+ROzc/U5bkGt0FwiG042wbpU=
|
||||
github.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE=
|
||||
github.com/onsi/gomega v1.18.1/go.mod h1:0q+aL8jAiMXy9hbwj2mr5GziHiwhAIQpFmmtT5hitRs=
|
||||
golang.org/x/net v0.8.0 h1:Zrh2ngAOFYneWTAIAPethzeaQLuHwhuBkuV6ZiRnUaQ=
|
||||
golang.org/x/net v0.8.0/go.mod h1:QVkue5JL9kW//ek3r6jTKnTFis1tRmNAW2P1shuFdJc=
|
||||
golang.org/x/sync v0.1.0 h1:wsuoTGHzEhffawBOhz5CYhcrV4IdKZbEyZjBMuTp12o=
|
||||
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sys v0.6.0 h1:MVltZSvRTcU2ljQOhs94SXPftV6DCNnZViHeQps87pQ=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/text v0.8.0 h1:57P1ETyNKtuIjB4SRd15iJxuhj8Gc416Y78H3qgMh68=
|
||||
golang.org/x/text v0.8.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
|
||||
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=
|
||||
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
|
||||
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
||||
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
||||
390
quantification/quantification.go
Normal file
390
quantification/quantification.go
Normal file
@@ -0,0 +1,390 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"math"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/go-redis/redis/v8"
|
||||
_ "github.com/lib/pq"
|
||||
|
||||
MQTT "github.com/eclipse/paho.mqtt.golang"
|
||||
)
|
||||
|
||||
// Define your MQTT broker address and port
|
||||
const (
|
||||
brokerAddress = "ws://localhost:9001" // Change this to your MQTT broker address
|
||||
host = "localhost"
|
||||
port = 5432
|
||||
user = "postgres"
|
||||
password = "postgres"
|
||||
dbname = "sleep_monitoring"
|
||||
)
|
||||
|
||||
var DB *sql.DB
|
||||
|
||||
type Message struct {
|
||||
Event string `json:"event"`
|
||||
Data interface{} `json:"data"`
|
||||
}
|
||||
|
||||
type SleepQuality struct {
|
||||
ID uint `gorm:"primaryKey" json:"id"`
|
||||
Value string `json:"value,omitempty"`
|
||||
DeepSleepDuration float64 `json:"deep_sleep_duration,omitempty"`
|
||||
AwakeDuration float64 `json:"awake_duration,omitempty"`
|
||||
TotalSleepDuration float64 `json:"total_sleep_duration,omitempty"`
|
||||
InputTime time.Time `json:"input_time,omitempty"`
|
||||
UserId uint `json:"user_id,omitempty"`
|
||||
}
|
||||
|
||||
// FuzzySet represents a fuzzy set with a membership function.
|
||||
type FuzzySet struct {
|
||||
Name string
|
||||
Terms map[string]func(float64) float64
|
||||
}
|
||||
|
||||
// Fuzzify function calculates the membership values for each term in the fuzzy set.
|
||||
func (fs *FuzzySet) Fuzzify(value float64) string {
|
||||
max := 0.0
|
||||
result := ""
|
||||
|
||||
for term, mf := range fs.Terms {
|
||||
membership := mf(value)
|
||||
if membership > max {
|
||||
max = membership
|
||||
result = term
|
||||
}
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
// ApplyRule function evaluates the conditions and returns the corresponding action.
|
||||
func ApplyRule(conditions map[string]float64) (string, bool) {
|
||||
// Apply all rules
|
||||
switch {
|
||||
case conditions["awakeDuration"] == 1 && conditions["deepSleepTime"] == 1 && conditions["totalSleepTime"] == 1:
|
||||
return "LEVEL 9", true
|
||||
case conditions["awakeDuration"] == 1 && conditions["deepSleepTime"] == 1 && conditions["totalSleepTime"] == 0.5:
|
||||
return "LEVEL 8", true
|
||||
case conditions["awakeDuration"] == 1 && conditions["deepSleepTime"] == 1 && conditions["totalSleepTime"] == 0:
|
||||
return "LEVEL 7", true
|
||||
case conditions["awakeDuration"] == 1 && conditions["deepSleepTime"] == 0.5 && conditions["totalSleepTime"] == 1:
|
||||
return "LEVEL 7", true
|
||||
case conditions["awakeDuration"] == 1 && conditions["deepSleepTime"] == 0.5 && conditions["totalSleepTime"] == 0.5:
|
||||
return "LEVEL 6", true
|
||||
case conditions["awakeDuration"] == 1 && conditions["deepSleepTime"] == 0.5 && conditions["totalSleepTime"] == 0:
|
||||
return "LEVEL 5", true
|
||||
case conditions["awakeDuration"] == 1 && conditions["deepSleepTime"] == 0 && conditions["totalSleepTime"] == 1:
|
||||
return "LEVEL 5", true
|
||||
case conditions["awakeDuration"] == 1 && conditions["deepSleepTime"] == 0 && conditions["totalSleepTime"] == 0.5:
|
||||
return "LEVEL 4", true
|
||||
case conditions["awakeDuration"] == 1 && conditions["deepSleepTime"] == 0 && conditions["totalSleepTime"] == 0:
|
||||
return "LEVEL 3", true
|
||||
case conditions["awakeDuration"] == 0.5 && conditions["deepSleepTime"] == 1 && conditions["totalSleepTime"] == 1:
|
||||
return "LEVEL 8", true
|
||||
case conditions["awakeDuration"] == 0.5 && conditions["deepSleepTime"] == 1 && conditions["totalSleepTime"] == 0.5:
|
||||
return "LEVEL 7", true
|
||||
case conditions["awakeDuration"] == 0.5 && conditions["deepSleepTime"] == 1 && conditions["totalSleepTime"] == 0:
|
||||
return "LEVEL 6", true
|
||||
case conditions["awakeDuration"] == 0.5 && conditions["deepSleepTime"] == 0.5 && conditions["totalSleepTime"] == 1:
|
||||
return "LEVEL 6", true
|
||||
case conditions["awakeDuration"] == 0.5 && conditions["deepSleepTime"] == 0.5 && conditions["totalSleepTime"] == 0.5:
|
||||
return "LEVEL 5", true
|
||||
case conditions["awakeDuration"] == 0.5 && conditions["deepSleepTime"] == 0.5 && conditions["totalSleepTime"] == 0:
|
||||
return "LEVEL 4", true
|
||||
case conditions["awakeDuration"] == 0.5 && conditions["deepSleepTime"] == 0 && conditions["totalSleepTime"] == 1:
|
||||
return "LEVEL 4", true
|
||||
case conditions["awakeDuration"] == 0.5 && conditions["deepSleepTime"] == 0 && conditions["totalSleepTime"] == 0.5:
|
||||
return "LEVEL 3", true
|
||||
case conditions["awakeDuration"] == 0.5 && conditions["deepSleepTime"] == 0 && conditions["totalSleepTime"] == 0:
|
||||
return "LEVEL 2", true
|
||||
case conditions["awakeDuration"] == 0 && conditions["deepSleepTime"] == 1 && conditions["totalSleepTime"] == 1:
|
||||
return "LEVEL 7", true
|
||||
case conditions["awakeDuration"] == 0 && conditions["deepSleepTime"] == 1 && conditions["totalSleepTime"] == 0.5:
|
||||
return "LEVEL 6", true
|
||||
case conditions["awakeDuration"] == 0 && conditions["deepSleepTime"] == 1 && conditions["totalSleepTime"] == 0:
|
||||
return "LEVEL 5", true
|
||||
case conditions["awakeDuration"] == 0 && conditions["deepSleepTime"] == 0.5 && conditions["totalSleepTime"] == 1:
|
||||
return "LEVEL 5", true
|
||||
case conditions["awakeDuration"] == 0 && conditions["deepSleepTime"] == 0.5 && conditions["totalSleepTime"] == 0.5:
|
||||
return "LEVEL 4", true
|
||||
case conditions["awakeDuration"] == 0 && conditions["deepSleepTime"] == 0.5 && conditions["totalSleepTime"] == 0:
|
||||
return "LEVEL 3", true
|
||||
case conditions["awakeDuration"] == 0 && conditions["deepSleepTime"] == 0 && conditions["totalSleepTime"] == 1:
|
||||
return "LEVEL 3", true
|
||||
case conditions["awakeDuration"] == 0 && conditions["deepSleepTime"] == 0 && conditions["totalSleepTime"] == 0.5:
|
||||
return "LEVEL 2", true
|
||||
case conditions["awakeDuration"] == 0 && conditions["deepSleepTime"] == 0 && conditions["totalSleepTime"] == 0:
|
||||
return "LEVEL 1", true
|
||||
}
|
||||
|
||||
// No rule matched
|
||||
return "", false
|
||||
}
|
||||
|
||||
var messagePubHandler MQTT.MessageHandler = func(client MQTT.Client, msg MQTT.Message) {
|
||||
fmt.Printf("Message Received from [%s] : %s\n", msg.Topic(), msg.Payload())
|
||||
|
||||
if string(msg.Payload()) == "sleepStageReady" {
|
||||
quantifyData()
|
||||
}
|
||||
}
|
||||
|
||||
func setUpDB() *sql.DB {
|
||||
// Construct the connection string
|
||||
connStr := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=disable",
|
||||
os.Getenv("DB_HOST"), os.Getenv("DB_PORT"),
|
||||
os.Getenv("DB_USER"), os.Getenv("DB_PASSWORD"),
|
||||
os.Getenv("DB_NAME"))
|
||||
|
||||
// Open a connection to the database
|
||||
db, err := sql.Open("postgres", connStr)
|
||||
if err != nil {
|
||||
log.Fatal("Error connecting to the database: ", err)
|
||||
}
|
||||
|
||||
// Attempt to ping the database to verify connection
|
||||
err = db.Ping()
|
||||
if err != nil {
|
||||
log.Fatal("Error pinging the database: ", err)
|
||||
}
|
||||
|
||||
fmt.Println("Successfully connected to the database!")
|
||||
|
||||
return db
|
||||
}
|
||||
|
||||
var red *redis.Client
|
||||
|
||||
func quantifyData() {
|
||||
|
||||
var awake float64 = 0
|
||||
var deepSleep float64 = 0
|
||||
var totalSleep float64 = 0
|
||||
|
||||
db := setUpDB()
|
||||
DB = db
|
||||
value, err := red.Get(red.Context(), "sleep-stages").Result()
|
||||
if value == "" || err != nil {
|
||||
// Example query
|
||||
start := time.Now().UnixMilli()
|
||||
rows, err := DB.Query("SELECT * FROM sleep_stages")
|
||||
end := time.Now().UnixMilli()
|
||||
fmt.Println("Start : ", start)
|
||||
fmt.Println("End : ", end)
|
||||
fmt.Println("Duration : ", end-start, "ms")
|
||||
if err != nil {
|
||||
log.Fatal("Error querying the database: ", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
// Iterate over the rows
|
||||
for rows.Next() {
|
||||
var id int
|
||||
var reference_id string
|
||||
var value string
|
||||
var method any
|
||||
var timestamp time.Time
|
||||
err := rows.Scan(&id, &reference_id, &value, &method, ×tamp)
|
||||
if err != nil {
|
||||
log.Println("Error scanning row: ", err)
|
||||
}
|
||||
if value == "N1" || value == "N2" || value == "REM" {
|
||||
totalSleep++
|
||||
}
|
||||
if value == "N3" {
|
||||
totalSleep++
|
||||
deepSleep++
|
||||
}
|
||||
if value == "AWAKE" {
|
||||
awake++
|
||||
}
|
||||
}
|
||||
if err := rows.Err(); err != nil {
|
||||
log.Println("Error iterating over rows: ", err)
|
||||
}
|
||||
} else {
|
||||
type SleepStage struct {
|
||||
ID uint `json:"id"`
|
||||
ReferenceID uint `json:"reference_id"`
|
||||
Value string `json:"value"`
|
||||
Method string `json:"method"`
|
||||
Timestamp time.Time `json:"timestamp"`
|
||||
}
|
||||
|
||||
type SleepStages struct {
|
||||
Stages []SleepStage `json:"sleep-stages"`
|
||||
}
|
||||
|
||||
// Unmarshal the JSON into a SleepStages struct
|
||||
var stages SleepStages
|
||||
err := json.Unmarshal([]byte(value), &stages)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
// Loop through the stages and print values
|
||||
for _, stage := range stages.Stages {
|
||||
if stage.Value == "N1" || stage.Value == "N2" || stage.Value == "REM" {
|
||||
totalSleep++
|
||||
}
|
||||
if stage.Value == "N3" {
|
||||
totalSleep++
|
||||
deepSleep++
|
||||
}
|
||||
if stage.Value == "AWAKE" {
|
||||
awake++
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Define fuzzy sets and input values
|
||||
awakeDurationSet := FuzzySet{
|
||||
Name: "awakeDuration",
|
||||
Terms: map[string]func(float64) float64{
|
||||
"low": func(x float64) float64 { return math.Max(0, 1.0-0.5*x) },
|
||||
"medium": func(x float64) float64 { return math.Max(0, 1-2*math.Abs(x-0.5)) },
|
||||
"high": func(x float64) float64 { return math.Max(0, 2*x-1) },
|
||||
},
|
||||
}
|
||||
deepSleepDurationSet := FuzzySet{
|
||||
Name: "awakeDuration",
|
||||
Terms: map[string]func(float64) float64{
|
||||
"low": func(x float64) float64 { return math.Max(0, 1.0-0.5*x) },
|
||||
"medium": func(x float64) float64 { return math.Max(0, 1-2*math.Abs(x-0.5)) },
|
||||
"high": func(x float64) float64 { return math.Max(0, 2*x-1) },
|
||||
},
|
||||
}
|
||||
totalSleepDurationSet := FuzzySet{
|
||||
Name: "awakeDuration",
|
||||
Terms: map[string]func(float64) float64{
|
||||
"low": func(x float64) float64 { return math.Max(0, 1.0-0.5*x) },
|
||||
"medium": func(x float64) float64 { return math.Max(0, 1-2*math.Abs(x-0.5)) },
|
||||
"high": func(x float64) float64 { return math.Max(0, 2*x-1) },
|
||||
},
|
||||
}
|
||||
|
||||
inputValues := map[string]float64{
|
||||
"awakeDuration": awake / 60,
|
||||
"deepSleepTime": deepSleep / 60,
|
||||
"totalSleepTime": totalSleep / 60,
|
||||
}
|
||||
|
||||
// Fuzzify the sleep data
|
||||
awakeTerm := awakeDurationSet.Fuzzify(inputValues["awakeDuration"])
|
||||
deepSleepTerm := deepSleepDurationSet.Fuzzify(inputValues["deepSleepTime"])
|
||||
totalSleepTerm := totalSleepDurationSet.Fuzzify(inputValues["totalSleepTime"])
|
||||
// Fuzzify the other sleep data...
|
||||
|
||||
// Print the intermediate results
|
||||
fmt.Println("Fuzzy Conditions:")
|
||||
fmt.Printf("Awake Duration: %s (%f)\n", awakeTerm, inputValues["awakeDuration"])
|
||||
fmt.Printf("Deep Sleep Duration: %s (%f)\n", deepSleepTerm, inputValues["deepSleepTime"])
|
||||
fmt.Printf("Total Sleep Duration: %s (%f)\n", totalSleepTerm, inputValues["totalSleepTime"])
|
||||
// Print the other fuzzy conditions...
|
||||
|
||||
// Map the linguistic terms to their respective numerical values
|
||||
linguisticValues := map[string]float64{
|
||||
"high": 1.0,
|
||||
"medium": 0.5,
|
||||
"low": 0.0,
|
||||
}
|
||||
|
||||
// Create a map for all conditions
|
||||
allConditions := map[string]float64{
|
||||
"awakeDuration": linguisticValues[awakeTerm],
|
||||
"deepSleepTime": linguisticValues[deepSleepTerm],
|
||||
"totalSleepTime": linguisticValues[totalSleepTerm],
|
||||
}
|
||||
|
||||
action, ruleMatched := ApplyRule(allConditions)
|
||||
|
||||
// Print the final results
|
||||
if ruleMatched {
|
||||
fmt.Println("\nRule Matched! Action:", action)
|
||||
} else {
|
||||
fmt.Println("\nNo Rule Matched.")
|
||||
}
|
||||
|
||||
sleepQuality := SleepQuality{
|
||||
Value: action,
|
||||
DeepSleepDuration: inputValues["deepSleepTime"],
|
||||
TotalSleepDuration: inputValues["totalSleepTime"],
|
||||
AwakeDuration: inputValues["awakeDuration"],
|
||||
InputTime: time.Now(),
|
||||
UserId: 1,
|
||||
}
|
||||
|
||||
sleepQualityJSON, err := json.Marshal(sleepQuality)
|
||||
if err != nil {
|
||||
fmt.Println("Error:", err)
|
||||
return
|
||||
}
|
||||
|
||||
err = red.Set(red.Context(), "sleep_quality_1", sleepQualityJSON, 0).Err()
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to save to redis: %v", err)
|
||||
} else {
|
||||
fmt.Println("Saved to redis")
|
||||
}
|
||||
|
||||
// Prepare the SQL statement for inserting data.
|
||||
stmt, err := db.Prepare("INSERT INTO sleep_qualities (value, deep_sleep_duration, awake_duration, total_sleep_duration, input_time, user_id) VALUES ($1, $2, $3, $4, $5, $6)")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
defer stmt.Close()
|
||||
|
||||
// Example data to be inserted.
|
||||
value1 := action
|
||||
value2 := inputValues["deepSleepTime"]
|
||||
value3 := inputValues["awakeDuration"]
|
||||
value4 := inputValues["totalSleepTime"]
|
||||
value5 := time.Now()
|
||||
value6 := 1
|
||||
|
||||
// Execute the SQL statement with the provided values.
|
||||
_, err = stmt.Exec(value1, value2, value3, value4, value5, value6)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func main() {
|
||||
red = redis.NewClient(&redis.Options{
|
||||
Addr: os.Getenv("REDIS_ADDR"), // Redis server address
|
||||
Password: "", // no password set
|
||||
DB: 0, // use default DB
|
||||
})
|
||||
|
||||
// Set MQTT client options
|
||||
|
||||
opts := MQTT.NewClientOptions().AddBroker("ws://" + os.Getenv("MQTT_BROKER"))
|
||||
opts.SetClientID("MQTT_simulator")
|
||||
opts.SetDefaultPublishHandler(messagePubHandler)
|
||||
|
||||
topic := "sleep_monitoring"
|
||||
|
||||
// Create and start a client using the above options
|
||||
client := MQTT.NewClient(opts)
|
||||
if token := client.Connect(); token.Wait() && token.Error() != nil {
|
||||
fmt.Println("Error connecting to MQTT broker:", token.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// Subscribe to MQTT topics
|
||||
if token := client.Subscribe(topic, 0, messagePubHandler); token.Wait() && token.Error() != nil {
|
||||
fmt.Println("Error subscribing to MQTT topic:", token.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
fmt.Printf("Subscribed to topic: %s\n", topic)
|
||||
// Keep the program running indefinitely
|
||||
select {}
|
||||
}
|
||||
Reference in New Issue
Block a user