mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 14:20:23 -04:00
168 lines
3.7 KiB
Go
168 lines
3.7 KiB
Go
// +build !windows
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/docker/docker/daemon/graphdriver/devmapper"
|
|
"github.com/docker/docker/pkg/devicemapper"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func usage() {
|
|
fmt.Fprintf(os.Stderr, "Usage: %s <flags> [status] | [list] | [device id] | [resize new-pool-size] | [snap new-id base-id] | [remove id] | [mount id mountpoint]\n", os.Args[0])
|
|
flag.PrintDefaults()
|
|
os.Exit(1)
|
|
}
|
|
|
|
func byteSizeFromString(arg string) (int64, error) {
|
|
digits := ""
|
|
rest := ""
|
|
last := strings.LastIndexAny(arg, "0123456789")
|
|
if last >= 0 {
|
|
digits = arg[:last+1]
|
|
rest = arg[last+1:]
|
|
}
|
|
|
|
val, err := strconv.ParseInt(digits, 10, 64)
|
|
if err != nil {
|
|
return val, err
|
|
}
|
|
|
|
rest = strings.ToLower(strings.TrimSpace(rest))
|
|
|
|
var multiplier int64 = 1
|
|
switch rest {
|
|
case "":
|
|
multiplier = 1
|
|
case "k", "kb":
|
|
multiplier = 1024
|
|
case "m", "mb":
|
|
multiplier = 1024 * 1024
|
|
case "g", "gb":
|
|
multiplier = 1024 * 1024 * 1024
|
|
case "t", "tb":
|
|
multiplier = 1024 * 1024 * 1024 * 1024
|
|
default:
|
|
return 0, fmt.Errorf("Unknown size unit: %s", rest)
|
|
}
|
|
|
|
return val * multiplier, nil
|
|
}
|
|
|
|
func main() {
|
|
root := flag.String("r", "/var/lib/docker", "Docker root dir")
|
|
flDebug := flag.Bool("D", false, "Debug mode")
|
|
|
|
flag.Parse()
|
|
|
|
if *flDebug {
|
|
os.Setenv("DEBUG", "1")
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
}
|
|
|
|
if flag.NArg() < 1 {
|
|
usage()
|
|
}
|
|
|
|
args := flag.Args()
|
|
|
|
home := path.Join(*root, "devicemapper")
|
|
devices, err := devmapper.NewDeviceSet(home, false, nil, nil, nil)
|
|
if err != nil {
|
|
fmt.Println("Can't initialize device mapper: ", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
switch args[0] {
|
|
case "status":
|
|
status := devices.Status()
|
|
fmt.Printf("Pool name: %s\n", status.PoolName)
|
|
fmt.Printf("Data Loopback file: %s\n", status.DataLoopback)
|
|
fmt.Printf("Metadata Loopback file: %s\n", status.MetadataLoopback)
|
|
fmt.Printf("Sector size: %d\n", status.SectorSize)
|
|
fmt.Printf("Data use: %d of %d (%.1f %%)\n", status.Data.Used, status.Data.Total, 100.0*float64(status.Data.Used)/float64(status.Data.Total))
|
|
fmt.Printf("Metadata use: %d of %d (%.1f %%)\n", status.Metadata.Used, status.Metadata.Total, 100.0*float64(status.Metadata.Used)/float64(status.Metadata.Total))
|
|
case "list":
|
|
ids := devices.List()
|
|
sort.Strings(ids)
|
|
for _, id := range ids {
|
|
fmt.Println(id)
|
|
}
|
|
case "device":
|
|
if flag.NArg() < 2 {
|
|
usage()
|
|
}
|
|
status, err := devices.GetDeviceStatus(args[1])
|
|
if err != nil {
|
|
fmt.Println("Can't get device info: ", err)
|
|
os.Exit(1)
|
|
}
|
|
fmt.Printf("Id: %d\n", status.DeviceID)
|
|
fmt.Printf("Size: %d\n", status.Size)
|
|
fmt.Printf("Transaction Id: %d\n", status.TransactionID)
|
|
fmt.Printf("Size in Sectors: %d\n", status.SizeInSectors)
|
|
fmt.Printf("Mapped Sectors: %d\n", status.MappedSectors)
|
|
fmt.Printf("Highest Mapped Sector: %d\n", status.HighestMappedSector)
|
|
case "resize":
|
|
if flag.NArg() < 2 {
|
|
usage()
|
|
}
|
|
|
|
size, err := byteSizeFromString(args[1])
|
|
if err != nil {
|
|
fmt.Println("Invalid size: ", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
err = devices.ResizePool(size)
|
|
if err != nil {
|
|
fmt.Println("Error resizing pool: ", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
case "snap":
|
|
if flag.NArg() < 3 {
|
|
usage()
|
|
}
|
|
|
|
err := devices.AddDevice(args[1], args[2], nil)
|
|
if err != nil {
|
|
fmt.Println("Can't create snap device: ", err)
|
|
os.Exit(1)
|
|
}
|
|
case "remove":
|
|
if flag.NArg() < 2 {
|
|
usage()
|
|
}
|
|
|
|
err := devicemapper.RemoveDevice(args[1])
|
|
if err != nil {
|
|
fmt.Println("Can't remove device: ", err)
|
|
os.Exit(1)
|
|
}
|
|
case "mount":
|
|
if flag.NArg() < 3 {
|
|
usage()
|
|
}
|
|
|
|
err := devices.MountDevice(args[1], args[2], "")
|
|
if err != nil {
|
|
fmt.Println("Can't mount device: ", err)
|
|
os.Exit(1)
|
|
}
|
|
default:
|
|
fmt.Printf("Unknown command %s\n", args[0])
|
|
usage()
|
|
|
|
os.Exit(1)
|
|
}
|
|
}
|