:art:服务端的初始化

repository/server.go

@@ -0,0 +1,41 @@
+package repository
+
+import (
+	"gorm.io/gorm"
+	"wireguard-dashboard/client"
+	"wireguard-dashboard/model/entity"
+	"wireguard-dashboard/model/vo"
+)
+
+type server struct {
+	*gorm.DB
+}
+
+func Server() server {
+	return server{
+		client.DB,
+	}
+}
+
+// GetServer
+// @description: 获取服务端信息
+// @receiver r
+// @return data
+// @return err
+func (r server) GetServer() (data *vo.Server, err error) {
+	err = r.Model(&entity.Server{}).First(&data).Error
+	return
+}
+
+// Save
+// @description: 新增/更新服务端信息
+// @receiver r
+// @param ent
+// @return err
+func (r server) Save(ent *entity.Server) (err error) {
+	if ent.Id != "" {
+		return r.Model(&entity.Server{}).Updates(&ent).Error
+	}
+
+	return r.Model(&entity.Server{}).Create(&ent).Error
+}