feat: add clean db script

This commit is contained in:
Sam 2023-03-13 23:26:12 +01:00
parent 72b54512aa
commit fb10f29e2b
No known key found for this signature in database
GPG Key ID: B4EF20DDE721CAA1
2 changed files with 114 additions and 0 deletions

View File

@ -55,6 +55,11 @@ const (
MaxLinkLength = 256 MaxLinkLength = 256
) )
const (
SelfDeleteAfter = 30 * 24 * time.Hour
ModDeleteAfter = 180 * 24 * time.Hour
)
// CreateUser creates a user with the given username. // CreateUser creates a user with the given username.
func (db *DB) CreateUser(ctx context.Context, tx pgx.Tx, username string) (u User, err error) { func (db *DB) CreateUser(ctx context.Context, tx pgx.Tx, username string) (u User, err error) {
// check if the username is valid // check if the username is valid

109
scripts/cleandb/main.go Normal file
View File

@ -0,0 +1,109 @@
package main
import (
"context"
"fmt"
"os"
"time"
dbpkg "codeberg.org/u1f320/pronouns.cc/backend/db"
"github.com/georgysavva/scany/pgxscan"
"github.com/joho/godotenv"
"github.com/rs/xid"
)
func main() {
err := godotenv.Load()
if err != nil {
fmt.Println("error loading .env file:", err)
os.Exit(1)
}
ctx := context.Background()
db, err := dbpkg.New()
if err != nil {
fmt.Println("error opening database:", err)
os.Exit(1)
}
defer db.Close()
fmt.Println("opened database")
fmt.Println("deleting invalidated tokens")
ct, err := db.Exec(ctx, "DELETE FROM tokens WHERE invalidated = true OR expires < $1", time.Now())
if err != nil {
fmt.Println("executing query:", err)
os.Exit(1)
}
fmt.Printf("deleted %v invalidated or expired tokens\n", ct.RowsAffected())
var users []dbpkg.User
err = pgxscan.Select(ctx, db, &users, `SELECT * FROM users WHERE
deleted_at IS NOT NULL AND
(self_delete = true AND deleted_at < $1)
OR (self_delete = false AND deleted_at < $2)
ORDER BY id`, time.Now().Add(-dbpkg.SelfDeleteAfter), time.Now().Add(-dbpkg.ModDeleteAfter))
if err != nil {
fmt.Println("error getting to-be-deleted users:", err)
os.Exit(1)
}
if len(users) == 0 {
fmt.Println("there are no users pending deletion")
os.Exit(0)
}
for _, u := range users {
members, err := db.UserMembers(ctx, u.ID)
if err != nil {
fmt.Printf("error getting members for user %v: %v\n", u.ID, err)
continue
}
for _, m := range members {
if m.Avatar == nil {
continue
}
fmt.Printf("deleting avatars for member %v\n", m.ID)
err = db.DeleteMemberAvatar(ctx, m.ID, *m.Avatar)
if err != nil {
fmt.Printf("error deleting avatars for member %v: %v", m.ID, err)
continue
}
fmt.Printf("deleted avatars for member %v\n", m.ID)
}
if u.Avatar == nil {
continue
}
fmt.Printf("deleting avatars for user %v\n", u.ID)
err = db.DeleteUserAvatar(ctx, u.ID, *u.Avatar)
if err != nil {
fmt.Printf("error deleting avatars for user %v: %v", u.ID, err)
continue
}
fmt.Printf("deleted avatars for user %v\n", u.ID)
}
ids := make([]xid.ID, len(users))
for _, u := range users {
ids = append(ids, u.ID)
}
ct, err = db.Exec(ctx, "DELETE FROM users WHERE id = ANY($1)", ids)
if err != nil {
fmt.Printf("error deleting users: %v\n", err)
os.Exit(1)
}
fmt.Printf("deleted %v users!\n", ct.RowsAffected())
}