Message-Push-Nest/service/settings_service/init_site.go

package settings_service

import (
	"github.com/sirupsen/logrus"
	"message-nest/models"
	"message-nest/pkg/constant"
)

// 初始化环境的设置数据
type InitSettingService struct {
}

func (es *InitSettingService) CommonAddSetting(section string, key string, value string) {
	setting, _ := models.GetSettingByKey(section, key)
	if setting.ID <= 0 {
		err := models.AddOneSetting(models.Settings{
			Section: section,
			Key:     key,
			Value:   value,
		})
		if err != nil {
			logrus.Error("初始化%s:%s失败", section, key)
		} else {
			logrus.Infof("初始化%s:%s成功", section, key)
		}
	}
}

// InitSiteConfig 初始化、重置站点信息设置
func (es *InitSettingService) InitSiteConfig() {
	section := constant.SiteSettingSectionName
	for key, value := range constant.SiteSiteDefaultValueMap {
		es.CommonAddSetting(section, key, value)
	}
}

// InitLogConfig 初始化日志清理设置
func (es *InitSettingService) InitLogConfig() {
	section := constant.LogsCleanSectionName
	for key, value := range constant.LogsCleanDefaultValueMap {
		es.CommonAddSetting(section, key, value)
	}
}

// InitHostedMsgConfig 初始化托管消息清理设置
func (es *InitSettingService) InitHostedMsgConfig() {
	section := constant.HostedMsgCleanSectionName
	for key, value := range constant.HostedMsgCleanDefaultValueMap {
		es.CommonAddSetting(section, key, value)
	}
}