Browse Source

check if trial table is empty

abhishek9686 1 year ago
parent
commit
dbc48b1b2f
1 changed files with 6 additions and 4 deletions
  1. 6 4
      pro/trial.go

+ 6 - 4
pro/trial.go

@@ -15,7 +15,6 @@ import (
 	"github.com/gravitl/netmaker/models"
 	"github.com/gravitl/netmaker/netclient/ncutils"
 	"golang.org/x/crypto/nacl/box"
-	"golang.org/x/exp/slog"
 )
 
 type TrialInfo struct {
@@ -49,11 +48,14 @@ func initTrial() error {
 	if telData.Hosts > 0 || telData.Networks > 0 || telData.Users > 0 {
 		return nil
 	}
-	err = database.CreateTable(trial_table_name)
-	if err != nil {
-		slog.Error("failed to create table", "table name", trial_table_name, "err", err.Error())
+	database.CreateTable(trial_table_name)
+	records, err := database.FetchRecords(trial_table_name)
+	if err != nil && !database.IsEmptyRecord(err) {
 		return err
 	}
+	if len(records) > 0 {
+		return nil
+	}
 	// setup encryption keys
 	trafficPubKey, trafficPrivKey, err := box.GenerateKey(rand.Reader) // generate traffic keys
 	if err != nil {