extern/zrok
1
1
Fork 0
mirror of https://github.com/openziti/zrok.git synced 2025-08-15 18:42:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

roughed in store implementation for organizations and memberships (#537)

Browse Source
This commit is contained in:
Michael Quigley
2024-12-09 11:46:13 -05:00
parent d6c8f206b2
commit 842f3e8c20
3 changed files with 90 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
controller/store/organizationMember.go Normal file
View File

@ -0,0 +1,54 @@
package store
import (
"github.com/jmoiron/sqlx"
"github.com/pkg/errors"
)
func (str *Store) AddAccountToOrganization(acctId, orgId int, trx *sqlx.Tx) error {
stmt, err := trx.Prepare("insert into organization_members (organization_id, account_id) values ($1, $2)")
if err != nil {
return errors.Wrap(err, "error preparing organization_members insert statement")
}
_, err = stmt.Exec(acctId, orgId)
if err != nil {
return errors.Wrap(err, "error executing organization_members insert statement")
}
return nil
}
func (str *Store) IsAccountInOrganization(acctId, orgId int, trx *sqlx.Tx) (bool, error) {
stmt, err := trx.Prepare("select count(0) from organization_members where organization_id = $1 and account_id = $2")
if err != nil {
return false, errors.Wrap(err, "error preparing organization_members count statement")
}
var count int
if err := stmt.QueryRow(acctId, orgId).Scan(&count); err != nil {
return false, errors.Wrap(err, "error executing organization_members count statement")
}
return count > 0, nil
}
func (str *Store) IsAccountAdminOfOrganization(acctId, orgId int, trx *sqlx.Tx) (bool, error) {
stmt, err := trx.Prepare("select count(0) from organization_members where organization_id = $1 and account_id = $2 and admin")
if err != nil {
return false, errors.Wrap(err, "error preparing organization_members count statement")
}
var count int
if err := stmt.QueryRow(acctId, orgId).Scan(&count); err != nil {
return false, errors.Wrap(err, "error executing organization_members count statement")
}
return count > 0, nil
}
func (str *Store) RemoveAccountFromOrganization(acctId, orgId int, trx *sqlx.Tx) error {
stmt, err := trx.Prepare("delete from organization_members where organization_id = $1 and account_id = $2")
if err != nil {
return errors.Wrap(err, "error preparing organization_members delete statement")
}
_, err = stmt.Exec(acctId, orgId)
if err != nil {
return errors.Wrap(err, "error executing organization_members delete statement")
}
return nil
}