56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package docker
|
||
|
||
import (
|
||
"github.com/docker/docker/client"
|
||
"github.com/gofiber/fiber/v2/log"
|
||
|
||
"gitee.ltd/lxh/wechat-robot/internal/config"
|
||
)
|
||
|
||
var (
|
||
dockerClient *client.Client
|
||
)
|
||
|
||
// InitClient 初始化Docker客户端
|
||
func InitClient() {
|
||
var err error
|
||
defer func() {
|
||
if err != nil {
|
||
log.Panicf("Docker客户端初始化失败: %v", err)
|
||
}
|
||
}()
|
||
|
||
options := []client.Opt{
|
||
client.WithAPIVersionNegotiation(),
|
||
}
|
||
|
||
// 如果指定了Docker主机地址
|
||
if config.Scd.Docker.Host != "" {
|
||
options = append(options, client.WithHost(config.Scd.Docker.Host))
|
||
}
|
||
|
||
// 如果指定了API版本
|
||
if config.Scd.Docker.APIVersion != "" {
|
||
options = append(options, client.WithVersion(config.Scd.Docker.APIVersion))
|
||
}
|
||
|
||
// 创建Docker客户端
|
||
dockerClient, err = client.NewClientWithOpts(options...)
|
||
}
|
||
|
||
// GetClient 获取Docker客户端实例
|
||
func GetClient() *client.Client {
|
||
if dockerClient == nil {
|
||
panic("Docker客户端未初始化，先调用InitClient")
|
||
}
|
||
return dockerClient
|
||
}
|
||
|
||
// CloseClient 关闭Docker客户端
|
||
func CloseClient() error {
|
||
if dockerClient != nil {
|
||
return dockerClient.Close()
|
||
}
|
||
return nil
|
||
}
|