EdgeAPI/internal/db/models/node_dao_test.go

//go:build plus
// +build plus

package models_test

import (
	"encoding/json"
	"testing"
	"time"

	teaconst "github.com/TeaOSLab/EdgeAPI/internal/const"
	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
	"github.com/TeaOSLab/EdgeAPI/internal/utils"
	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
	_ "github.com/go-sql-driver/mysql"
	"github.com/iwind/TeaGo/dbs"
	"github.com/iwind/TeaGo/logs"
)

func TestNodeDAO_FindAllNodeIdsMatch(t *testing.T) {
	var tx *dbs.Tx
	dbs.NotifyReady()
	nodeIds, err := models.SharedNodeDAO.FindAllNodeIdsMatch(tx, 1, true, 0)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(nodeIds)
}

func TestNodeDAO_UpdateNodeUp(t *testing.T) {
	dbs.NotifyReady()
	var tx *dbs.Tx
	err := models.SharedNodeDAO.UpdateNodeUp(tx, 57, false)
	if err != nil {
		t.Fatal(err)
	}
	t.Log("ok")
}

func TestNodeDAO_FindEnabledNodeClusterIds(t *testing.T) {
	dbs.NotifyReady()
	var tx *dbs.Tx
	clusterIds, err := models.NewNodeDAO().FindEnabledAndOnNodeClusterIds(tx, 48)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(clusterIds)
}

func TestNodeDAO_ComposeNodeConfig(t *testing.T) {
	dbs.NotifyReady()

	var before = time.Now()

	var tx *dbs.Tx
	var cacheMap = utils.NewCacheMap()
	var dataMap = shared.NewDataMap()
	//var dataMap *nodeconfigs.DataMap
	nodeConfig, err := models.SharedNodeDAO.ComposeNodeConfig(tx, 48, dataMap, cacheMap)
	if err != nil {
		t.Fatal(err)
	}
	nodeConfig.DataMap = dataMap
	t.Log(len(nodeConfig.Servers), "servers")
	t.Log(cacheMap.Len(), "items")

	t.Log(time.Since(before).Seconds()*1000, "ms")

	data, err := json.Marshal(nodeConfig)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(len(data), "bytes")

	{
		nodeConfig, err = models.SharedNodeDAO.ComposeNodeConfig(tx, 148, dataMap, cacheMap)
		if err != nil {
			t.Fatal(err)
		}
		t.Log(len(nodeConfig.DataMap.Map), "items in dataMap")
		data, err = json.Marshal(nodeConfig)
		if err != nil {
			t.Fatal(err)
		}
		t.Log(len(data), "bytes")
	}
}

func TestNodeDAO_ComposeNodeConfig_ParentNodes(t *testing.T) {
	dbs.NotifyReady()

	teaconst.IsPlus = true

	var tx *dbs.Tx
	var cacheMap = utils.NewCacheMap()
	nodeConfig, err := models.SharedNodeDAO.ComposeNodeConfig(tx, 48, nil, cacheMap)
	if err != nil {
		t.Fatal(err)
	}
	logs.PrintAsJSON(nodeConfig.ParentNodes, t)
}

func TestNodeDAO_FindEnabledNodeIdWithUniqueId(t *testing.T) {
	dbs.NotifyReady()

	var tx *dbs.Tx
	// init
	{
		_, err := models.SharedNodeDAO.FindEnabledNodeIdWithUniqueId(tx, "a186380dbd26ccd49e75d178ec59df1b")
		if err != nil {
			t.Fatal(err)
		}
	}

	var before = time.Now()
	nodeId, err := models.SharedNodeDAO.FindEnabledNodeIdWithUniqueId(tx, "a186380dbd26ccd49e75d178ec59df1b")
	if err != nil {
		t.Fatal(err)
	}
	t.Log("cost:", time.Since(before).Seconds()*1000, "ms")
	t.Log("nodeId:", nodeId)

	{
		before = time.Now()
		nodeId, err := models.SharedNodeDAO.FindEnabledNodeIdWithUniqueId(tx, "a186380dbd26ccd49e75d178ec59df1b")
		if err != nil {
			t.Fatal(err)
		}
		t.Log("cost:", time.Since(before).Seconds()*1000, "ms")
		t.Log("nodeId:", nodeId)
	}
}