69 lines
2.7 KiB
Go
69 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"github.com/jackc/pgx/pgxpool"
|
|
)
|
|
|
|
func postHandler(reportPostChan chan ReportPost) {
|
|
for { // Write posts
|
|
v := <-reportPostChan
|
|
go writePost(v)
|
|
}
|
|
}
|
|
|
|
func writePost(reportpost ReportPost) {
|
|
conn, err := pool.Acquire(context.Background())
|
|
if err != nil {
|
|
log.Fatal("Error connecting to database:", err)
|
|
}
|
|
defer conn.Release()
|
|
|
|
// Insert new account if new
|
|
var accountid int
|
|
err = conn.QueryRow(context.Background(), "INSERT INTO accounts (acct, avatar, bot, created_at, display_name, uri) VALUES($1, $2, $3, $4, $5, $6) ON CONFLICT(acct) DO UPDATE SET acct=EXCLUDED.acct RETURNING id", reportpost.Account.Acct, reportpost.Account.Avatar, reportpost.Account.Bot, reportpost.Account.Created_at, reportpost.Account.Display_name, reportpost.Account.Url).Scan(&accountid)
|
|
if err != nil {
|
|
log.Print("First ", err)
|
|
log.Print("--------------------------")
|
|
log.Print("Reported error: ", err)
|
|
log.Print("Account Acct: ", reportpost.Account.Acct)
|
|
log.Print("Account Avatar: ", reportpost.Account.Avatar)
|
|
log.Print("Account Avatar len: ", len(reportpost.Account.Avatar))
|
|
log.Print("Account Bot: ", reportpost.Account.Bot)
|
|
log.Print("Account Created_at: ", reportpost.Account.Created_at)
|
|
log.Print("Account Display: ", reportpost.Account.Display_name)
|
|
log.Print("Account URL: ", reportpost.Account.Url)
|
|
log.Print(reportpost)
|
|
log.Print("--------------------------")
|
|
log.Fatal("Unable to write record to database")
|
|
}
|
|
|
|
// Insert new post if new
|
|
_, err = conn.Exec(context.Background(), "INSERT INTO posts (uri, content, created_at, normalized, account_id, posthash) VALUES ($1, $2, $3, $4, $5, $6) ON CONFLICT (posthash) DO NOTHING", reportpost.Uri, reportpost.Content, reportpost.Created_at, reportpost.normalized, accountid, reportpost.posthash)
|
|
if err != nil { // For now I want to know why this failed.
|
|
log.Print("Second ", err)
|
|
log.Print("--------------------------")
|
|
log.Print("Reported error: ", err)
|
|
log.Print("Uri: ", reportpost.Uri)
|
|
log.Print("Content: ", reportpost.Content)
|
|
log.Print("Created_at: ", reportpost.Created_at)
|
|
log.Print("normalized: ", reportpost.normalized)
|
|
log.Print("account_id", accountid)
|
|
log.Print("posthash: ", reportpost.posthash)
|
|
log.Print("--------------------------")
|
|
log.Fatal("Unable to write record to database")
|
|
}
|
|
}
|
|
|
|
func getDbPool() *pgxpool.Pool {
|
|
// Setup Database
|
|
dbURI := fmt.Sprintf("postgres://%s:%s@%s:%d/fedilogue", settings.Database.Username, settings.Database.Password, settings.Database.Host, settings.Database.Port)
|
|
pool, err := pgxpool.Connect(context.Background(), dbURI)
|
|
if err != nil {
|
|
log.Fatal("Unable to connect to database:", err)
|
|
}
|
|
return pool
|
|
}
|