DataContainer-Utils.psm1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
function Backup-DataContainer ($Config) {
    $backupContainer = "${Config.container}-data"
    $volumeArgs = -join ($Config.volumes | ForEach-Object { "-v /backup/${_}:${_} " })
    $cpCmds = -join ($Config.volumes | ForEach-Object { "cp -Rf /backup/${_} ${_}" })
    # backup data container's volumes
    docker "run --volumes-from ${Config.container} --name $backupContainer $volumeArgs alpine $cpCmds"
    if ($LASTEXITCODE -ne 0) {
        exit $LASTEXITCODE
    }

    # commit backup to registry
    $remote = "${Config.registry}$backupContainer"
    docker commit $backupContainer $remote
    if ($LASTEXITCODE -ne 0) {
        exit $LASTEXITCODE
    }

    docker push $registry
    if ($LASTEXITCODE -ne 0) {
        exit $LASTEXITCODE
    }
}

function Restore-DataContainer ($Config) {
    $backupContainer = "${Config.container}-data"
    $remote = "${Config.registry}$backupContainer"
    $volumeArgs = -join ($Config.volumes | ForEach-Object { "-v /backup/${_}:${_} " })
    docker "run $volumeArgs --entrypoint "bin/sh" --name $backupContainer $remote"
    if ($LASTEXITCODE -ne 0) {
        exit $LASTEXITCODE
    }
}

function Backup-FromConfig {
    param(
        # ConfigFile
        [Parameter(Mandatory = $true)]
        [string]
        $ConfigFile
    )
    $ErrorActionPreference = 'stop'
    $configs = Get-Content $ConfigFile -Raw | ConvertFrom-Json

    foreach ($cfg in $configs) {
        Backup-DataContainer($cfg)
    }
}

function Restore-FromConfig {
    param(
        # ConfigFile
        [Parameter(Mandatory = $true)]
        [string]
        $ConfigFile
    )
    $ErrorActionPreference = 'stop'
    $configs = Get-Content $ConfigFile -Raw | ConvertFrom-Json

    foreach ($cfg in $configs) {
        Restore-DataContainer($cfg)
    }
}

function Get-DataContainerConfig {
    param(
        # ConfigFile
        [Parameter(Mandatory = $true)]
        [string]
        $ConfigFile
    )
    foreach ($cfg in $configs) {
        Write-Output "${cfg.container} >> ${cfg.registry}${cfg.container}-data"
        foreach ($path in $cfg.volumes) {
            Write-Output " $path"
        }
        Write-Output ""
    }
}



Export-ModuleMember -Function Backup-FromConfig
Export-ModuleMember -Function Restore-FromConfig
Export-ModuleMember -Function Get-DataContainerConfig