Get DB size from Postgres

db/stats.sql.go

@@ -0,0 +1,21 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.25.0
+// source: stats.sql
+
+package db
+
+import (
+	"context"
+)
+
+const dBSize = `-- name: DBSize :one
+SELECT pg_size_pretty(pg_database_size(current_database()))
+`
+
+func (q *Queries) DBSize(ctx context.Context) (string, error) {
+	row := q.db.QueryRow(ctx, dBSize)
+	var pg_size_pretty string
+	err := row.Scan(&pg_size_pretty)
+	return pg_size_pretty, err
+}

handlers.go

@@ -31,7 +31,7 @@ func FeedsHandler(w http.ResponseWriter, _ *http.Request) {
 		Keywords:     "RSS, Atom, Feed, Archive",
 		Author:       "TheLovinator",
 		CanonicalURL: "http://localhost:8000/feeds",
-		Content:      "<p>Here be </p>",
+		Content:      "<p>No feeds yet.</p>",
 	}
 	html := FullHTML(htmlData)
 	_, err := w.Write([]byte(html))
@@ -163,7 +163,10 @@ func IndexHandler(w http.ResponseWriter, _ *http.Request) {
 		Content:      content,
 	}
 	html := FullHTML(htmlData)
-	w.Write([]byte(html))
+	_, err := w.Write([]byte(html))
+	if err != nil {
+		log.Println("Error writing response:", err)
+	}
 }
 
 func UploadOpmlHandler(w http.ResponseWriter, r *http.Request) {
@@ -218,5 +221,8 @@ func UploadOpmlHandler(w http.ResponseWriter, r *http.Request) {
 		ParseResult:  parseResult,
 	}
 	html := FullHTML(htmlData)
-	w.Write([]byte(html))
+	_, err = w.Write([]byte(html))
+	if err != nil {
+		log.Println("Error writing response:", err)
+	}
 }

sql/queries/stats.sql

@@ -0,0 +1,2 @@
+-- name: DBSize :one
+SELECT pg_size_pretty(pg_database_size(current_database()));

sqlc.yaml

@@ -8,7 +8,5 @@ sql:
               out: "db"
               sql_package: "pgx/v5"
               emit_prepared_queries: true
-              emit_interface: true
               emit_empty_slices: true
-              emit_exported_queries: true
               emit_json_tags: true

stats.go

@@ -1,8 +1,7 @@
 package main
 
 import (
-	"fmt"
-	"os"
+	"context"
 	"time"
 )
 
@@ -14,46 +13,22 @@ type Cache struct {
 var cache Cache
 
 func GetDBSize() (string, error) {
-	// If cache is less than 10 minutes old, return cached data
-	if time.Since(cache.timestamp).Minutes() < 10 {
-		return cache.data, nil
+	// If the cache is older than 5 minutes, get the database size from the database
+	if time.Since(cache.timestamp) > 5*time.Minute {
+		dbSize, err := getDBSizeFromDB()
+		if err != nil {
+			return "", err
+		}
+		cache = Cache{timestamp: time.Now(), data: dbSize}
 	}
+	return cache.data, nil
+}
 
-	// TODO: This should be read from the environment
-	fileInfo, err := os.Stat("feedvault.db")
+func getDBSizeFromDB() (string, error) {
+	// Get database size from the PostgreSQL database
+	dbSize, err := DB.DBSize(context.Background())
 	if err != nil {
 		return "", err
 	}
-
-	// Get the file size in bytes
-	fileSize := fileInfo.Size()
-
-	// Convert to human readable size and append the unit (KiB, MiB, GiB, TiB)
-	var size float64
-
-	if fileSize < 1024*1024 {
-		// If the file size is less than 1 KiB, return the size in bytes
-		size = float64(fileSize) / 1024
-		cache.data = fmt.Sprintf("%.2f KiB", size)
-
-	} else if fileSize < 1024*1024*1024 {
-		// If the file size is less than 1 MiB, return the size in KiB
-		size = float64(fileSize) / (1024 * 1024)
-		cache.data = fmt.Sprintf("%.2f MiB", size)
-
-	} else if fileSize < 1024*1024*1024*1024 {
-		// If the file size is less than 1 GiB, return the size in MiB
-		size = float64(fileSize) / (1024 * 1024 * 1024)
-		cache.data = fmt.Sprintf("%.2f GiB", size)
-	} else {
-		// If the file size is 1 GiB or more, return the size in TiB
-		size = float64(fileSize) / (1024 * 1024 * 1024 * 1024)
-		cache.data = fmt.Sprintf("%.2f TiB", size)
-	}
-
-	// Update cache timestamp
-	cache.timestamp = time.Now()
-
-	// Return the human readable size
-	return cache.data, nil
+	return dbSize, nil
 }