基于Golang實現統一加載資源的入口

更新時間：2023年05月29日 09:06:30 作者：uccs

當我們需要在?main?函數中做一些初始化的工作，比如初始化日志，初始化配置文件，都需要統一初始化入口函數，所以本文就來編寫一個統一加載資源的入口吧

統一初始化入口

什么時候需要統一初始化入口函數呢？

當我們需要在 main 函數中做一些初始化的工作，比如初始化日志，初始化配置文件，初始化數據庫等等

這些都是需要在 main 函數中做的初始化工作，這個時候就可以使用統一初始化入口的方式來實現

抽象出生命周期接口

使用統一初始化入口的方式，需要我們抽象出生命周期接口
比如下面我們定義了 Init、Setup、Start、StartBlocking、Stop 五個接口，用來完成五件事情：

Init: 系統啟動，初始化一些基礎資源，比如數據庫，消息隊列等
Setup: 安裝一些基礎資源，比如初始化數據庫表，或者其他基礎資源的初始化
Start: 啟動基礎資源，比如啟動數據庫連接池，消息隊列消費者
StartBlocking: 啟動器是否可阻塞，比如啟動 RPC 服務，啟動器需要一直阻塞才能提供 RPC 服務
Stop: 系統停止，一般是一些基礎資源的回收和停止，比如數據庫連接池，消息隊列連接池等

type Starter interface {
    Init(ctx StarterContext)
    Setup(ctx StarterContext)
    Start(ctx StarterContext)
    StartBlocking() bool
    Stop(ctx StarterContext)
}

實現生命周期接口

下一步是實現這個接口，在 go 中 interface 是隱式實現的，只要實現了接口中的方法，就是實現了這個接口

type BaseStarter struct{}

func (b *BaseStarter) Init(ctx StarterContext)  {}
func (b *BaseStarter) Setup(ctx StarterContext) {}
func (b *BaseStarter) Start(ctx StarterContext) {}
func (b *BaseStarter) StartBlocking() bool      { return false }
func (b *BaseStarter) Stop(ctx StarterContext)  {}

實現自動裝配

所有的啟動器都實現了 Starter 接口

這時就可以將所有的啟動器都放在一個切片中，然后遍歷切片，依次調用 Init、Setup、Start、StartBlocking、Stop 方法

type StarterRegister struct {
    starters []Starter
}
func (sr *StarterRegister) Register(s Starter) {
    sr.starters = append(sr.starters, s)
}
func (sr *StarterRegister) AllStarters() []Starter {
    return sr.starters
}

提供一個注冊和運行的方法，這樣就可以實現自動裝配了

func Register(s Starter) {
    starterRegister.Register(s)
}
func SystemRun() {
    ctx := StarterContext{}
    for _, s := range starterRegister.AllStarters() {
        s.Init(ctx)
        s.Setup(ctx)
        s.Start(ctx)
        s.StartBlocking()
        s.Stop(ctx)
    }
}

使用

1.為每個資源創(chuàng)建一個 config.go 文件，然后在 init 函數中注冊

func init() {
	fmt.Println("config")
	Register(&Config{})
}

type Config struct{}

func (c *Config) Init(ctx StarterContext) {
	fmt.Println("Init")
}

func (c *Config) Setup(ctx StarterContext) {
	fmt.Println("Setup")
}

func (c *Config) Start(ctx StarterContext) {
	fmt.Println("Start")
}

func (c *Config) StartBlocking() bool {
  fmt.Println("StartBlocking")
	return true
}

func (c *Config) Stop(ctx StarterContext) {
	fmt.Println("Stop")
}

2.在 main 函數中調用 SystemRun 方法

通過這種方式，就可以實現自動裝配了，有多少資源，就創(chuàng)建多少 config.go 文件，并實現 Starter 接口

到此這篇關于基于Golang實現統一加載資源的入口的文章就介紹到這了,更多相關Golang統一入口內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: