Skip to content

Commit 999d284

Browse files
sonakshireddymSonakshi Reddy Morramganti
andauthored
Get backup field values in backup.yaml from environment variables (#693)
* Add backup fields as env vars Signed-off-by: Sonakshi Reddy Morramganti <sonakshi.reddymorramganti@reddit.com> * Add backup fields as env vars Signed-off-by: Sonakshi Reddy Morramganti <sonakshi.reddymorramganti@reddit.com> --------- Signed-off-by: Sonakshi Reddy Morramganti <sonakshi.reddymorramganti@reddit.com> Co-authored-by: Sonakshi Reddy Morramganti <sonakshi.reddymorramganti@reddit.com>
1 parent 0b9e819 commit 999d284

1 file changed

Lines changed: 43 additions & 0 deletions

File tree

core/paramtable/base_table.go

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,7 @@ func (gp *BaseTable) loadFromYaml(file string) {
143143
func (gp *BaseTable) tryLoadFromEnv() {
144144
gp.loadMinioConfig()
145145
gp.loadMilvusConfig()
146+
gp.loadBackupConfig()
146147
}
147148

148149
// Load loads an object with @key.
@@ -562,3 +563,45 @@ func (gp *BaseTable) loadMilvusConfig() {
562563
_ = gp.Save("milvus.rpcChannelName", milvusRPCChannelName)
563564
}
564565
}
566+
567+
func (gp *BaseTable) loadBackupConfig() {
568+
backupMaxSegmentGroupSize := os.Getenv("BACKUP_MAX_SEGMENT_GROUP_SIZE")
569+
if backupMaxSegmentGroupSize != "" {
570+
_ = gp.Save("backup.maxSegmentGroupSize", backupMaxSegmentGroupSize)
571+
}
572+
573+
backupParallelismBackupCollection := os.Getenv("BACKUP_PARALLELISM_BACKUP_COLLECTION")
574+
if backupParallelismBackupCollection != "" {
575+
_ = gp.Save("backup.parallelism.backupCollection", backupParallelismBackupCollection)
576+
}
577+
578+
backupParallelismCopydata := os.Getenv("BACKUP_PARALLELISM_COPYDATA")
579+
if backupParallelismCopydata != "" {
580+
_ = gp.Save("backup.parallelism.copydata", backupParallelismCopydata)
581+
}
582+
583+
backupParallelismRestoreCollection := os.Getenv("BACKUP_PARALLELISM_RESTORE_COLLECTION")
584+
if backupParallelismRestoreCollection != "" {
585+
_ = gp.Save("backup.parallelism.restoreCollection", backupParallelismRestoreCollection)
586+
}
587+
588+
backupKeepTempFiles := os.Getenv("BACKUP_KEEP_TEMP_FILES")
589+
if backupKeepTempFiles != "" {
590+
_ = gp.Save("backup.keepTempFiles", backupKeepTempFiles)
591+
}
592+
593+
backupGCPauseEnable := os.Getenv("BACKUP_GC_PAUSE_ENABLE")
594+
if backupGCPauseEnable != "" {
595+
_ = gp.Save("backup.gcPause.enable", backupGCPauseEnable)
596+
}
597+
598+
backupGCPauseSeconds := os.Getenv("BACKUP_GC_PAUSE_SECONDS")
599+
if backupGCPauseSeconds != "" {
600+
_ = gp.Save("backup.gcPause.seconds", backupGCPauseSeconds)
601+
}
602+
603+
backupGCPauseAddress := os.Getenv("BACKUP_GC_PAUSE_ADDRESS")
604+
if backupGCPauseAddress != "" {
605+
_ = gp.Save("backup.gcPause.address", backupGCPauseAddress)
606+
}
607+
}

0 commit comments

Comments
 (0)