Added a new .env variable

Author: kpachhai

Diff for: .env.example

@@ -4,4 +4,5 @@ DB_USER=postgres
 DB_PASSWORD=postgres
 DB_NAME=nuklaivm
 DB_SSLMODE=require # Or "disable" if you don't want to use SSL
+DB_RESET=true # Set to "true" to reset the database on every restart
 GRPC_WHITELISTED_BLOCKCHAIN_NODES="172.17.1.56" # "127.0.0.1,localhost,::1" is already included by default. You can even include something like myblockchain.aws.com

Diff for: config/config.go

@@ -16,12 +16,12 @@ import (
 func GetDatabaseURL() string {
 	// Retrieve env variable for DB_HOST and set default value if not present
 
-	host := getEnv("DB_HOST", "localhost")
-	port := getEnv("DB_PORT", "5432")
-	user := getEnv("DB_USER", "postgres")
-	password := getEnv("DB_PASSWORD", "postgres")
-	dbName := getEnv("DB_NAME", "nuklaivm")
-	sslMode := getEnv("DB_SSL_MODE", "disable")
+	host := GetEnv("DB_HOST", "localhost")
+	port := GetEnv("DB_PORT", "5432")
+	user := GetEnv("DB_USER", "postgres")
+	password := GetEnv("DB_PASSWORD", "postgres")
+	dbName := GetEnv("DB_NAME", "nuklaivm")
+	sslMode := GetEnv("DB_SSL_MODE", "disable")
 
 	// Encode password to handle special characters
 	encodedPassword := url.QueryEscape(password)
@@ -36,7 +36,7 @@ func GetDatabaseURL() string {
 // and resolves domain names to IPs.
 // GetWhitelistIPs retrieves the list of whitelisted IPs and CIDR ranges
 func GetWhitelistIPs() ([]string, []string) {
-	ipList := getEnv("GRPC_WHITELISTED_BLOCKCHAIN_NODES", "127.0.0.1,localhost,::1")
+	ipList := GetEnv("GRPC_WHITELISTED_BLOCKCHAIN_NODES", "127.0.0.1,localhost,::1")
 	entries := strings.Split(ipList, ",")
 
 	whitelistIPs := []string{}
@@ -63,9 +63,9 @@ func GetWhitelistIPs() ([]string, []string) {
 	return uniqueStrings(whitelistIPs), uniqueStrings(whitelistCIDRs)
 }
 
-// getEnv retrieves the value of the environment variable named by the key.
+// GetEnv retrieves the value of the environment variable named by the key.
 // If the variable is not present, it returns the default value.
-func getEnv(key, defaultValue string) string {
+func GetEnv(key, defaultValue string) string {
 	if value, exists := os.LookupEnv(key); exists {
 		return value
 	}

Diff for: db/db.go

@@ -9,6 +9,7 @@ import (
 	"log"
 
 	_ "github.com/lib/pq"
+	"github.com/nuklai/nuklaivm-external-subscriber/config"
 )
 
 // InitDB initializes the database connection and creates the schema if it doesn't exist
@@ -24,6 +25,18 @@ func InitDB(connStr string) (*sql.DB, error) {
 
 	log.Println("Database connection established")
 
+	reset := config.GetEnv("DB_RESET", "false") == "true"
+	if reset {
+		// Drop all existing tables
+		log.Println("Resetting the database...")
+		_, err := db.Exec(`
+			DROP TABLE IF EXISTS blocks, transactions, actions, assets, genesis_data CASCADE;
+		`)
+		if err != nil {
+			return nil, fmt.Errorf("error resetting the database: %w", err)
+		}
+	}
+
 	// Ensure schema is created
 	if err := CreateSchema(db); err != nil {
 		return nil, fmt.Errorf("error creating schema: %w", err)

Diff for: infra/aws/task-definition-subscriber.json

@@ -67,6 +67,10 @@
           "name": "DB_PASSWORD",
           "valueFrom": "arn:aws:ssm:${AWS_REGION}:${AWS_ACCOUNT_ID}:parameter/${ENVIRONMENT}/${PRODUCT}/${APPLICATION}/db_password"
         },
+        {
+          "name": "DB_RESET",
+          "valueFrom": "arn:aws:ssm:${AWS_REGION}:${AWS_ACCOUNT_ID}:parameter/${ENVIRONMENT}/${PRODUCT}/${APPLICATION}/db_reset"
+        },
         {
           "name": "GRPC_WHITELISTED_BLOCKCHAIN_NODES",
           "valueFrom": "arn:aws:ssm:${AWS_REGION}:${AWS_ACCOUNT_ID}:parameter/${ENVIRONMENT}/${PRODUCT}/${APPLICATION}/grpc_whitelisted_blockchain_nodes"