完善个人数据维护, 初步建立局域网链接逻辑

service/server.go

@@ -0,0 +1,72 @@
+package service
+
+import (
+	"errors"
+	"github.com/gin-gonic/gin"
+	"log"
+	"net/http"
+	"work_cation/cfg"
+	"work_cation/global"
+	"work_cation/repo"
+)
+
+type serverService struct {
+	Status string
+	server *http.Server
+}
+
+const (
+	StatusOnline  = "online"
+	StatusOffline = "offline"
+)
+
+var Server = &serverService{Status: StatusOffline}
+
+func (s *serverService) Online() error {
+	if s.Status == StatusOnline {
+		return nil
+	}
+	// 启动服务发现
+	user := repo.User.GetUserInfo(global.DB)
+	if err := Zeroconf.Register(user); err != nil {
+		return err
+	}
+	// 启动服务
+	if err := s.StartListenServer(); err != nil {
+		return err
+	}
+	// 修改状态
+	s.Status = StatusOnline
+	return nil
+}
+
+func (s *serverService) StatusOffline() error {
+	if err := s.server.Close(); err != nil {
+		return err
+	}
+	// 关闭服务发现
+	Zeroconf.Close()
+	// 修改状态
+	s.Status = StatusOffline
+	return nil
+}
+
+func (s *serverService) StartListenServer() error {
+	router := gin.New()
+	router.GET("/", func(c *gin.Context) {
+		c.JSON(200, gin.H{"message": "hello"})
+	})
+
+	srv := &http.Server{
+		Addr:    cfg.T.ServerAddr,
+		Handler: router,
+	}
+
+	go func() {
+		if err := srv.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
+			log.Fatalf("listen: %s\n", err)
+		}
+	}()
+	s.server = srv
+	return nil
+}