|
@@ -11,15 +11,17 @@ import (
|
|
)
|
|
)
|
|
|
|
|
|
// CreateRelay - creates a relay
|
|
// CreateRelay - creates a relay
|
|
-func CreateRelay(relay models.RelayRequest) (models.Node, error) {
|
|
|
|
|
|
+func CreateRelay(relay models.RelayRequest) ([]models.Node, models.Node, error) {
|
|
|
|
+ var returnnodes []models.Node
|
|
|
|
+
|
|
node, err := GetNodeByID(relay.NodeID)
|
|
node, err := GetNodeByID(relay.NodeID)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return models.Node{}, err
|
|
|
|
|
|
+ return returnnodes, models.Node{}, err
|
|
}
|
|
}
|
|
|
|
|
|
err = ValidateRelay(relay)
|
|
err = ValidateRelay(relay)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return models.Node{}, err
|
|
|
|
|
|
+ return returnnodes, models.Node{}, err
|
|
}
|
|
}
|
|
node.IsRelay = "yes"
|
|
node.IsRelay = "yes"
|
|
node.RelayAddrs = relay.RelayAddrs
|
|
node.RelayAddrs = relay.RelayAddrs
|
|
@@ -28,32 +30,31 @@ func CreateRelay(relay models.RelayRequest) (models.Node, error) {
|
|
node.PullChanges = "yes"
|
|
node.PullChanges = "yes"
|
|
nodeData, err := json.Marshal(&node)
|
|
nodeData, err := json.Marshal(&node)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return node, err
|
|
|
|
|
|
+ return returnnodes, node, err
|
|
}
|
|
}
|
|
if err = database.Insert(node.ID, string(nodeData), database.NODES_TABLE_NAME); err != nil {
|
|
if err = database.Insert(node.ID, string(nodeData), database.NODES_TABLE_NAME); err != nil {
|
|
- return models.Node{}, err
|
|
|
|
|
|
+ return returnnodes, models.Node{}, err
|
|
}
|
|
}
|
|
- err = SetRelayedNodes("yes", node.Network, node.RelayAddrs)
|
|
|
|
|
|
+ returnnodes, err = SetRelayedNodes("yes", node.Network, node.RelayAddrs)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return node, err
|
|
|
|
|
|
+ return returnnodes, node, err
|
|
}
|
|
}
|
|
-
|
|
|
|
if err = NetworkNodesUpdatePullChanges(node.Network); err != nil {
|
|
if err = NetworkNodesUpdatePullChanges(node.Network); err != nil {
|
|
- return models.Node{}, err
|
|
|
|
|
|
+ return returnnodes, models.Node{}, err
|
|
}
|
|
}
|
|
- return node, nil
|
|
|
|
|
|
+ return returnnodes, node, nil
|
|
}
|
|
}
|
|
|
|
|
|
// SetRelayedNodes- set relayed nodes
|
|
// SetRelayedNodes- set relayed nodes
|
|
-func SetRelayedNodes(yesOrno string, networkName string, addrs []string) error {
|
|
|
|
-
|
|
|
|
|
|
+func SetRelayedNodes(yesOrno string, networkName string, addrs []string) ([]models.Node, error) {
|
|
|
|
+ var returnnodes []models.Node
|
|
collections, err := database.FetchRecords(database.NODES_TABLE_NAME)
|
|
collections, err := database.FetchRecords(database.NODES_TABLE_NAME)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return err
|
|
|
|
|
|
+ return returnnodes, err
|
|
}
|
|
}
|
|
network, err := GetNetworkSettings(networkName)
|
|
network, err := GetNetworkSettings(networkName)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return err
|
|
|
|
|
|
+ return returnnodes, err
|
|
}
|
|
}
|
|
|
|
|
|
for _, value := range collections {
|
|
for _, value := range collections {
|
|
@@ -61,7 +62,7 @@ func SetRelayedNodes(yesOrno string, networkName string, addrs []string) error {
|
|
var node models.Node
|
|
var node models.Node
|
|
err := json.Unmarshal([]byte(value), &node)
|
|
err := json.Unmarshal([]byte(value), &node)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return err
|
|
|
|
|
|
+ return returnnodes, err
|
|
}
|
|
}
|
|
if node.Network == networkName && !(node.IsServer == "yes") {
|
|
if node.Network == networkName && !(node.IsServer == "yes") {
|
|
for _, addr := range addrs {
|
|
for _, addr := range addrs {
|
|
@@ -74,14 +75,15 @@ func SetRelayedNodes(yesOrno string, networkName string, addrs []string) error {
|
|
}
|
|
}
|
|
data, err := json.Marshal(&node)
|
|
data, err := json.Marshal(&node)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return err
|
|
|
|
|
|
+ return returnnodes, err
|
|
}
|
|
}
|
|
database.Insert(node.ID, string(data), database.NODES_TABLE_NAME)
|
|
database.Insert(node.ID, string(data), database.NODES_TABLE_NAME)
|
|
|
|
+ returnnodes = append(returnnodes, node)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
- return nil
|
|
|
|
|
|
+ return returnnodes, nil
|
|
}
|
|
}
|
|
|
|
|
|
// SetNodeIsRelayed - Sets IsRelayed to on or off for relay
|
|
// SetNodeIsRelayed - Sets IsRelayed to on or off for relay
|
|
@@ -128,28 +130,30 @@ func ValidateRelay(relay models.RelayRequest) error {
|
|
}
|
|
}
|
|
|
|
|
|
// UpdateRelay - updates a relay
|
|
// UpdateRelay - updates a relay
|
|
-func UpdateRelay(network string, oldAddrs []string, newAddrs []string) {
|
|
|
|
|
|
+func UpdateRelay(network string, oldAddrs []string, newAddrs []string) []models.Node {
|
|
|
|
+ var returnnodes []models.Node
|
|
time.Sleep(time.Second / 4)
|
|
time.Sleep(time.Second / 4)
|
|
- err := SetRelayedNodes("no", network, oldAddrs)
|
|
|
|
|
|
+ returnnodes, err := SetRelayedNodes("no", network, oldAddrs)
|
|
if err != nil {
|
|
if err != nil {
|
|
logger.Log(1, err.Error())
|
|
logger.Log(1, err.Error())
|
|
}
|
|
}
|
|
- err = SetRelayedNodes("yes", network, newAddrs)
|
|
|
|
|
|
+ returnnodes, err = SetRelayedNodes("yes", network, newAddrs)
|
|
if err != nil {
|
|
if err != nil {
|
|
logger.Log(1, err.Error())
|
|
logger.Log(1, err.Error())
|
|
}
|
|
}
|
|
|
|
+ return returnnodes
|
|
}
|
|
}
|
|
|
|
|
|
// DeleteRelay - deletes a relay
|
|
// DeleteRelay - deletes a relay
|
|
-func DeleteRelay(network, nodeid string) (models.Node, error) {
|
|
|
|
-
|
|
|
|
|
|
+func DeleteRelay(network, nodeid string) ([]models.Node, models.Node, error) {
|
|
|
|
+ var returnnodes []models.Node
|
|
node, err := GetNodeByID(nodeid)
|
|
node, err := GetNodeByID(nodeid)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return models.Node{}, err
|
|
|
|
|
|
+ return returnnodes, models.Node{}, err
|
|
}
|
|
}
|
|
- err = SetRelayedNodes("no", node.Network, node.RelayAddrs)
|
|
|
|
|
|
+ _, err = SetRelayedNodes("no", node.Network, node.RelayAddrs)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return node, err
|
|
|
|
|
|
+ return returnnodes, node, err
|
|
}
|
|
}
|
|
|
|
|
|
node.IsRelay = "no"
|
|
node.IsRelay = "no"
|
|
@@ -159,13 +163,13 @@ func DeleteRelay(network, nodeid string) (models.Node, error) {
|
|
|
|
|
|
data, err := json.Marshal(&node)
|
|
data, err := json.Marshal(&node)
|
|
if err != nil {
|
|
if err != nil {
|
|
- return models.Node{}, err
|
|
|
|
|
|
+ return returnnodes, models.Node{}, err
|
|
}
|
|
}
|
|
if err = database.Insert(nodeid, string(data), database.NODES_TABLE_NAME); err != nil {
|
|
if err = database.Insert(nodeid, string(data), database.NODES_TABLE_NAME); err != nil {
|
|
- return models.Node{}, err
|
|
|
|
|
|
+ return returnnodes, models.Node{}, err
|
|
}
|
|
}
|
|
if err = NetworkNodesUpdatePullChanges(network); err != nil {
|
|
if err = NetworkNodesUpdatePullChanges(network); err != nil {
|
|
- return models.Node{}, err
|
|
|
|
|
|
+ return returnnodes, models.Node{}, err
|
|
}
|
|
}
|
|
- return node, nil
|
|
|
|
|
|
+ return returnnodes, node, nil
|
|
}
|
|
}
|