Browse Source

added a cleanup

0xdcarns 3 years ago
parent
commit
19426fc3e1
3 changed files with 8 additions and 4 deletions
  1. 4 3
      logic/telemetry.go
  2. 1 1
      logic/timer.go
  3. 3 0
      netclient/functions/common.go

+ 4 - 3
logic/telemetry.go

@@ -74,11 +74,12 @@ func fetchTelemetryData() (telemetryData, error) {
 }
 
 // setTelemetryTimestamp - Give the entry in the DB a new timestamp
-func setTelemetryTimestamp(uuid string) error {
+func setTelemetryTimestamp(telRecord *models.Telemetry) error {
 	lastsend := time.Now().Unix()
 	var serverTelData = models.Telemetry{
-		UUID:     uuid,
-		LastSend: lastsend,
+		UUID:       telRecord.UUID,
+		LastSend:   lastsend,
+		TrafficKey: telRecord.TrafficKey,
 	}
 	jsonObj, err := json.Marshal(&serverTelData)
 	if err != nil {

+ 1 - 1
logic/timer.go

@@ -31,7 +31,7 @@ func TimerCheckpoint() error {
 		runHooks()
 	}
 	// set telemetry timestamp for server, restarts 24 hour cycle
-	return setTelemetryTimestamp(telRecord.UUID)
+	return setTelemetryTimestamp(&telRecord)
 }
 
 // AddHook - adds a hook function to run every 24hrs

+ 3 - 0
netclient/functions/common.go

@@ -253,6 +253,9 @@ func WipeLocal(network string) error {
 	if ncutils.FileExists(home + "secret-" + network) {
 		_ = os.Remove(home + "secret-" + network)
 	}
+	if ncutils.FileExists(home + "traffic-" + network) {
+		_ = os.Remove(home + "traffic-" + network)
+	}
 	if ncutils.FileExists(home + "wgkey-" + network) {
 		_ = os.Remove(home + "wgkey-" + network)
 	}