mirror of
https://github.com/planetmint/planetmint-go.git
synced 2025-03-30 15:08:28 +00:00
e99b614e4a
* added a bit of logging * added update messages to mqtt (every second with a timestamp) * added isConnectionOpen to the interface and analysis this during execution * added connection loss handler and status check * put some global objects into an objects to avoid inter service testing issues * TestingLocker Mutex to RWMutex (increase performance) * termintationMutex became a sync.RWMutex * added logging to the mqtt mock client * made monitor Mutex a RWMutex * added Mutex protection to the numberOfElements varialbe * added another Waiting block to the machine attestation methods (CI tests) * had to adjust the test cases to the impact of that change. Signed-off-by: Jürgen Eckel <juergen@riddleandcode.com>
112 lines
3.2 KiB
Go
112 lines
3.2 KiB
Go
package monitor
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/syndtr/goleveldb/leveldb/iterator"
|
|
)
|
|
|
|
type LastSeenEvent struct {
|
|
Address string `binding:"required" json:"address"`
|
|
Timestamp int64 `binding:"required" json:"timestamp"`
|
|
}
|
|
|
|
func (mms *MqttMonitor) AddParticipant(address string, lastSeenTS int64) (err error) {
|
|
// store receive address - planetmint address pair
|
|
var lastSeen LastSeenEvent
|
|
lastSeen.Address = address
|
|
lastSeen.Timestamp = lastSeenTS
|
|
|
|
lastSeenBytes, err := json.Marshal(lastSeen)
|
|
if err != nil {
|
|
log.Println("[app] [Monitor] Error serializing ConversionRequest: " + err.Error())
|
|
return
|
|
}
|
|
|
|
// returns an error if the entry does not exist (we have to increase the counter in this case)
|
|
_, err = mms.db.Get([]byte(address), nil)
|
|
if err != nil {
|
|
mms.setNumDBElements(mms.getNumDBElements() + 1)
|
|
}
|
|
mms.dbMutex.Lock()
|
|
err = mms.db.Put([]byte(address), lastSeenBytes, nil)
|
|
mms.dbMutex.Unlock()
|
|
if err != nil {
|
|
log.Println("[app] [Monitor] error storing addresses in DB: " + err.Error())
|
|
} else {
|
|
log.Println("[app] [Monitor] stored address in DB: " + address)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (mms *MqttMonitor) deleteEntry(key []byte) (err error) {
|
|
mms.setNumDBElements(mms.getNumDBElements() - 1)
|
|
mms.dbMutex.Lock()
|
|
err = mms.db.Delete(key, nil)
|
|
mms.dbMutex.Unlock()
|
|
return
|
|
}
|
|
|
|
func (mms *MqttMonitor) getAmountOfElements() (amount int64, err error) {
|
|
iter := mms.db.NewIterator(nil, nil)
|
|
defer iter.Release()
|
|
|
|
for iter.Next() {
|
|
amount++
|
|
}
|
|
|
|
// Check for any errors encountered during iteration
|
|
if err := iter.Error(); err != nil {
|
|
log.Println("[app] [Monitor] " + err.Error())
|
|
} else {
|
|
log.Println("[app] [Monitor] elements: " + strconv.FormatInt(amount, 10))
|
|
}
|
|
|
|
return
|
|
}
|
|
func (mms *MqttMonitor) getDataFromIter(iter iterator.Iterator) (lastSeen LastSeenEvent, err error) {
|
|
key := iter.Key()
|
|
value := iter.Value()
|
|
err = json.Unmarshal(value, &lastSeen)
|
|
if err != nil {
|
|
log.Println("[app] [Monitor] Failed to unmarshal entry: " + string(key) + " - " + err.Error())
|
|
}
|
|
return
|
|
}
|
|
|
|
func (mms *MqttMonitor) CleanupDB() {
|
|
// Create an iterator for the database
|
|
log.Println("[app] [Monitor] Starting clean-up process")
|
|
iter := mms.db.NewIterator(nil, nil)
|
|
defer iter.Release() // Make sure to release the iterator at the end
|
|
|
|
// Iterate over all elements in the database
|
|
for iter.Next() && !mms.IsTerminated() {
|
|
// Use iter.Key() and iter.Value() to access the key and value
|
|
lastSeen, err := mms.getDataFromIter(iter)
|
|
if err != nil {
|
|
log.Println("[app] [Monitor] Failed to unmarshal entry: " + string(iter.Key()) + " - " + err.Error())
|
|
continue
|
|
}
|
|
timeThreshold := time.Now().Add(-1 * mms.CleanupPeriodicityInMinutes * time.Minute).Unix()
|
|
if lastSeen.Timestamp <= timeThreshold {
|
|
// If the entry is older than 12 hours, delete it
|
|
err := mms.deleteEntry(iter.Key())
|
|
if err != nil {
|
|
log.Println("[app] [Monitor] Failed to delete entry: " + err.Error())
|
|
} else {
|
|
log.Println("[app] [Monitor] Delete entry: " + string(iter.Key()))
|
|
}
|
|
}
|
|
}
|
|
|
|
// Check for any errors encountered during iteration
|
|
if err := iter.Error(); err != nil {
|
|
log.Println("[app] [Monitor] error during cleanup : " + err.Error())
|
|
}
|
|
}
|