基于Golang實(shí)現(xiàn)統(tǒng)一加載資源的入口

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

當(dāng)我們需要在?main?函數(shù)中做一些初始化的工作，比如初始化日志，初始化配置文件，都需要統(tǒng)一初始化入口函數(shù)，所以本文就來(lái)編寫一個(gè)統(tǒng)一加載資源的入口吧

統(tǒng)一初始化入口

什么時(shí)候需要統(tǒng)一初始化入口函數(shù)呢？

當(dāng)我們需要在 main 函數(shù)中做一些初始化的工作，比如初始化日志，初始化配置文件，初始化數(shù)據(jù)庫(kù)等等

這些都是需要在 main 函數(shù)中做的初始化工作，這個(gè)時(shí)候就可以使用統(tǒng)一初始化入口的方式來(lái)實(shí)現(xiàn)

抽象出生命周期接口

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

Init: 系統(tǒng)啟動(dòng)，初始化一些基礎(chǔ)資源，比如數(shù)據(jù)庫(kù)，消息隊(duì)列等
Setup: 安裝一些基礎(chǔ)資源，比如初始化數(shù)據(jù)庫(kù)表，或者其他基礎(chǔ)資源的初始化
Start: 啟動(dòng)基礎(chǔ)資源，比如啟動(dòng)數(shù)據(jù)庫(kù)連接池，消息隊(duì)列消費(fèi)者
StartBlocking: 啟動(dòng)器是否可阻塞，比如啟動(dòng) RPC 服務(wù)，啟動(dòng)器需要一直阻塞才能提供 RPC 服務(wù)
Stop: 系統(tǒng)停止，一般是一些基礎(chǔ)資源的回收和停止，比如數(shù)據(jù)庫(kù)連接池，消息隊(duì)列連接池等

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

實(shí)現(xiàn)生命周期接口

下一步是實(shí)現(xiàn)這個(gè)接口，在 go 中 interface 是隱式實(shí)現(xiàn)的，只要實(shí)現(xiàn)了接口中的方法，就是實(shí)現(xiàn)了這個(gè)接口

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)  {}

實(shí)現(xiàn)自動(dòng)裝配

所有的啟動(dòng)器都實(shí)現(xiàn)了 Starter 接口

這時(shí)就可以將所有的啟動(dòng)器都放在一個(gè)切片中，然后遍歷切片，依次調(diào)用 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
}

提供一個(gè)注冊(cè)和運(yùn)行的方法，這樣就可以實(shí)現(xiàn)自動(dòng)裝配了

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.為每個(gè)資源創(chuàng)建一個(gè) config.go 文件，然后在 init 函數(shù)中注冊(cè)

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 函數(shù)中調(diào)用 SystemRun 方法

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

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

您可能感興趣的文章: