Initial changes - reading all banned ips from db

main.go

@@ -41,6 +41,8 @@ func main() {
 
 	db := sqlx.MustConnect("pgx", dsn)
 	models.DB = db
+	models.Migrate()
+	models.InitializeModelsFromDB()
 
 	if *shouldMigrate {
 		err := models.Migrate()

models/dataaccess.go

@@ -23,3 +23,29 @@ func GetIpAddressFromBannedIpsTable(ipAddressToBeQueried string) (string, error)
 
 	return bannedIp, nil
 }
+
+func GetAllIpAddressesFromBannedIpsTable() ([]string, error) {
+	rows, err := DB.Query(`
+								SELECT host(ip)
+								FROM bannedips`)
+
+	if err != nil {
+		log.Fatal(err)
+		return nil, err
+	}
+
+	defer rows.Close()
+	ipAddressList := make([]string, 0)
+
+	for rows.Next() {
+		var ipAddress string
+		scanRowErr := rows.Scan(&ipAddress)
+		if scanRowErr != nil {
+			log.Fatal(scanRowErr)
+			return nil, scanRowErr
+		}
+		ipAddressList = append(ipAddressList, ipAddress)
+	}
+
+	return ipAddressList, nil
+}

models/initializemodels.go

@@ -0,0 +1,17 @@
+package models
+
+import "github.com/golang/glog"
+
+func InitializeModelsFromDB() {
+	initializeBannedIpListModelFromDB()
+}
+
+func initializeBannedIpListModelFromDB() {
+	ipAddressList, err := GetAllIpAddressesFromBannedIpsTable()
+	if err != nil {
+		glog.Errorf("Failed to load banned ip address from the DB")
+		return
+	}
+
+	BannedIpAddresses = ipAddressList
+}

models/models.go

@@ -1,9 +1,11 @@
-// Copyright (c) 2021 Orange Forum authors. All rights reserved.
-// Use of this source code is governed by a BSD-style license that can be
-// found in the LICENSE file.
-
-package models
-
-import "github.com/jmoiron/sqlx"
-
-var DB *sqlx.DB
+// Copyright (c) 2021 Orange Forum authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package models
+
+import "github.com/jmoiron/sqlx"
+
+var DB *sqlx.DB
+
+var BannedIpAddresses []string