zishuo/netmaker
1
0
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-10-18 14:54:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

unexport cache funcs

Browse Source
This commit is contained in:
Abhishek Kondur
2023-06-26 11:18:43 +05:30
parent 4071bb2bac
commit 307203105d
2 changed files with 65 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
logic/nodes.go
View File

@@ -23,44 +23,44 @@ import (
)
var (
NodeCacheMutex = &sync.RWMutex{}
NodesCacheMap = make(map[string]models.Node)
nodeCacheMutex = &sync.RWMutex{}
nodesCacheMap = make(map[string]models.Node)
)
func GetNodeFromCache(nodeID string) (node models.Node, ok bool) {
NodeCacheMutex.RLock()
node, ok = NodesCacheMap[nodeID]
NodeCacheMutex.RUnlock()
func getNodeFromCache(nodeID string) (node models.Node, ok bool) {
nodeCacheMutex.RLock()
node, ok = nodesCacheMap[nodeID]
nodeCacheMutex.RUnlock()
return
}
func GetNodesFromCache() (nMap map[string]models.Node) {
NodeCacheMutex.RLock()
nMap = NodesCacheMap
NodeCacheMutex.RUnlock()
func getNodesFromCache() (nMap map[string]models.Node) {
nodeCacheMutex.RLock()
nMap = nodesCacheMap
nodeCacheMutex.RUnlock()
return
}
func DeleteNodeFromCache(nodeID string) {
NodeCacheMutex.Lock()
delete(NodesCacheMap, nodeID)
NodeCacheMutex.Unlock()
func deleteNodeFromCache(nodeID string) {
nodeCacheMutex.Lock()
delete(nodesCacheMap, nodeID)
nodeCacheMutex.Unlock()
}
func StoreNodeInCache(node models.Node) {
NodeCacheMutex.Lock()
NodesCacheMap[node.ID.String()] = node
NodeCacheMutex.Unlock()
func storeNodeInCache(node models.Node) {
nodeCacheMutex.Lock()
nodesCacheMap[node.ID.String()] = node
nodeCacheMutex.Unlock()
}
func LoadNodesIntoCache(nMap map[string]models.Node) {
NodeCacheMutex.Lock()
NodesCacheMap = nMap
NodeCacheMutex.Unlock()
func loadNodesIntoCache(nMap map[string]models.Node) {
nodeCacheMutex.Lock()
nodesCacheMap = nMap
nodeCacheMutex.Unlock()
}
func ClearNodeCache() {
NodeCacheMutex.Lock()
NodesCacheMap = make(map[string]models.Node)
NodeCacheMutex.Unlock()
nodeCacheMutex.Lock()
nodesCacheMap = make(map[string]models.Node)
nodeCacheMutex.Unlock()
}
const (
@@ -117,7 +117,7 @@ func UpdateNodeCheckin(node *models.Node) error {
if err != nil {
return err
}
StoreNodeInCache(*node)
storeNodeInCache(*node)
return nil
}
@@ -132,7 +132,7 @@ func UpsertNode(newNode *models.Node) error {
if err != nil {
return err
}
StoreNodeInCache(*newNode)
storeNodeInCache(*newNode)
return nil
}
@@ -169,7 +169,7 @@ func UpdateNode(currentNode *models.Node, newNode *models.Node) error {
if err != nil {
return err
}
StoreNodeInCache(*newNode)
storeNodeInCache(*newNode)
return nil
}
}
@@ -228,7 +228,7 @@ func deleteNodeByID(node *models.Node) error {
return err
}
}
DeleteNodeFromCache(node.ID.String())
deleteNodeFromCache(node.ID.String())
if servercfg.IsDNSMode() {
SetDNS()
}
@@ -294,7 +294,7 @@ func IsFailoverPresent(network string) bool {
// GetAllNodes - returns all nodes in the DB
func GetAllNodes() ([]models.Node, error) {
var nodes []models.Node
nodesMap := GetNodesFromCache()
nodesMap := getNodesFromCache()
if len(nodesMap) != 0 {
for _, node := range nodesMap {
nodes = append(nodes, node)
@@ -302,7 +302,7 @@ func GetAllNodes() ([]models.Node, error) {
return nodes, nil
}
nodesMap = make(map[string]models.Node)
defer LoadNodesIntoCache(nodesMap)
defer loadNodesIntoCache(nodesMap)
collection, err := database.FetchRecords(database.NODES_TABLE_NAME)
if err != nil {
if database.IsEmptyRecord(err) {
@@ -376,7 +376,7 @@ func GetRecordKey(id string, network string) (string, error) {
}
func GetNodeByID(uuid string) (models.Node, error) {
if node, ok := GetNodeFromCache(uuid); ok {
if node, ok := getNodeFromCache(uuid); ok {
return node, nil
}
var record, err = database.FetchRecord(database.NODES_TABLE_NAME, uuid)
@@ -387,7 +387,7 @@ func GetNodeByID(uuid string) (models.Node, error) {
if err = json.Unmarshal([]byte(record), &node); err != nil {
return models.Node{}, err
}
StoreNodeInCache(node)
storeNodeInCache(node)
return node, nil
}
@@ -537,7 +537,7 @@ func createNode(node *models.Node) error {
if err != nil {
return err
}
StoreNodeInCache(*node)
storeNodeInCache(*node)
_, err = nodeacls.CreateNodeACL(nodeacls.NetworkID(node.Network), nodeacls.NodeID(node.ID.String()), defaultACLVal)
if err != nil {
logger.Log(1, "failed to create node ACL for node,", node.ID.String(), "err:", err.Error())