Fixed nil err in isempty record 2

This commit is contained in:
worker-9
2021-08-05 13:11:21 -04:00
parent 34b61f8e79
commit 736bcc3e55
2 changed files with 23 additions and 2 deletions

View File

@@ -354,8 +354,8 @@ func deleteNetwork(w http.ResponseWriter, r *http.Request) {
} }
func DeleteNetwork(network string) error { func DeleteNetwork(network string) error {
_, err := database.FetchRecords(database.NODES_TABLE_NAME) nodeCount, err := functions.GetNetworkNodeCount(network)
if database.IsEmptyRecord(err) { if nodeCount == 0 || database.IsEmptyRecord(err) {
return database.DeleteRecord(database.NETWORKS_TABLE_NAME, network) return database.DeleteRecord(database.NETWORKS_TABLE_NAME, network)
} }
return errors.New("node check failed. All nodes must be deleted before deleting network") return errors.New("node check failed. All nodes must be deleted before deleting network")

View File

@@ -323,6 +323,27 @@ func IsMacAddressUnique(macaddress string, networkName string) (bool, error) {
return true, nil return true, nil
} }
func GetNetworkNodeCount(networkName string) (int, error) {
collection, err := database.FetchRecords(database.NODES_TABLE_NAME)
count := 0
if err != nil && !database.IsEmptyRecord(err) {
return count, err
}
for _, value := range collection {
var node models.Node
if err = json.Unmarshal([]byte(value), &node); err != nil {
return count, err
} else {
if node.Network == networkName {
count++
}
}
}
return count, nil
}
//Checks to see if access key is valid //Checks to see if access key is valid
//Does so by checking against all keys and seeing if any have the same value //Does so by checking against all keys and seeing if any have the same value
//may want to hash values before comparing...consider this //may want to hash values before comparing...consider this