extern/hishtory
1
0
Fork 0
mirror of https://github.com/ddworken/hishtory.git synced 2025-06-24 14:01:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update custom column support to also automatically retry DB errors to further harden against issues like #119

Browse Source
This commit is contained in:
David Dworken 2023-10-30 17:50:47 -07:00
parent 5f9ff9e158
commit f8b515c328
No known key found for this signature in database
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
client/lib/lib.go
View File

@ -4,6 +4,7 @@ import (
"bufio"
"bytes"
"context"
"database/sql"
"encoding/json"
"errors"
"fmt"
@ -937,15 +938,17 @@ func parseAtomizedToken(ctx context.Context, token string) (string, any, any, er
func getAllCustomColumnNames(ctx context.Context) ([]string, error) {
db := hctx.GetDb(ctx)
query := `
SELECT DISTINCT json_extract(value, '$.name') as cc_name
FROM history_entries
JOIN json_each(custom_columns)
WHERE value IS NOT NULL
LIMIT 10`
rows, err := db.Raw(query).Rows()
rows, err := RetryingDbFunctionWithResult(func() (*sql.Rows, error) {
query := `
SELECT DISTINCT json_extract(value, '$.name') as cc_name
FROM history_entries
JOIN json_each(custom_columns)
WHERE value IS NOT NULL
LIMIT 10`
return db.Raw(query).Rows()
})
if err != nil {
return nil, err
return nil, fmt.Errorf("failed to query for list of custom columns: %v", err)
}
ccNames := make([]string, 0)
for rows.Next() {