fix: DLE restore configuration(#230)

Author: agneum

pkg/retrieval/engine/postgres/physical/custom.go

@@ -41,6 +41,7 @@ func (c *custom) GetRecoveryConfig(pgVersion float64) map[string]string {
 
 		if pgVersion < defaults.PGVersion12 {
 			recoveryCfg["standby_mode"] = "on"
+			recoveryCfg["recovery_target_timeline"] = "latest"
 		}
 	}
 

pkg/retrieval/engine/postgres/physical/custom_test.go

@@ -13,8 +13,9 @@ func TestCustomRecoveryConfig(t *testing.T) {
 
 	recoveryConfig := customTool.GetRecoveryConfig(11.7)
 	expectedResponse11 := map[string]string{
-		"restore_command": "pg_basebackup -X stream -D dataDirectory",
-		"standby_mode":    "on",
+		"restore_command":          "pg_basebackup -X stream -D dataDirectory",
+		"standby_mode":             "on",
+		"recovery_target_timeline": "latest",
 	}
 	assert.Equal(t, expectedResponse11, recoveryConfig)
 

pkg/retrieval/engine/postgres/physical/wal_g.go

@@ -44,6 +44,7 @@ func (w *walg) GetRecoveryConfig(pgVersion float64) map[string]string {
 
 	if pgVersion < defaults.PGVersion12 {
 		recoveryCfg["standby_mode"] = "on"
+		recoveryCfg["recovery_target_timeline"] = "latest"
 	}
 
 	return recoveryCfg

pkg/retrieval/engine/postgres/physical/wal_g_test.go

@@ -11,8 +11,9 @@ func TestWALGRecoveryConfig(t *testing.T) {
 
 	recoveryConfig := walg.GetRecoveryConfig(11.7)
 	expectedResponse11 := map[string]string{
-		"restore_command": "wal-g wal-fetch %f %p",
-		"standby_mode":    "on",
+		"restore_command":          "wal-g wal-fetch %f %p",
+		"standby_mode":             "on",
+		"recovery_target_timeline": "latest",
 	}
 	assert.Equal(t, expectedResponse11, recoveryConfig)
 

pkg/services/provision/databases/postgres/pgconfig/configuration.go

@@ -146,6 +146,10 @@ func (m *Manager) isInitialized() (bool, error) {
 
 	postgresFile, err := os.Open(pgConfDst)
 	if err != nil {
+		if _, ok := err.(*os.PathError); ok {
+			return false, nil
+		}
+
 		return false, err
 	}
 
@@ -303,6 +307,10 @@ func (m *Manager) AdjustRecoveryFiles() error {
 
 // ApplyRecovery applies recovery configuration parameters.
 func (m *Manager) ApplyRecovery(cfg map[string]string) error {
+	if err := tools.TouchFile(path.Join(m.dataDir, m.recoveryFilename())); err != nil {
+		return err
+	}
+
 	if err := appendExtraConf(path.Join(m.dataDir, m.recoveryFilename()), cfg); err != nil {
 		return err
 	}
@@ -373,11 +381,11 @@ func (m *Manager) getConfigPath(configName string) string {
 
 // recoveryFilename returns the name of a recovery configuration file.
 func (m Manager) recoveryFilename() string {
-	if m.pgVersion > defaults.PGVersion12 {
+	if m.pgVersion >= defaults.PGVersion12 {
 		return configPrefix + pgConfName
 	}
 
-	return configPrefix + recoveryConfName
+	return recoveryConfName
 }
 
 // rewriteConfig completely rewrite a configuration file with provided parameters.