心得分享，如何组织入口文件

[Gota]

这里讨论的是不用框架只用工具库（ Toolkit ）的情况，否则组织形式只能按照框架的规定，也就没什么讨论空间了。

其实同样的思路还可以用来组织具体的业务代码，整合第三方服务等不少场景。 这里先讲个大概，等过年回家之后再整理篇完整的博客。

入口部分主要是声明退出方式，然后开始初始化流程。

// cmd/app/main.go
package main

func main() {
	// 在第一次收到 SIG_INT ，SIG_TERM 时结束 Context ，
	// 第二次收到时 os.Exit()
	ctx, cancel := module.GracefulContext()
	defer cancel()

	// 执行业务逻辑
	if err := business.Run(ctx); err != nil {
		log.Fatalf("Error exit: %s", err)
	}
}

组装第一层程序结构，主要是：命令行，外部依赖，端口监听。 这里以常见的 HTTP 服务器为例。

// business/app.go
package business

type App struct {
	module.CLI    // 命令行
	module.SQLite // 数据库
	module.HTTP   // HTTP 服务
}

func Run(ctx context.Context) (err error) {
	var app App
	return module.Run(ctx, &app)
}

module.Run() 会依次初始化 App 中声明的每个 module ，然后运行所有实现了 Run() 的 module 。

初始化的大致流程为：module1.PreInit() -> module1.Init() -> module1.PostInit() -> module2.PreInit() -> ...

至于模块如何声明，就以 module.SQLite 为例：其中 Init() 和 Close() 都是 module 的可选接口。 module.HTTP 还需要实现 Run()。

// internal/module/sqlite.go
package module

type SQLite struct {
	DSN string
	DB  *sql.DB
}

func (m *SQLite) Init(ctx context.Context) (err error) {
	if m.DB, err = sql.Open("sqlite3", m.DSN); err != nil {
		return
	}

	ctx, cancel := context.WithTimeout(ctx, time.Second)
	defer cancel()

	return m.DB.PingContext(ctx)
}

func (m *SQLite) Close() error {
	return m.DB.Close()
}

module.HTTP 和 module.CLI 代码比较长就不贴了，完整的 Demo 在 Github： https://github.com/GotaX/module-demo

有疑问或者想分享下自己是如何组织代码的也欢迎留言讨论。

[dacapoday]

我选择 fx
https://pkg.go.dev/go.uber.org/fx#section-readme