relay.go 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133
  1. package logic
  2. import (
  3. "encoding/json"
  4. "errors"
  5. "time"
  6. "github.com/gravitl/netmaker/database"
  7. "github.com/gravitl/netmaker/logger"
  8. "github.com/gravitl/netmaker/models"
  9. )
  10. // CreateRelay - creates a relay
  11. func CreateRelay(relay models.RelayRequest) (models.Node, error) {
  12. node, err := GetNodeByID(relay.NodeID)
  13. if err != nil {
  14. return models.Node{}, err
  15. }
  16. if node.OS == "macos" { // add in darwin later
  17. return models.Node{}, errors.New(node.OS + " is unsupported for relay")
  18. }
  19. err = ValidateRelay(relay)
  20. if err != nil {
  21. return models.Node{}, err
  22. }
  23. node.IsRelay = "yes"
  24. node.RelayAddrs = relay.RelayAddrs
  25. node.SetLastModified()
  26. node.PullChanges = "yes"
  27. nodeData, err := json.Marshal(&node)
  28. if err != nil {
  29. return node, err
  30. }
  31. if err = database.Insert(node.ID, string(nodeData), database.NODES_TABLE_NAME); err != nil {
  32. return models.Node{}, err
  33. }
  34. err = SetRelayedNodes("yes", node.Network, node.RelayAddrs)
  35. if err != nil {
  36. return node, err
  37. }
  38. if err = NetworkNodesUpdatePullChanges(node.Network); err != nil {
  39. return models.Node{}, err
  40. }
  41. return node, nil
  42. }
  43. // SetRelayedNodes- set relayed nodes
  44. func SetRelayedNodes(yesOrno string, networkName string, addrs []string) error {
  45. collections, err := database.FetchRecords(database.NODES_TABLE_NAME)
  46. if err != nil {
  47. return err
  48. }
  49. for _, value := range collections {
  50. var node models.Node
  51. err := json.Unmarshal([]byte(value), &node)
  52. if err != nil {
  53. return err
  54. }
  55. if node.Network == networkName {
  56. for _, addr := range addrs {
  57. if addr == node.Address || addr == node.Address6 {
  58. node.IsRelayed = yesOrno
  59. data, err := json.Marshal(&node)
  60. if err != nil {
  61. return err
  62. }
  63. database.Insert(node.ID, string(data), database.NODES_TABLE_NAME)
  64. }
  65. }
  66. }
  67. }
  68. return nil
  69. }
  70. // ValidateRelay - checks if relay is valid
  71. func ValidateRelay(relay models.RelayRequest) error {
  72. var err error
  73. //isIp := functions.IsIpCIDR(gateway.RangeString)
  74. empty := len(relay.RelayAddrs) == 0
  75. if empty {
  76. err = errors.New("IP Ranges Cannot Be Empty")
  77. }
  78. return err
  79. }
  80. // UpdateRelay - updates a relay
  81. func UpdateRelay(network string, oldAddrs []string, newAddrs []string) {
  82. time.Sleep(time.Second / 4)
  83. err := SetRelayedNodes("no", network, oldAddrs)
  84. if err != nil {
  85. logger.Log(1, err.Error())
  86. }
  87. err = SetRelayedNodes("yes", network, newAddrs)
  88. if err != nil {
  89. logger.Log(1, err.Error())
  90. }
  91. }
  92. // DeleteRelay - deletes a relay
  93. func DeleteRelay(network, nodeid string) (models.Node, error) {
  94. node, err := GetNodeByID(nodeid)
  95. if err != nil {
  96. return models.Node{}, err
  97. }
  98. err = SetRelayedNodes("no", node.Network, node.RelayAddrs)
  99. if err != nil {
  100. return node, err
  101. }
  102. node.IsRelay = "no"
  103. node.RelayAddrs = []string{}
  104. node.SetLastModified()
  105. node.PullChanges = "yes"
  106. data, err := json.Marshal(&node)
  107. if err != nil {
  108. return models.Node{}, err
  109. }
  110. if err = database.Insert(nodeid, string(data), database.NODES_TABLE_NAME); err != nil {
  111. return models.Node{}, err
  112. }
  113. if err = NetworkNodesUpdatePullChanges(network); err != nil {
  114. return models.Node{}, err
  115. }
  116. return node, nil
  117. }