Make tile server export script require a valid TOTP token

[chef.git] / cookbooks / tile / templates / default / replicate.erb

diff --git a/cookbooks/tile/templates/default/replicate.erb b/cookbooks/tile/templates/default/replicate.erb
index 5d940375bf36e60d6e96a19c5b5692fc6b3e6a1e..1b6c61bbc96f588ee5810a9f6e71b9cfce5ffd86 100644 (file)
--- a/cookbooks/tile/templates/default/replicate.erb
+++ b/cookbooks/tile/templates/default/replicate.erb
@@ -2,9 +2,15 @@
 
 # DO NOT EDIT - This file is being maintained by Chef
 
 
 # DO NOT EDIT - This file is being maintained by Chef
 

-# Initialize timestamp with day of latest planet dump
-# Setting to midnight ensures we get conistent data after first run
-# osmosis --read-replication-interval-init
+# Before running updates, the replication needs to be set up with the timestamp
+# set to the day of the latest planet dump. Setting to midnight ensures we get
+# conistent data after first run. osmosis --read-replication-interval-init is
+# used to initially create the state file
+
+# Define exit handler
+function onexit {
+    [ -f state-prev.txt ] && mv state-prev.txt state.txt
+}

 
 # Send output to the log
 exec > /var/log/tile/replicate.log 2>&1
 
 # Send output to the log
 exec > /var/log/tile/replicate.log 2>&1


@@ -12,6 +18,9 @@ exec > /var/log/tile/replicate.log 2>&1
 # Change to the replication state directory
 cd /var/lib/replicate
 
 # Change to the replication state directory
 cd /var/lib/replicate
 

+# Install exit handler
+trap onexit EXIT
+

 # Read in initial state
 . state.txt
 
 # Read in initial state
 . state.txt
 


@@ -21,6 +30,9 @@ do
     # Work out the name of the next file
     file="changes-${sequenceNumber}.osm.gz"
 
     # Work out the name of the next file
     file="changes-${sequenceNumber}.osm.gz"
 

+    # Save state file so we can rollback if an error occurs
+    cp state.txt state-prev.txt
+

     # Fetch the next set of changes
     osmosis --read-replication-interval --simc --write-xml-change file="${file}" compressionMethod="gzip"
 
     # Fetch the next set of changes
     osmosis --read-replication-interval --simc --write-xml-change file="${file}" compressionMethod="gzip"
 


@@ -28,50 +40,56 @@ do
     if [ $? -eq 0 ]
     then
         # Enable exit on error
     if [ $? -eq 0 ]
     then
         # Enable exit on error

-       set -e
+        set -e

 
         # Remember the previous sequence number
 
         # Remember the previous sequence number

-       prevSequenceNumber=$sequenceNumber
+        prevSequenceNumber=$sequenceNumber

 
         # Read in new state
 
         # Read in new state

-       . state.txt
+        . state.txt

 
         # Did we get any new data?
 
         # Did we get any new data?

-       if [ "${sequenceNumber}" == "${prevSequenceNumber}" ]
+        if [ "${sequenceNumber}" == "${prevSequenceNumber}" ]

         then
             # Log the lack of data
         then
             # Log the lack of data

-           echo "No new data available. Sleeping..."
+            echo "No new data available. Sleeping..."

 
 

-           # Remove file, it will just be an empty changeset
-           rm ${file}
+            # Remove file, it will just be an empty changeset
+            rm ${file}
+
+            # No need to rollback now
+            rm state-prev.txt

 
             # Sleep for a short while
 
             # Sleep for a short while

-           sleep 30
-       else
+            sleep 30
+        else

             # Log the new data
             # Log the new data

-           echo "Fetched new data from ${prevSequenceNumber} to ${sequenceNumber} into ${file}"
+            echo "Fetched new data from ${prevSequenceNumber} to ${sequenceNumber} into ${file}"

 
             # Apply the changes to the database
 <% if node[:tile][:node_file] -%>
 
             # Apply the changes to the database
 <% if node[:tile][:node_file] -%>

-            osm2pgsql --slim --append --flat-nodes=<%= node[:tile][:node_file] %> ${file}
+            osm2pgsql --slim --append --number-processes=1 --flat-nodes=<%= node[:tile][:node_file] %> ${file}

 <% else -%>
 <% else -%>

-            osm2pgsql --slim --append ${file}
+            osm2pgsql --slim --append --number-processes=1 ${file}

 <% end -%>
 
 <% end -%>
 

+            # No need to rollback now
+            rm state-prev.txt
+

             # Expire tiles which are touched by the changes
             sudo -u www-data /usr/local/bin/expire-tiles ${file} > /dev/null 2>&1 &
             # Expire tiles which are touched by the changes
             sudo -u www-data /usr/local/bin/expire-tiles ${file} > /dev/null 2>&1 &

-       fi
+        fi

 
         # Delete old downloads
 
         # Delete old downloads

-       find . -name 'changes-*.gz' -mmin +300 -exec rm -f {} \;
+        find . -name 'changes-*.gz' -mmin +300 -exec rm -f {} \;

 
         # Disable exit on error
         set +e
     else
         # Log our failure to fetch changes
 
         # Disable exit on error
         set +e
     else
         # Log our failure to fetch changes

-       echo "Failed to fetch changes - waiting a few minutes before retry"
+        echo "Failed to fetch changes - waiting a few minutes before retry"

 
         # Wait five minutes and have another go
 
         # Wait five minutes and have another go

-       sleep 300
+        sleep 300

     fi
 done
     fi
 done