package primary import ( "blackforestbytes.com/simplecloudnotifier/models" "database/sql" "errors" "gogs.mikescher.com/BlackForestBytes/goext/sq" ) func (db *Database) CreateCompatID(ctx TxContext, idtype string, newid string) (int64, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return 0, err } rows, err := tx.Query(ctx, "SELECT COALESCE(MAX(old), 0) FROM compat_ids", sq.PP{}) if err != nil { return 0, err } if !rows.Next() { return 0, errors.New("failed to query MAX(old)") } var oldid int64 err = rows.Scan(&oldid) if err != nil { return 0, err } oldid++ _, err = tx.Exec(ctx, "INSERT INTO compat_ids (old, new, type) VALUES (:old, :new, :typ)", sq.PP{ "old": oldid, "new": newid, "typ": idtype, }) if err != nil { return 0, err } return oldid, nil } func (db *Database) ConvertCompatID(ctx TxContext, oldid int64, idtype string) (*string, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return nil, err } rows, err := tx.Query(ctx, "SELECT new FROM compat_ids WHERE old = :old AND type = :typ", sq.PP{ "old": oldid, "typ": idtype, }) if err != nil { return nil, err } if !rows.Next() { return nil, nil } var newid string err = rows.Scan(&newid) if err == sql.ErrNoRows { return nil, nil } if err != nil { return nil, err } return &newid, nil } func (db *Database) ConvertToCompatID(ctx TxContext, newid string) (*int64, *string, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return nil, nil, err } rows, err := tx.Query(ctx, "SELECT old, type FROM compat_ids WHERE new = :new", sq.PP{"new": newid}) if err != nil { return nil, nil, err } if !rows.Next() { return nil, nil, nil } var oldid int64 var idtype string err = rows.Scan(&oldid, &idtype) if err == sql.ErrNoRows { return nil, nil, nil } if err != nil { return nil, nil, err } return &oldid, &idtype, nil } func (db *Database) ConvertToCompatIDOrCreate(ctx TxContext, idtype string, newid string) (int64, error) { id1, _, err := db.ConvertToCompatID(ctx, newid) if err != nil { return 0, err } if id1 != nil { return *id1, nil } id2, err := db.CreateCompatID(ctx, idtype, newid) if err != nil { return 0, err } return id2, nil } func (db *Database) GetAck(ctx TxContext, msgid models.MessageID) (bool, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return false, err } rows, err := tx.Query(ctx, "SELECT * FROM compat_acks WHERE message_id = :msgid LIMIT 1", sq.PP{ "msgid": msgid, }) if err != nil { return false, err } res := rows.Next() err = rows.Close() if err != nil { return false, err } return res, nil } func (db *Database) SetAck(ctx TxContext, userid models.UserID, msgid models.MessageID) error { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return err } _, err = tx.Exec(ctx, "INSERT INTO compat_acks (user_id, message_id) VALUES (:uid, :mid)", sq.PP{ "uid": userid, "mid": msgid, }) if err != nil { return err } return nil }