postgres.go

package foxkit

import (
	"fmt"
	"net/http"
	"os"

	"github.com/gin-gonic/gin"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

// creates a connection to Postgres using ENV
func ConnectSQL() *gorm.DB {
	// build connection URI
	uri := fmt.Sprintf("host=%v user=%v password=%v dbname=%v port=%v sslmode=%v TimeZone=%v",
		os.Getenv("POSTGRES_HOST"), os.Getenv("POSTGRES_USER"), os.Getenv("POSTGRES_PASS"),
		os.Getenv("POSTGRES_DB"), os.Getenv("POSTGRES_PORT"), os.Getenv("POSTGRES_SSLMODE"),
		os.Getenv("POSTGRES_TIMEZONE"))

	// connect to the DB
	db, err := gorm.Open(postgres.Open(uri), &gorm.Config{Logger: logger.Default.LogMode(logger.Warn)})
	if err != nil {
		ErrorFatal("FoxKit", err)
	}

	// check for errors
	if db.Error != nil {
		ErrorFatal("FoxKit", db.Error)
	}

	return db
}

// migrates all tables, panics on error
func AutoMigrateSQL(pg_conn *gorm.DB, inf ...interface{}) {
	if err := pg_conn.AutoMigrate(inf...); err != nil {
		ErrorFatal("FoxKit", err)
	}
}

// tries to push the data into the DB table, sets status to 500 if false
func StoreDB(c *gin.Context, pg_conn *gorm.DB, inf interface{}) bool {
	if err := pg_conn.Create(inf).Error; err != nil {
		c.AbortWithStatus(http.StatusInternalServerError)
		LogError("FoxKit", err)
		return false
	}
	return true
}

// Finds Data Entry with condition and updates it with new data, sets status to 500 if false
func UpdateDB(c *gin.Context, pg_conn *gorm.DB, condition interface{}, inf interface{}) bool {
	if err := pg_conn.Where(condition).Updates(inf).Error; err != nil {
		c.AbortWithStatus(http.StatusInternalServerError)
		LogError("FoxKit", err)
		return false
	}
	return true
}

// Finds and Returns true with the Data Entry or false if not found, sets status to 500 (error) or 404 (not found) if false
func GetDB(c *gin.Context, pg_conn *gorm.DB, condition interface{}, inf interface{}) bool {
	err := pg_conn.Where(condition).First(inf).Error
	if err == gorm.ErrRecordNotFound {
		c.AbortWithStatus(http.StatusNotFound)
		LogError("FoxKit", err)
		return false
	} else if err != nil {
		c.AbortWithStatus(http.StatusInternalServerError)
		LogError("FoxKit", err)
		return false
	}
	return true
}

// returns true if an entry for the given condition exists
func ExistsDB(c *gin.Context, pg_conn *gorm.DB, model interface{}, condition interface{}) (bool, error) {
	var count int64
	err := pg_conn.Model(model).Where(condition).Count(&count).Error
	if err != nil {
		// error
		return false, err
	} else if count == 0 {
		// doesn't exists
		return false, nil
	}
	// exists
	return true, nil
}

// returns the number of rows with the given condition
func CountDB(c *gin.Context, pg_conn *gorm.DB, model interface{}, condition interface{}) (int64, error) {
	var count int64
	err := pg_conn.Model(model).Where(condition).Count(&count).Error
	if err != nil {
		return 0, err
	}
	return count, nil
}

// Deletes Data Entry with condition, sets the status to 500 if false
func DeleteDB(c *gin.Context, pg_conn *gorm.DB, condition interface{}, inf interface{}) bool {
	if err := pg_conn.Where(condition).Delete(inf).Error; err != nil {
		c.AbortWithStatus(http.StatusInternalServerError)
		LogError("FoxKit", err)
		return false
	}
	return true
}