Fix CONFIG_DB_INITIALIZED flag check logic and set/reset flag for war…

…mboot (#15685)

* Fix CONFIG_DB_INITIALIZED flag check logic and set/reset flag for warm-reboot
* Fix db-cli usage
* Handle same image warm-reboot and generalize handling of INIT flag
* Cover boot from ONIE case: set config init flag when minigraph, config_db are missing
* Handle case: first boot of SONiC
* Check for config init flag
* Simplify logic, and do not call db_migrator for same image reboot

files/build_templates/docker_image_ctl.j2

@@ -229,6 +229,7 @@ function postStartAction()
  ($(docker exec -i database$DEV sonic-db-cli PING | grep -c PONG) -gt 0) ]]; do
  sleep 1;
  done
+
  if [[ ("$BOOT_TYPE" == "warm" || "$BOOT_TYPE" == "fastfast" || "$BOOT_TYPE" == "fast") && -f $WARM_DIR/dump.rdb ]]; then
  # retain the dump file from last boot for debugging purposes
  mv $WARM_DIR/dump.rdb $WARM_DIR/dump.rdb.old
@@ -246,28 +247,18 @@ function postStartAction()
  $SONIC_CFGGEN -j /etc/sonic/config_db$DEV.json --write-to-db
  fi
  fi
-
- if [[ "$BOOT_TYPE" == "fast" ]]; then
- # set the key to expire in 3 minutes
- $SONIC_DB_CLI STATE_DB SET "FAST_REBOOT|system" "1" "EX" "180"
- fi
-
- $SONIC_DB_CLI CONFIG_DB SET "CONFIG_DB_INITIALIZED" "1"
  fi
 
- if [ -e /tmp/pending_config_migration ]; then
+ if [ -e /tmp/pending_config_migration ] || [ -e /tmp/pending_config_initialization ]; then
  # this is first boot to a new image, config-setup execution is pending.
- # For fast/cold reboot case, DB contains nothing at this point
- # Call db_migrator after config-setup loads the config (from old config or minigraph)
- echo "Delaying db_migrator until config migration is over"
+ # for warmboot case, DB is loaded but migration is still pending
+ # For firstbboot/fast/cold reboot case, DB contains nothing at this point
+ # unset CONFIG_DB_INITIALIZED to indicate pending config load and migration
+ # This flag will be set to "1" after DB migration/initialization is completed as part of config-setup
+ $SONIC_DB_CLI CONFIG_DB SET "CONFIG_DB_INITIALIZED" "0"
  else
- # this is not a first time boot to a new image. Datbase container starts w/ old pre-existing config
- if [[ -x /usr/local/bin/db_migrator.py ]]; then
- # Migrate the DB to the latest schema version if needed
- if [ -z "$DEV" ]; then
- /usr/local/bin/db_migrator.py -o migrate
- fi
- fi
+ # set CONFIG_DB_INITIALIZED to indicate end of config load and migration
+ $SONIC_DB_CLI CONFIG_DB SET "CONFIG_DB_INITIALIZED" "1"
  fi
  # Add redis UDS to the redis group and give read/write access to the group
  REDIS_SOCK="/var/run/redis${DEV}/redis.sock"

files/image_config/config-setup/config-setup

@@ -252,6 +252,7 @@ do_config_initialization()
  fi
 
  rm -f /tmp/pending_config_initialization
+ sonic-db-cli CONFIG_DB SET "CONFIG_DB_INITIALIZED" "1"
 }
 
 # Restore config-setup post migration hooks from a backup copy
@@ -301,13 +302,14 @@ check_all_config_db_present()
 }
 
 # DB schema is subject to change between two images
-# Perform DB schema migration after loading backup config from previous image
+# Perform DB schema migration after loading backup config/minigraph from previous image
 do_db_migration()
 {
  if [[ -x /usr/local/bin/db_migrator.py ]]; then
  # Migrate the DB to the latest schema version if needed
  /usr/local/bin/db_migrator.py -o migrate
  fi
+ sonic-db-cli CONFIG_DB SET "CONFIG_DB_INITIALIZED" "1"
 }
 
 # Perform configuration migration from backup copy.

files/image_config/warmboot-finalizer/finalize-warmboot.sh

@@ -74,7 +74,7 @@ function wait_for_database_service()
  done
 
  # Wait for configDB initialization
- until [[ $(sonic-db-cli CONFIG_DB GET "CONFIG_DB_INITIALIZED") ]];
+ until [[ $(sonic-db-cli CONFIG_DB GET "CONFIG_DB_INITIALIZED") -eq 1 ]];
  do sleep 1;
  done
 

files/scripts/swss.sh

@@ -87,7 +87,7 @@ function wait_for_database_service()
  done
 
  # Wait for configDB initialization
- until [[ $($SONIC_DB_CLI CONFIG_DB GET "CONFIG_DB_INITIALIZED") ]];
+ until [[ $($SONIC_DB_CLI CONFIG_DB GET "CONFIG_DB_INITIALIZED") -eq 1 ]];
  do sleep 1;
  done
 }

files/scripts/syncd_common.sh

@@ -66,7 +66,7 @@ function wait_for_database_service()
  done
 
  # Wait for configDB initialization
- until [[ $($SONIC_DB_CLI CONFIG_DB GET "CONFIG_DB_INITIALIZED") ]];
+ until [[ $($SONIC_DB_CLI CONFIG_DB GET "CONFIG_DB_INITIALIZED") -eq 1 ]];
  do sleep 1;
  done
 }