Adds init.sql for user defined commands at startup

Dockerfile

@@ -7,9 +7,11 @@ RUN pip install -U pip setuptools
 RUN pip install python-etcd
 
 RUN mkdir -p /governor/helpers
-ADD governor.py /governor/governor.py
-ADD helpers /governor/helpers
-ADD postgres0.yml /governor/
+COPY governor.py /governor/governor.py
+COPY helpers /governor/helpers
+COPY postgres0.yml /governor/
+COPY pg_hba.conf /governor/
+COPY init.sql /governor/
 
 RUN mkdir -p /data/postgres && \
   chown -R postgres /data && \
@@ -20,4 +22,3 @@ WORKDIR /governor
 
 CMD gosu postgres /governor/governor.py
 
-

governor.py

@@ -55,7 +55,7 @@ def shutdown():
             logging.info("Governor Shutting Down: Abdicating Leadership")
             etcd.abdicate(postgresql.name)
 
-        logging.info("Governor Shutting Down: Remiving Membership")
+        logging.info("Governor Shutting Down: Removing Membership")
         etcd.delete_member(postgresql.name)
     except:
         pass
@@ -93,6 +93,7 @@ def shutdown():
         postgresql.start(master=True)
     else:
         logging.info("Governor Starting up: Initialisation Race ... LOST")
+        time.sleep(20)
         logging.info("Governor Starting up: Sync Postgres from Leader")
         synced_from_leader = False
         while not synced_from_leader:

helpers/postgresql.py

@@ -71,8 +71,10 @@ def initialize(self):
         if os.system("initdb -D %s" % self.data_dir) == 0:
             #self.write_pg_hba()
             self.copy_pg_hba()
+
             self.start()
             self.create_replication_user()
+            self.run_init_sql()
             self.stop()
 
             return True
@@ -245,6 +247,15 @@ def demote(self, leader):
         self.stop()
         self.start(master=False)
 
+    def run_init_sql(self):
+        if os.path.exists('init.sql'):
+            logger.info("Running init.sql")
+            sql_lines = open("init.sql", "r").read().split("\n")
+            for sql in sql_lines:
+                if not sql:
+                    continue
+                self.query(sql)
+
     def create_replication_user(self):
         #logger.info("Governor Starting Up: Running postgres single user mode to create repliaction user")
         #os.system("postgres --single -jE << CREATE USER '%s' WITH REPLICATION ENCRYPTED PASSWORD '%s';" % (self.replication["username"], self.replication["password"]))

init.sql.example

@@ -0,0 +1,4 @@
+
+CREATE USER example WITH ENCRYPTED PASSWORD 'examplepass';
+
+CREATE DATABASE example;