59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
package stats
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
type Cache struct {
|
|
timestamp time.Time
|
|
data string
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
// TODO: This should be read from the environment
|
|
fileInfo, err := os.Stat("feedvault.db")
|
|
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
|
|
}
|