快捷導(dǎo)航

Go語言實(shí)現(xiàn)配置熱加載的方法分享

更新時(shí)間：2023年05月24日 11:41:22 作者：tracy小貓

web項(xiàng)目，經(jīng)常需要熱啟動(dòng)各種各樣的配置信息，一旦這些服務(wù)發(fā)生變更，我們需要重新啟動(dòng)web server，以使配置生效，實(shí)現(xiàn)配置熱加載，本文為大家整理了幾個(gè)方法實(shí)現(xiàn)這個(gè)需求，需要的可以參考下

概述

web項(xiàng)目，經(jīng)常需要熱啟動(dòng)各種各樣的配置信息，一旦這些服務(wù)發(fā)生變更，我們需要重新啟動(dòng)web server，以使配置生效，實(shí)現(xiàn)配置熱加載。這里有幾種方法實(shí)現(xiàn)這個(gè)需求。

go 定時(shí)器協(xié)程實(shí)現(xiàn)

項(xiàng)目結(jié)構(gòu)

首先來看一下整個(gè)項(xiàng)目的目錄結(jié)構(gòu):

- dyconfig // 項(xiàng)目地址
- config // 配置文件目錄
- api.yaml // 采用yaml格式文件
- global // 代碼文件夾
- config
- config_define
- init
- reload
- go.mod // go package管理依賴的包文件
- go.sum // go package管理打包產(chǎn)生的文件
- main.go // web server的入口，主函數(shù)

代碼細(xì)節(jié)

接下來依次看一下各文件的主體內(nèi)容：

conf/api.yaml文件定義了配置項(xiàng)，包含server的host及port信息。

service:
    server:
          env: dev
          host: 127.0.0.1
          port: 9902

global/init.go

package global
import (
 "context"
 "path"
)
type Config struct {
 Conf struct {
    FilePath       string
    LastModifyTime int64
 }
 ctx    context.Context
 cancel context.CancelFunc
}
func NewConfig() (*Config, error) {
 conf := new(Config)
 conf.ctx, conf.cancel = context.WithCancel(context.Background())
 conf.Conf.FilePath = path.Join("./config", "api.yaml")
 APIconfig = conf.loadRoute()
 go conf.reload() //開啟協(xié)程監(jiān)聽routeNotify
 go func() {
    for {
       select {
       case lastModifyTime, ok := <-routeNotify:
          if !ok {
             return
          }
          conf.Conf.LastModifyTime = lastModifyTime
          route := routeAtomic.Load().(*APIConf)
          if route != nil {
             APIconfig = route
          }
       }
    }
 }()
 return conf, nil
}
func (c *Config) Stop() {
 c.cancel()
}

定義Config 根據(jù)LastModifyTime 判斷是否有發(fā)生變化，F(xiàn)ilePath為文件路徑

go conf.reload()

開啟協(xié)程監(jiān)聽routeNotify，routeNotify內(nèi)容是文件修改時(shí)間的timestamp

global/reload.go

package global
import (
   "fmt"
   "gopkg.in/yaml.v3"
   "io/ioutil"
   "os"
   "sync/atomic"
   "time"
)
const (
   CheckInterval = 5 * time.Second
)
var (
   routeAtomic atomic.Value //原子性，解決并發(fā)問題
   routeNotify = make(chan int64) //channel 放入timestamp
)
func (c *Config) reload() {
   ticker := time.NewTicker(CheckInterval)
   defer ticker.Stop()
   for {
      select {
      case <-c.ctx.Done():
         close(routeNotify)
         return
      case <-ticker.C:
         if f, err := os.Stat(c.Route.FilePath); err != nil {
            fmt.Println(err)
         } else if f.ModTime().Unix() != c.Route.LastModifyTime {
            if c.Route.LastModifyTime == 0 {
               c.Route.LastModifyTime = f.ModTime().Unix()
            } else {
               routeAtomic.Store(c.loadConfig())
               routeNotify <- f.ModTime().Unix()
               fmt.Println("配置文件發(fā)生變化")
            }
         }
      }
   }
}
func (c *Config) loadConfig() *APIConf {
   if fp, err := ioutil.ReadFile(c.Route.FilePath); err != nil {
      fmt.Println(err)
      return nil
   } else {
      route := new(APIConf)
      if err := yaml.Unmarshal(fp, &route); err != nil {
         return nil
      }
      return route
   }
}

定時(shí)器監(jiān)聽文件的修改時(shí)間與LastModifyTime是否相同，如果不同，則

package global
var (
   APIconfig = new(APIConf)
)

package global
type ServiceConf struct {
   Server struct {
      Env  string `yaml:"env"` 
      Host string `yaml:"host"`
      Port string  `yaml:"port"`
   } `yaml:"server"`
}
type APIConf struct { 
   Service ServiceConf `yaml:"service"`
}

mian

package main
import (
   "dyconfig/global"
   "fmt"
   "github.com/gin-gonic/gin"
)
func main() {
   if conf, err := global.NewConfig(); err != nil { // 初始化配置
      fmt.Println(err)
   } else {
      defer conf.Stop()
   }
   gin.SetMode(gin.DebugMode)
   r := gin.Default()
   r.GET("/ping", func(context *gin.Context) {
      fmt.Println("當(dāng)前host是: ", global.APIconfig.Service.Server.Host)
      fmt.Println("當(dāng)前port是: ", global.APIconfig.Service.Server.Port)
      context.JSON(
         200, gin.H{
            "host": global.APIconfig.Service.Server.Host,
            "port": &global.APIconfig.Service.Server.Port,
         })
   })
   port := global.APIconfig.Service.Server.Port
   fmt.Println("當(dāng)前host是: ", global.APIconfig.Service.Server.Host)
   fmt.Println("當(dāng)前port是: ", global.APIconfig.Service.Server.Port)
   port = ":" + port
   _ = r.Run(port)
}