Golang Go语言中 心得分享，如何组织入口文件

这里讨论的是不用框架只用工具库（ 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

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