From 1e73f185d899753b32cfecf8e1f1f7afa4d71978 Mon Sep 17 00:00:00 2001 From: Michael Quigley Date: Fri, 1 Mar 2024 16:27:32 -0500 Subject: [PATCH] access_grant store implementation (#432) --- controller/store/accessGrant.go | 52 +++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100644 controller/store/accessGrant.go diff --git a/controller/store/accessGrant.go b/controller/store/accessGrant.go new file mode 100644 index 00000000..022c93e3 --- /dev/null +++ b/controller/store/accessGrant.go @@ -0,0 +1,52 @@ +package store + +import ( + "github.com/jmoiron/sqlx" + "github.com/pkg/errors" +) + +type AccessGrant struct { + Model + ShareId int + AccountId int +} + +func (str *Store) CreateAccessGrant(shareId, accountId, tx *sqlx.Tx) (int, error) { + stmt, err := tx.Prepare("insert into access_grants (share_id, account_id) values ($1, $2) returning id") + if err != nil { + return 0, errors.Wrap(err, "error preparing access_grant insert statement") + } + var id int + if err := stmt.QueryRow(shareId, accountId).Scan(&id); err != nil { + return 0, errors.Wrap(err, "error executing access_grant insert statement") + } + return id, nil +} + +func (str *Store) FindAccessGrantsForAccount(accountId int, tx *sqlx.Tx) ([]*AccessGrant, error) { + rows, err := tx.Queryx("select access_grants.* from access_grants where account_id = $1 and not deleted", accountId) + if err != nil { + return nil, errors.Wrap(err, "error selecting access_grants by account_id") + } + var ags []*AccessGrant + for rows.Next() { + ag := &AccessGrant{} + if err := rows.StructScan(ag); err != nil { + return nil, errors.Wrap(err, "error scanning access_grant") + } + ags = append(ags, ag) + } + return ags, nil +} + +func (str *Store) DeleteAccessGrant(id int, tx *sqlx.Tx) error { + stmt, err := tx.Prepare("update access_grants set updated_at = current_timestamp, deleted = true where id = $1") + if err != nil { + return errors.Wrap(err, "error preparing access_grants delete statement") + } + _, err = stmt.Exec(id) + if err != nil { + return errors.Wrap(err, "error executing access_grants delete statement") + } + return nil +}