feat(passport): add replica database support

Author: duanhongyi

charts/passport/templates/_helpers.tpl

@@ -73,6 +73,13 @@ env:
     secretKeyRef:
       name: passport-creds
       key: database-url
+{{- if (.Values.databaseUrl) }}
+- name: DRYCC_DATABASE_REPLICA_URL
+  valueFrom:
+    secretKeyRef:
+      name: passport-creds
+      key: database-replica-url
+{{- end }}
 {{- else if eq .Values.global.databaseLocation "on-cluster" }}
 - name: DRYCC_DATABASE_USER
   valueFrom:
@@ -85,7 +92,9 @@ env:
       name: database-creds
       key: password
 - name: DRYCC_DATABASE_URL
-  value: "postgres://$(DRYCC_DATABASE_USER):$(DRYCC_DATABASE_PASSWORD)@$(DRYCC_DATABASE_SERVICE_HOST):$(DRYCC_DATABASE_SERVICE_PORT)/passport"
+  value: "postgres://$(DRYCC_DATABASE_USER):$(DRYCC_DATABASE_PASSWORD)@drycc-database.{{.Release.Namespace}}.svc.{{.Values.global.clusterDomain}}:5432/passport"
+- name: DRYCC_DATABASE_REPLICA_URL
+  value: "postgres://$(DRYCC_DATABASE_USER):$(DRYCC_DATABASE_PASSWORD)@drycc-database-replica.{{.Release.Namespace}}.svc.{{.Values.global.clusterDomain}}:5432/passport"
 {{- end }}
 - name: DRYCC_REDIS_ADDRS
   valueFrom:

charts/passport/templates/passport-secret-creds.yaml

@@ -11,6 +11,9 @@ data:
   {{- if (.Values.databaseUrl) }}
   database-url: {{ .Values.databaseUrl | b64enc }}
   {{- end }}
+  {{- if (.Values.databaseReplicaUrl) }}
+  database-replica-url: {{ .Values.databaseReplicaUrl | b64enc }}
+  {{- end }}
   django-secret-key: {{ randAscii 64 | b64enc }}
   drycc-passport-controller-key: {{ randAlphaNum 40 | b64enc }}
   drycc-passport-controller-secret: {{ randAlphaNum 64 | b64enc }}

charts/passport/values.yaml

@@ -25,8 +25,9 @@ podAntiAffinityPreset:
 replicas: 1
 # limitsCpu: "100m"
 # limitsMemory: "50Mi"
-## Configuring this will no longer use the built-in database component
+## databaseUrl and databaseReplicaUrl are will no longer use the built-in database component
 databaseUrl: ""
+databaseReplicaUrl: ""
 # Any custom controller environment variables
 # can be specified as key-value pairs under environment
 # this is usually a non required setting.

rootfs/api/routers.py

@@ -0,0 +1,18 @@
+from django.conf import settings
+
+
+class DefaultReplicaRouter(object):
+
+    def db_for_read(self, model, **hints):
+        if 'replica' in settings.DATABASES:
+            return 'replica'
+        return 'default'
+
+    def db_for_write(self, model, **hints):
+        return 'default'
+
+    def allow_relation(self, obj1, obj2, **hints):
+        return True
+
+    def allow_migrate(self, db, app_label, model_name=None, **hints):
+        return True

rootfs/api/settings/production.py

@@ -265,12 +265,20 @@
 random_secret = ')u_jckp95wule8#wxd8sm!0tj2j&aveozu!nnpgl)2x&&16gfj'
 SECRET_KEY = os.environ.get('DRYCC_SECRET_KEY', random_secret)
 
-# database setting
+# database default setting
 DRYCC_DATABASE_URL = os.environ.get('DRYCC_DATABASE_URL', 'postgres://postgres:123456@49.232.207.93:5432/drycc_passport')  # noqa
 DATABASES = {
-    'default': dj_database_url.config(default=DRYCC_DATABASE_URL,
-                                      conn_max_age=600)
+    'default': dj_database_url.config(
+        default=DRYCC_DATABASE_URL, conn_max_age=600)
 }
+# database replica setting
+DRYCC_DATABASE_REPLICA_URL = os.environ.get('DRYCC_DATABASE_REPLICA_URL', None)
+if DRYCC_DATABASE_REPLICA_URL is not None:
+    DATABASES["replica"] = dj_database_url.config(
+        default=DRYCC_DATABASE_REPLICA_URL, conn_max_age=600)
+
+# database routers
+DATABASE_ROUTERS = ['api.routers.DefaultReplicaRouter', ]
 
 # Static files (CSS, JavaScript, Images)
 # https://docs.djangoproject.com/en/2.2/howto/static-files/