1Panel/backend/app/service/container_volume.go

135 lines
3.3 KiB
Go
Raw Normal View History

2022-10-18 10:39:45 +00:00
package service
import (
"context"
2022-12-06 10:24:26 +00:00
"sort"
2023-02-07 10:48:32 +00:00
"strings"
2022-10-18 10:39:45 +00:00
"time"
"github.com/1Panel-dev/1Panel/backend/app/dto"
"github.com/1Panel-dev/1Panel/backend/buserr"
2023-03-11 03:32:24 +00:00
"github.com/1Panel-dev/1Panel/backend/constant"
2022-10-18 10:39:45 +00:00
"github.com/1Panel-dev/1Panel/backend/utils/docker"
"github.com/docker/docker/api/types/filters"
"github.com/docker/docker/api/types/volume"
)
2023-02-07 10:48:32 +00:00
func (u *ContainerService) PageVolume(req dto.SearchWithPage) (int64, interface{}, error) {
2022-10-18 10:39:45 +00:00
client, err := docker.NewDockerClient()
if err != nil {
return 0, nil, err
}
list, err := client.VolumeList(context.TODO(), volume.ListOptions{})
2022-10-18 10:39:45 +00:00
if err != nil {
return 0, nil, err
}
2023-02-07 10:48:32 +00:00
if len(req.Info) != 0 {
length, count := len(list.Volumes), 0
for count < length {
2023-02-07 10:48:32 +00:00
if !strings.Contains(list.Volumes[count].Name, req.Info) {
list.Volumes = append(list.Volumes[:count], list.Volumes[(count+1):]...)
length--
2023-02-07 10:48:32 +00:00
} else {
count++
}
}
}
2022-10-18 10:39:45 +00:00
var (
data []dto.Volume
records []*volume.Volume
2022-10-18 10:39:45 +00:00
)
2022-12-06 10:24:26 +00:00
sort.Slice(list.Volumes, func(i, j int) bool {
return list.Volumes[i].CreatedAt > list.Volumes[j].CreatedAt
})
2022-10-18 10:39:45 +00:00
total, start, end := len(list.Volumes), (req.Page-1)*req.PageSize, req.Page*req.PageSize
if start > total {
records = make([]*volume.Volume, 0)
2022-10-18 10:39:45 +00:00
} else {
if end >= total {
end = total
}
records = list.Volumes[start:end]
}
for _, item := range records {
tag := make([]string, 0)
for _, val := range item.Labels {
tag = append(tag, val)
}
2023-02-23 03:39:47 +00:00
if len(item.CreatedAt) > 19 {
item.CreatedAt = item.CreatedAt[0:19]
}
createTime, _ := time.Parse("2006-01-02T15:04:05", item.CreatedAt)
2022-10-18 10:39:45 +00:00
data = append(data, dto.Volume{
CreatedAt: createTime,
Name: item.Name,
Driver: item.Driver,
Mountpoint: item.Mountpoint,
Labels: tag,
})
}
return int64(total), data, nil
}
func (u *ContainerService) ListVolume() ([]dto.Options, error) {
client, err := docker.NewDockerClient()
if err != nil {
return nil, err
}
list, err := client.VolumeList(context.TODO(), volume.ListOptions{})
2022-10-18 10:39:45 +00:00
if err != nil {
return nil, err
}
var datas []dto.Options
2022-10-18 10:39:45 +00:00
for _, item := range list.Volumes {
datas = append(datas, dto.Options{
2022-10-18 10:39:45 +00:00
Option: item.Name,
})
}
sort.Slice(datas, func(i, j int) bool {
return datas[i].Option < datas[j].Option
})
return datas, nil
2022-10-18 10:39:45 +00:00
}
func (u *ContainerService) DeleteVolume(req dto.BatchDelete) error {
client, err := docker.NewDockerClient()
if err != nil {
return err
}
for _, id := range req.Names {
2022-10-18 10:39:45 +00:00
if err := client.VolumeRemove(context.TODO(), id, true); err != nil {
if strings.Contains(err.Error(), "volume is in use") {
return buserr.WithDetail(constant.ErrInUsed, id, nil)
}
2022-10-18 10:39:45 +00:00
return err
}
}
return nil
}
func (u *ContainerService) CreateVolume(req dto.VolumeCreate) error {
2022-10-18 10:39:45 +00:00
client, err := docker.NewDockerClient()
if err != nil {
return err
}
2023-03-11 03:32:24 +00:00
var array []filters.KeyValuePair
array = append(array, filters.Arg("name", req.Name))
vos, _ := client.VolumeList(context.TODO(), volume.ListOptions{})
2023-03-11 03:32:24 +00:00
if len(vos.Volumes) != 0 {
for _, v := range vos.Volumes {
if v.Name == req.Name {
return constant.ErrRecordExist
}
}
}
options := volume.CreateOptions{
2022-10-18 10:39:45 +00:00
Name: req.Name,
Driver: req.Driver,
DriverOpts: stringsToMap(req.Options),
Labels: stringsToMap(req.Labels),
}
2023-03-15 04:58:56 +00:00
if _, err := client.VolumeCreate(context.TODO(), options); err != nil {
2022-10-18 10:39:45 +00:00
return err
}
return nil
}