mirror of
https://github.com/acepanel/panel.git
synced 2026-02-04 17:17:13 +08:00
132 lines
3.5 KiB
Go
132 lines
3.5 KiB
Go
package data
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"slices"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/go-resty/resty/v2"
|
|
|
|
"github.com/TheTNB/panel/internal/biz"
|
|
"github.com/TheTNB/panel/internal/http/request"
|
|
"github.com/TheTNB/panel/pkg/shell"
|
|
"github.com/TheTNB/panel/pkg/types"
|
|
"github.com/TheTNB/panel/pkg/types/docker/network"
|
|
)
|
|
|
|
type containerNetworkRepo struct {
|
|
client *resty.Client
|
|
}
|
|
|
|
func NewContainerNetworkRepo(sock ...string) biz.ContainerNetworkRepo {
|
|
if len(sock) == 0 {
|
|
sock = append(sock, "/var/run/docker.sock")
|
|
}
|
|
client := resty.New()
|
|
client.SetTimeout(1 * time.Minute)
|
|
client.SetRetryCount(2)
|
|
client.SetTransport(&http.Transport{
|
|
DialContext: func(ctx context.Context, _ string, _ string) (net.Conn, error) {
|
|
return (&net.Dialer{}).DialContext(ctx, "unix", sock[0])
|
|
},
|
|
})
|
|
client.SetBaseURL("http://d/v1.40")
|
|
|
|
return &containerNetworkRepo{
|
|
client: client,
|
|
}
|
|
}
|
|
|
|
// List 列出网络
|
|
func (r *containerNetworkRepo) List() ([]types.ContainerNetwork, error) {
|
|
var resp []network.Network
|
|
_, err := r.client.R().SetResult(&resp).Get("/networks")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var networks []types.ContainerNetwork
|
|
for _, item := range resp {
|
|
ipamConfigs := make([]types.ContainerNetworkIPAMConfig, 0)
|
|
for _, ipam := range item.IPAM.Config {
|
|
ipamConfigs = append(ipamConfigs, types.ContainerNetworkIPAMConfig{
|
|
Subnet: ipam.Subnet,
|
|
IPRange: ipam.IPRange,
|
|
Gateway: ipam.Gateway,
|
|
AuxAddress: ipam.AuxAddress,
|
|
})
|
|
}
|
|
networks = append(networks, types.ContainerNetwork{
|
|
ID: item.ID,
|
|
Name: item.Name,
|
|
Driver: item.Driver,
|
|
IPv6: item.EnableIPv6,
|
|
Internal: item.Internal,
|
|
Attachable: item.Attachable,
|
|
Ingress: item.Ingress,
|
|
Scope: item.Scope,
|
|
CreatedAt: item.Created,
|
|
IPAM: types.ContainerNetworkIPAM{
|
|
Driver: item.IPAM.Driver,
|
|
Options: types.MapToKV(item.IPAM.Options),
|
|
Config: ipamConfigs,
|
|
},
|
|
Options: types.MapToKV(item.Options),
|
|
Labels: types.MapToKV(item.Labels),
|
|
})
|
|
}
|
|
|
|
slices.SortFunc(networks, func(a types.ContainerNetwork, b types.ContainerNetwork) int {
|
|
return strings.Compare(a.Name, b.Name)
|
|
})
|
|
|
|
return networks, nil
|
|
}
|
|
|
|
// Create 创建网络
|
|
func (r *containerNetworkRepo) Create(req *request.ContainerNetworkCreate) (string, error) {
|
|
var sb strings.Builder
|
|
|
|
sb.WriteString(fmt.Sprintf("docker network create --driver %s", req.Driver))
|
|
sb.WriteString(fmt.Sprintf(" %s", req.Name))
|
|
|
|
if req.Ipv4.Enabled {
|
|
sb.WriteString(fmt.Sprintf(" --subnet %s", req.Ipv4.Subnet))
|
|
sb.WriteString(fmt.Sprintf(" --gateway %s", req.Ipv4.Gateway))
|
|
if req.Ipv4.IPRange != "" {
|
|
sb.WriteString(fmt.Sprintf(" --ip-range %s", req.Ipv4.IPRange))
|
|
}
|
|
}
|
|
if req.Ipv6.Enabled {
|
|
sb.WriteString(fmt.Sprintf(" --subnet %s", req.Ipv6.Subnet))
|
|
sb.WriteString(fmt.Sprintf(" --gateway %s", req.Ipv6.Gateway))
|
|
if req.Ipv6.IPRange != "" {
|
|
sb.WriteString(fmt.Sprintf(" --ip-range %s", req.Ipv6.IPRange))
|
|
}
|
|
}
|
|
for _, label := range req.Labels {
|
|
sb.WriteString(fmt.Sprintf(" --label %s=%s", label.Key, label.Value))
|
|
}
|
|
for _, option := range req.Options {
|
|
sb.WriteString(fmt.Sprintf(" --opt %s=%s", option.Key, option.Value))
|
|
}
|
|
|
|
return shell.Exec(sb.String())
|
|
}
|
|
|
|
// Remove 删除网络
|
|
func (r *containerNetworkRepo) Remove(id string) error {
|
|
_, err := shell.ExecfWithTimeout(120*time.Second, "docker network rm -f %s", id)
|
|
return err
|
|
}
|
|
|
|
// Prune 清理未使用的网络
|
|
func (r *containerNetworkRepo) Prune() error {
|
|
_, err := shell.ExecfWithTimeout(120*time.Second, "docker network prune -f")
|
|
return err
|
|
}
|