添加手动注册服务工具

Author: matteriot

gateway-go.yaml

@@ -1,4 +1,4 @@
-gatewayuuid: 62ded68f-5bf6-4d97-9c2e-ced2d5fd215f
+gatewayuuid:
 logconfig:
     enablestdout: true
     logfilepath: ""

netservice/services/connect/service/mdns/mDns.go

@@ -3,6 +3,7 @@ package mdns
 import (
 	"context"
 	"encoding/json"
+	"github.com/OpenIoTHub/gateway-go/v2/register"
 	"github.com/OpenIoTHub/utils/models"
 	"github.com/OpenIoTHub/utils/msg"
 	"github.com/grandcat/zeroconf"
@@ -55,6 +56,20 @@ func (mc *MdnsCtrl) FindAllmDNS(stream net.Conn, service *models.NewService) err
 	//if len(rst) > 0 {
 	//	log.Println(rst[0])
 	//}
+	registeredServices := register.GetRegisteredServices()
+	for _, registeredService := range registeredServices {
+		rst = append(rst, &models.MDNSResult{
+			Instance: registeredService.Instance,
+			Service:  registeredService.Service,
+			Domain:   registeredService.Domain,
+			HostName: registeredService.HostName,
+			Port:     registeredService.Port,
+			Text:     registeredService.Text,
+			TTL:      registeredService.TTL,
+			AddrIPv4: registeredService.AddrIPv4,
+			AddrIPv6: registeredService.AddrIPv6,
+		})
+	}
 	rstByte, err := json.Marshal(&rst)
 	if err != nil {
 		log.Println(err.Error())

register/register_service.go

@@ -0,0 +1,33 @@
+package register
+
+import (
+	"github.com/OpenIoTHub/utils/models"
+	"net"
+	"sync"
+)
+
+var registeredServices = make([]models.MDNSResult, 0)
+var registeredServicesLock sync.RWMutex
+
+func RegisterService(instance, service, domain, hostname string, port int, text []string, TTL uint32, AddrIPv4, AddrIPv6 []net.IP) (err error) {
+	registeredServicesLock.Lock()
+	defer registeredServicesLock.Unlock()
+	registeredServices = append(registeredServices, models.MDNSResult{
+		Instance: instance,
+		Service:  service,
+		Domain:   domain,
+		HostName: hostname,
+		Port:     port,
+		Text:     text,
+		TTL:      TTL,
+		AddrIPv4: AddrIPv4,
+		AddrIPv6: AddrIPv6,
+	})
+	return
+}
+
+func GetRegisteredServices() (services []models.MDNSResult) {
+	registeredServicesLock.RLock()
+	defer registeredServicesLock.RUnlock()
+	return registeredServices
+}