Move files for pg-backup-file from roles to postgres module
[mirror/dsa-puppet.git] / modules / postgres / files / backup_source / pg-backup-file
1 #!/bin/bash
2
3 # Copyright (c) 2010, 2011, 2013, 2014 Peter Palfrader
4 #
5 # Permission is hereby granted, free of charge, to any person obtaining
6 # a copy of this software and associated documentation files (the
7 # "Software"), to deal in the Software without restriction, including
8 # without limitation the rights to use, copy, modify, merge, publish,
9 # distribute, sublicense, and/or sell copies of the Software, and to
10 # permit persons to whom the Software is furnished to do so, subject to
11 # the following conditions:
12 #
13 # The above copyright notice and this permission notice shall be
14 # included in all copies or substantial portions of the Software.
15 #
16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17 # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18 # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
20 # LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
21 # OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
22 # WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
23
24 # Archive a postgresql file (WAL or BASE) to the archive server $backuphost
25 # can be set in /etc/dsa/pg-backup-file.conf
26
27 set -u
28
29 backuphost=debbackup@storace
30 myhost="`hostname`"
31 self="`basename "$0"`[$$]"
32
33 if [ "$#" != 3 ] ; then
34         echo >&2 "Usage: $self <cluster> <WAL|BASE> <file>"
35         exit 1
36 fi
37
38 ssh_options=""
39 ! [ -e /etc/dsa/pg-backup-file.conf ] || . /etc/dsa/pg-backup-file.conf
40
41 cluster="$1"
42 what="$2"
43 file="$3"
44
45 info() {
46         logger -p daemon.info -t "$self" "$1"
47 }
48
49 croak() {
50         logger -s -p daemon.warn -t "$self" "$1"
51         exit 1
52 }
53
54 if ! [ -e "$file" ] ; then
55         croak "file $file does not exist"
56 fi
57
58 size="`stat -c '%s' "$file"`"
59 checksum=""
60 bn="`basename "$file"`"
61 targetname="$cluster.$what.$bn"
62 logtuple_start="($myhost,$targetname,$size"
63 failures=""
64
65 for target in $backuphost; do
66         if [ "${target#/}" != "$target" ]; then
67                 info "Archiving to $target: ($targetname,$size)"
68                 cp "$file" "$target/$myhost-$targetname"
69         else
70                 [ -n "$checksum" ] || checksum="`sha512sum "$file" | awk '{print $1}'`"
71                 info "Archiving to $target: ($myhost,$targetname,$size,$checksum)"
72
73                 ssh -C "$target" $ssh_options -o BatchMode=yes -o ServerAliveInterval=90 $myhost store-file pg "$targetname" "$size" "$checksum" < "$file"
74                 if [ "$?" != 0 ]; then
75                         failures="$failures $logtuple_start,$checksum,$target)"
76                 fi
77         fi
78 done
79 if [ "$failures" != "" ]; then
80         croak "remote store for$failures failed."
81 fi