Compare commits

...

6 Commits

4 changed files with 584 additions and 10 deletions

2
.gitignore vendored
View File

@ -1 +1 @@
postgres/
postgres**/

View File

@ -5,14 +5,22 @@ import (
"fmt"
"net/http"
"os"
"strconv"
"time"
"git.highperfocused.tech/highperfocused/lumina-relay/relay/cache"
"git.highperfocused.tech/highperfocused/lumina-relay/relay/trending"
"github.com/fiatjaf/eventstore/postgresql"
"github.com/fiatjaf/khatru"
"github.com/fiatjaf/khatru/policies"
)
// Cache for storing generic data like event counts
var dataCache = cache.New()
const eventCountCacheKey = "total_event_count"
const eventCountCacheDuration = 1 * time.Minute
func getEnv(key, fallback string) string {
if value, ok := os.LookupEnv(key); ok {
return value
@ -20,6 +28,43 @@ func getEnv(key, fallback string) string {
return fallback
}
// Gets total event count, using cache if available
func getTotalEventCount(db *postgresql.PostgresBackend) (int, error) {
// Try getting from cache first
if cachedCount, ok := dataCache.Get(eventCountCacheKey); ok {
return cachedCount.(int), nil
}
// If not in cache, query the database
count := 0
row := db.DB.QueryRow("SELECT COUNT(*) FROM event")
if err := row.Scan(&count); err != nil {
return 0, err
}
// Update the cache
dataCache.Set(eventCountCacheKey, count, eventCountCacheDuration)
return count, nil
}
// Updates event count in the background periodically
func startEventCountUpdater(db *postgresql.PostgresBackend) {
go func() {
ticker := time.NewTicker(eventCountCacheDuration)
defer ticker.Stop()
for range ticker.C {
count := 0
row := db.DB.QueryRow("SELECT COUNT(*) FROM event")
if err := row.Scan(&count); err != nil {
fmt.Printf("Error updating event count: %v\n", err)
continue
}
dataCache.Set(eventCountCacheKey, count, eventCountCacheDuration)
fmt.Printf("Updated event count cache: %d events\n", count)
}
}()
}
func main() {
fmt.Print(`
LUMINA RELAY
@ -48,6 +93,20 @@ func main() {
panic(err)
}
// Initialize trending system to start background calculations
fmt.Println("Initializing trending system...")
if err := trending.Initialize(db.DB.DB); err != nil {
fmt.Printf("Warning: Error initializing trending system: %v\n", err)
}
// Initialize event count cache and start periodic updates
fmt.Println("Initializing event count cache...")
_, err := getTotalEventCount(&db)
if err != nil {
fmt.Printf("Warning: Error initializing event count cache: %v\n", err)
}
startEventCountUpdater(&db)
relay.StoreEvent = append(relay.StoreEvent, db.SaveEvent)
relay.QueryEvents = append(relay.QueryEvents, db.QueryEvents)
relay.DeleteEvent = append(relay.DeleteEvent, db.DeleteEvent)
@ -66,11 +125,11 @@ func main() {
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("content-type", "text/html")
// Query the total number of events
count := 0
row := db.DB.QueryRow("SELECT COUNT(*) FROM event")
if err := row.Scan(&count); err != nil {
fmt.Printf("Error counting events: %v\n", err)
// Get event count from cache
count, err := getTotalEventCount(&db)
if err != nil {
fmt.Printf("Error getting event count: %v\n", err)
count = 0 // Fall back to zero if there's an error
}
// Improved HTML content with link to stats page
@ -120,6 +179,7 @@ func main() {
<h1>Welcome to LUMINA Relay!</h1>
<p>Number of events stored: %d</p>
<p><a href="/stats">View Event Stats</a></p>
<p><a href="/trending/history">View Trending History</a></p>
</div>
</body>
</html>
@ -218,6 +278,13 @@ func main() {
mux.HandleFunc("/api/stats", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
// Get total count from cache
totalCount, err := getTotalEventCount(&db)
if err != nil {
http.Error(w, fmt.Sprintf("Error getting event count: %v", err), http.StatusInternalServerError)
return
}
// Query the number of events for each kind, sorted by kind
rows, err := db.DB.Query("SELECT kind, COUNT(*) FROM event GROUP BY kind ORDER BY kind")
if err != nil {
@ -227,7 +294,6 @@ func main() {
defer rows.Close()
stats := make(map[string]int)
totalCount := 0
for rows.Next() {
var kind string
var count int
@ -236,7 +302,6 @@ func main() {
return
}
stats[kind] = count
totalCount += count
}
// Add total count to the stats
@ -269,6 +334,276 @@ func main() {
}
})
// Add endpoint for trending history
mux.HandleFunc("/api/trending/history/kind20", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
// Parse query parameters for pagination
limitStr := r.URL.Query().Get("limit")
offsetStr := r.URL.Query().Get("offset")
limit := 10 // Default limit
offset := 0 // Default offset
if limitStr != "" {
if val, err := strconv.Atoi(limitStr); err == nil && val > 0 {
limit = val
}
}
if offsetStr != "" {
if val, err := strconv.Atoi(offsetStr); err == nil && val >= 0 {
offset = val
}
}
// Get trending history for kind 20
history, err := trending.GetTrendingHistoryForKind(db.DB.DB, 20, limit, offset)
if err != nil {
http.Error(w, fmt.Sprintf("Error getting trending history: %v", err), http.StatusInternalServerError)
return
}
// Get total count for pagination info
totalCount, err := trending.GetTrendingHistoryCount(db.DB.DB, 20)
if err != nil {
http.Error(w, fmt.Sprintf("Error getting trending history count: %v", err), http.StatusInternalServerError)
return
}
response := map[string]interface{}{
"history": history,
"pagination": map[string]interface{}{
"total": totalCount,
"limit": limit,
"offset": offset,
},
}
if err := json.NewEncoder(w).Encode(response); err != nil {
http.Error(w, fmt.Sprintf("Error encoding JSON: %v", err), http.StatusInternalServerError)
}
})
// Add UI for trending history
mux.HandleFunc("/trending/history", func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("content-type", "text/html")
fmt.Fprintf(w, `
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>LUMINA Relay - Trending History</title>
<style>
body {
font-family: Arial, sans-serif;
background-color: #f4f4f4;
margin: 0;
padding: 20px;
}
.container {
max-width: 1200px;
margin: 0 auto;
background-color: #fff;
padding: 20px;
border-radius: 8px;
box-shadow: 0 0 10px rgba(0, 0, 0, 0.1);
}
h1, h2 {
color: #333;
}
.history-item {
margin-bottom: 30px;
padding: 15px;
border: 1px solid #ddd;
border-radius: 5px;
}
.history-date {
font-weight: bold;
margin-bottom: 10px;
color: #555;
}
.posts-container {
display: grid;
grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
gap: 15px;
}
.post {
border: 1px solid #eee;
padding: 10px;
border-radius: 5px;
}
.post-content {
max-height: 100px;
overflow: hidden;
margin-bottom: 10px;
}
.post-id {
max-height: 100px;
overflow: hidden;
margin-bottom: 10px;
}
.post-reactions {
font-weight: bold;
color: #007bff;
}
.pagination {
margin-top: 20px;
display: flex;
justify-content: center;
}
.pagination a {
margin: 0 5px;
padding: 8px 12px;
border: 1px solid #ddd;
color: #007bff;
text-decoration: none;
border-radius: 3px;
}
.pagination a:hover {
background-color: #f8f8f8;
}
.loading {
text-align: center;
padding: 20px;
font-style: italic;
color: #777;
}
</style>
</head>
<body>
<div class="container">
<h1>Trending History</h1>
<p>Archive of trending posts calculations</p>
<div id="history-container">
<div class="loading">Loading trending history data...</div>
</div>
<div id="pagination" class="pagination"></div>
</div>
<script>
let currentOffset = 0;
const limit = 5;
let totalItems = 0;
// Format date for display
function formatDate(dateStr) {
const date = new Date(dateStr);
return date.toLocaleString();
}
// Truncate content for preview
function truncateContent(content, maxLength = 100) {
if (content.length <= maxLength) return content;
return content.substr(0, maxLength) + '...';
}
// Load trending history data
async function loadTrendingHistory(offset = 0) {
try {
const response = await fetch('/api/trending/history/kind20?limit=' + limit + '&offset=' + offset);
const data = await response.json();
if (!response.ok) throw new Error(data.message || 'Error loading trending history');
totalItems = data.pagination.total;
currentOffset = offset;
renderTrendingHistory(data.history);
renderPagination();
} catch (error) {
document.getElementById('history-container').innerHTML =
'<div class="error">Error loading trending history: ' + error.message + '</div>';
}
}
// Render trending history
function renderTrendingHistory(historyItems) {
const container = document.getElementById('history-container');
if (historyItems.length === 0) {
container.innerHTML = '<p>No trending history data available yet.</p>';
return;
}
let html = '';
historyItems.forEach(item => {
html += '<div class="history-item">' +
'<div class="history-date">' +
'Calculated on: ' + formatDate(item.calculation_time) +
'</div>' +
'<div class="posts-container">';
item.posts.forEach(post => {
html += '<div class="post">' +
'<div class="post-id">' + truncateContent(post.id) + '</div><hr />' +
'<div class="post-content">' + truncateContent(post.content) + '</div>' +
'<div class="post-reactions">Reactions: ' + post.reaction_count + '</div>' +
'</div>';
});
html += '</div></div>';
});
container.innerHTML = html;
}
// Render pagination controls
function renderPagination() {
const container = document.getElementById('pagination');
const totalPages = Math.ceil(totalItems / limit);
const currentPage = Math.floor(currentOffset / limit) + 1;
let html = '';
if (totalPages <= 1) {
container.innerHTML = '';
return;
}
// Previous button
if (currentPage > 1) {
html += '<a href="#" onclick="loadTrendingHistory(' + ((currentPage - 2) * limit) + '); return false;">Previous</a>';
}
// Page numbers
const maxPagesToShow = 5;
let startPage = Math.max(1, currentPage - Math.floor(maxPagesToShow / 2));
let endPage = Math.min(totalPages, startPage + maxPagesToShow - 1);
if (endPage - startPage + 1 < maxPagesToShow) {
startPage = Math.max(1, endPage - maxPagesToShow + 1);
}
for (let i = startPage; i <= endPage; i++) {
const offset = (i - 1) * limit;
const active = i === currentPage ? 'active' : '';
html += '<a href="#" class="' + active + '" onclick="loadTrendingHistory(' + offset + '); return false;">' + i + '</a>';
}
// Next button
if (currentPage < totalPages) {
html += '<a href="#" onclick="loadTrendingHistory(' + (currentPage * limit) + '); return false;">Next</a>';
}
container.innerHTML = html;
}
// Initial load
document.addEventListener('DOMContentLoaded', () => {
loadTrendingHistory();
});
</script>
</body>
</html>
`)
})
fmt.Println("running on :3334")
http.ListenAndServe(":3334", relay)
}

View File

@ -3,6 +3,7 @@ package trending
import (
"database/sql"
"encoding/json"
"fmt"
"time"
"git.highperfocused.tech/highperfocused/lumina-relay/relay/cache"
@ -20,15 +21,70 @@ type Post struct {
var (
trendingCache = cache.New()
cacheDuration = 5 * time.Minute
cacheDuration = 35 * time.Minute // Slightly longer than update interval
updateInterval = 30 * time.Minute
)
// Initialize sets up the trending system
// - Creates database tables
// - Performs initial calculation
// - Sets up periodic updates
func Initialize(db *sql.DB) error {
// Create necessary database tables
if err := CreateTablesIfNotExist(db); err != nil {
return fmt.Errorf("failed to create trending database tables: %w", err)
}
// Perform initial calculation
if _, err := calculateTrendingKind20(db); err != nil {
fmt.Printf("Error in initial trending calculation: %v\n", err)
}
// Set up periodic updates
go func() {
ticker := time.NewTicker(updateInterval)
defer ticker.Stop()
for range ticker.C {
if _, err := calculateTrendingKind20(db); err != nil {
fmt.Printf("Error updating trending data: %v\n", err)
} else {
fmt.Printf("Successfully updated trending data at %s\n", time.Now().Format(time.RFC3339))
}
}
}()
return nil
}
// GetTrendingKind20 returns the top 20 trending posts of kind 20 from the last 24 hours
// It first tries the in-memory cache, then the database, and calculates on-demand if needed
func GetTrendingKind20(db *sql.DB) ([]Post, error) {
// Try in-memory cache first
if cached, ok := trendingCache.Get("trending_kind_20"); ok {
return cached.([]Post), nil
}
// If not in memory, try getting from database
posts, calculationTime, err := GetLatestTrendingFromHistory(db, 20)
if err != nil {
return nil, err
}
// If we got data from the database and it's not too old, use it
if len(posts) > 0 && time.Since(calculationTime) < cacheDuration {
// Update in-memory cache
trendingCache.Set("trending_kind_20", posts, cacheDuration)
return posts, nil
}
// Calculate on-demand as a last resort
return calculateTrendingKind20(db)
}
// calculateTrendingKind20 performs the actual calculation of trending posts,
// updates the cache, and saves to database
func calculateTrendingKind20(db *sql.DB) ([]Post, error) {
query := `
WITH reactions AS (
SELECT
@ -76,6 +132,49 @@ func GetTrendingKind20(db *sql.DB) ([]Post, error) {
trendingPosts = append(trendingPosts, post)
}
// Update in-memory cache
trendingCache.Set("trending_kind_20", trendingPosts, cacheDuration)
// Save to database for historical records
postsJSON, err := json.Marshal(trendingPosts)
if err != nil {
return nil, fmt.Errorf("failed to marshal trending posts to JSON: %w", err)
}
_, err = db.Exec(`
INSERT INTO trending_history (calculation_time, kind, trending_data)
VALUES (NOW(), 20, $1)
`, postsJSON)
if err != nil {
fmt.Printf("Warning: failed to store trending data in database: %v\n", err)
// Don't return error here as we want to still return the trending posts even if saving fails
}
return trendingPosts, nil
}
// UnmarshalPosts unmarshals JSON data into a slice of Post objects
func UnmarshalPosts(data []byte) ([]Post, error) {
var posts []Post
if err := json.Unmarshal(data, &posts); err != nil {
return nil, err
}
return posts, nil
}
// GetTrendingHistoryCount returns the count of trending history entries for the given kind
func GetTrendingHistoryCount(db *sql.DB, kind int) (int, error) {
var count int
err := db.QueryRow(`
SELECT COUNT(*)
FROM trending_history
WHERE kind = $1
`, kind).Scan(&count)
if err != nil {
return 0, fmt.Errorf("failed to count trending history: %w", err)
}
return count, nil
}

140
relay/trending/schema.go Normal file
View File

@ -0,0 +1,140 @@
package trending
import (
"database/sql"
"fmt"
"time"
)
// Schema version to track database migrations
const SchemaVersion = 1
// CreateTablesIfNotExist ensures that all necessary database tables for the trending system exist
func CreateTablesIfNotExist(db *sql.DB) error {
// Create the trending_history table if it doesn't exist
_, err := db.Exec(`
CREATE TABLE IF NOT EXISTS trending_history (
id SERIAL PRIMARY KEY,
calculation_time TIMESTAMPTZ NOT NULL,
kind INTEGER NOT NULL,
trending_data JSONB NOT NULL
)
`)
if err != nil {
return fmt.Errorf("failed to create trending_history table: %w", err)
}
// Create an index on calculation_time and kind for faster queries
_, err = db.Exec(`
CREATE INDEX IF NOT EXISTS idx_trending_history_time_kind
ON trending_history (calculation_time DESC, kind)
`)
if err != nil {
return fmt.Errorf("failed to create index on trending_history: %w", err)
}
// Create schema version table if it doesn't exist
_, err = db.Exec(`
CREATE TABLE IF NOT EXISTS trending_schema_version (
version INTEGER PRIMARY KEY,
updated_at TIMESTAMPTZ NOT NULL
)
`)
if err != nil {
return fmt.Errorf("failed to create trending_schema_version table: %w", err)
}
// Check if we need to initialize the schema version
var count int
err = db.QueryRow(`SELECT COUNT(*) FROM trending_schema_version`).Scan(&count)
if err != nil {
return fmt.Errorf("failed to check trending_schema_version: %w", err)
}
if count == 0 {
_, err = db.Exec(`
INSERT INTO trending_schema_version (version, updated_at)
VALUES ($1, $2)
`, SchemaVersion, time.Now())
if err != nil {
return fmt.Errorf("failed to initialize trending_schema_version: %w", err)
}
}
return nil
}
// GetLatestTrendingFromHistory retrieves the most recent trending data for the specified kind
func GetLatestTrendingFromHistory(db *sql.DB, kind int) ([]Post, time.Time, error) {
var (
trendingData []byte
calculationTime time.Time
)
err := db.QueryRow(`
SELECT trending_data, calculation_time
FROM trending_history
WHERE kind = $1
ORDER BY calculation_time DESC
LIMIT 1
`, kind).Scan(&trendingData, &calculationTime)
if err != nil {
if err == sql.ErrNoRows {
return []Post{}, time.Time{}, nil
}
return nil, time.Time{}, fmt.Errorf("failed to get latest trending data: %w", err)
}
posts, err := UnmarshalPosts(trendingData)
if err != nil {
return nil, time.Time{}, fmt.Errorf("failed to unmarshal trending posts: %w", err)
}
return posts, calculationTime, nil
}
// GetTrendingHistoryForKind retrieves trending history for the specified kind
// limit defines how many records to return, offset is for pagination
func GetTrendingHistoryForKind(db *sql.DB, kind int, limit, offset int) ([]TrendingHistoryEntry, error) {
rows, err := db.Query(`
SELECT id, calculation_time, trending_data
FROM trending_history
WHERE kind = $1
ORDER BY calculation_time DESC
LIMIT $2 OFFSET $3
`, kind, limit, offset)
if err != nil {
return nil, fmt.Errorf("failed to query trending history: %w", err)
}
defer rows.Close()
var entries []TrendingHistoryEntry
for rows.Next() {
var (
entry TrendingHistoryEntry
trendingData []byte
)
err := rows.Scan(&entry.ID, &entry.CalculationTime, &trendingData)
if err != nil {
return nil, fmt.Errorf("failed to scan trending history entry: %w", err)
}
entry.Posts, err = UnmarshalPosts(trendingData)
if err != nil {
return nil, fmt.Errorf("failed to unmarshal trending posts: %w", err)
}
entries = append(entries, entry)
}
return entries, nil
}
// TrendingHistoryEntry represents a historical record of trending data
type TrendingHistoryEntry struct {
ID int `json:"id"`
CalculationTime time.Time `json:"calculation_time"`
Posts []Post `json:"posts"`
}