extern/hishtory
1
0
Fork 0
mirror of https://github.com/ddworken/hishtory.git synced 2025-06-20 11:57:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update CreateIndices() function so that it works with sqlite

Browse Source
This commit is contained in:
David Dworken 2023-09-24 18:49:40 -07:00
parent 64db341144
commit 3d0673e6e9
No known key found for this signature in database
1 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
backend/server/internal/database/db.go
View File

@ -4,6 +4,7 @@ import (
"context" "context"
"database/sql" "database/sql"
"fmt" "fmt"
"strings"
"github.com/ddworken/hishtory/shared" "github.com/ddworken/hishtory/shared"
"github.com/jackc/pgx/v4/stdlib" "github.com/jackc/pgx/v4/stdlib"
@ -65,15 +66,25 @@ func (db *DB) CreateIndices() error {
// Note: If adding a new index here, consider manually running it on the prod DB using CONCURRENTLY to // Note: If adding a new index here, consider manually running it on the prod DB using CONCURRENTLY to
// make server startup non-blocking. The benefit of this function is primarily for other people so they // make server startup non-blocking. The benefit of this function is primarily for other people so they
// don't have to manually create these indexes. // don't have to manually create these indexes.
var indices = []string{ indices := []struct {
`CREATE INDEX IF NOT EXISTS entry_id_idx ON enc_history_entries USING btree(encrypted_id);`, name string
`CREATE INDEX IF NOT EXISTS device_id_idx ON enc_history_entries USING btree(device_id);`, table string
`CREATE INDEX IF NOT EXISTS read_count_idx ON enc_history_entries USING btree(read_count);`, columns []string
`CREATE INDEX IF NOT EXISTS redact_idx ON enc_history_entries USING btree(user_id, device_id, date);`, }{
`CREATE INDEX IF NOT EXISTS del_user_idx ON deletion_requests USING btree(user_id);`, {"entry_id_idx", "enc_history_entries", []string{"encrypted_id"}},
{"device_id_idx", "enc_history_entries", []string{"device_id"}},
{"read_count_idx", "enc_history_entries", []string{"read_count"}},
{"redact_idx", "enc_history_entries", []string{"user_id", "device_id", "date"}},
{"del_user_idx", "deletion_requests", []string{"user_id"}},
} }
for _, index := range indices { for _, index := range indices {
r := db.Exec(index) sql := ""
if db.Name() == "sqlite" {
sql = fmt.Sprintf("CREATE INDEX IF NOT EXISTS %s ON %s (%s)", index.name, index.table, strings.Join(index.columns, ","))
} else {
sql = fmt.Sprintf("CREATE INDEX IF NOT EXISTS %s ON %s USING btree(%s)", index.name, index.table, strings.Join(index.columns, ","))
}
r := db.Exec(sql)
if r.Error != nil { if r.Error != nil {
return fmt.Errorf("failed to execute index creation sql=%#v: %w", index, r.Error) return fmt.Errorf("failed to execute index creation sql=%#v: %w", index, r.Error)
} }