mirror of
https://github.com/sourcegraph/sourcegraph.git
synced 2026-02-06 18:51:59 +00:00
c836d6a333
This moves a bunch of stuff out of enterprise. Notably, the final result is `EnterpriseDB` no longer exists. This is one of the biggest sticking points for moving other stuff out of the `enterprise` directory, so this should help speed up future cleanups.
132 lines
2.8 KiB
Go
132 lines
2.8 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
"github.com/keegancsmith/sqlf"
|
|
|
|
"github.com/sourcegraph/sourcegraph/internal/database/dbutil"
|
|
)
|
|
|
|
type Recipient struct {
|
|
ID int64
|
|
Email int64
|
|
NamespaceUserID *int32
|
|
NamespaceOrgID *int32
|
|
}
|
|
|
|
var recipientColumns = []*sqlf.Query{
|
|
sqlf.Sprintf("cm_recipients.id"),
|
|
sqlf.Sprintf("cm_recipients.email"),
|
|
sqlf.Sprintf("cm_recipients.namespace_user_id"),
|
|
sqlf.Sprintf("cm_recipients.namespace_org_id"),
|
|
}
|
|
|
|
const createRecipientFmtStr = `
|
|
INSERT INTO cm_recipients (email, namespace_user_id, namespace_org_id)
|
|
VALUES (%s,%s,%s)
|
|
RETURNING %s
|
|
`
|
|
|
|
func (s *codeMonitorStore) CreateRecipient(ctx context.Context, emailID int64, userID, orgID *int32) (*Recipient, error) {
|
|
q := sqlf.Sprintf(createRecipientFmtStr, emailID, userID, orgID, sqlf.Join(recipientColumns, ","))
|
|
row := s.QueryRow(ctx, q)
|
|
return scanRecipient(row)
|
|
}
|
|
|
|
const deleteRecipientFmtStr = `
|
|
DELETE FROM cm_recipients
|
|
WHERE email = %s
|
|
`
|
|
|
|
func (s *codeMonitorStore) DeleteRecipients(ctx context.Context, emailID int64) error {
|
|
q := sqlf.Sprintf(
|
|
deleteRecipientFmtStr,
|
|
emailID,
|
|
)
|
|
return s.Exec(ctx, q)
|
|
}
|
|
|
|
type ListRecipientsOpts struct {
|
|
EmailID *int64
|
|
First *int
|
|
After *int64
|
|
}
|
|
|
|
func (l ListRecipientsOpts) Conds() *sqlf.Query {
|
|
conds := []*sqlf.Query{sqlf.Sprintf("TRUE")}
|
|
if l.EmailID != nil {
|
|
conds = append(conds, sqlf.Sprintf("email = %s", *l.EmailID))
|
|
}
|
|
if l.After != nil {
|
|
conds = append(conds, sqlf.Sprintf("id > %s", *l.After))
|
|
}
|
|
return sqlf.Join(conds, "AND")
|
|
}
|
|
|
|
func (l ListRecipientsOpts) Limit() *sqlf.Query {
|
|
if l.First == nil {
|
|
return sqlf.Sprintf("ALL")
|
|
}
|
|
return sqlf.Sprintf("%s", *l.First)
|
|
}
|
|
|
|
const readRecipientQueryFmtStr = `
|
|
SELECT %s -- recipientColumns
|
|
FROM cm_recipients
|
|
WHERE %s
|
|
ORDER BY id ASC
|
|
LIMIT %s;
|
|
`
|
|
|
|
func (s *codeMonitorStore) ListRecipients(ctx context.Context, args ListRecipientsOpts) ([]*Recipient, error) {
|
|
q := sqlf.Sprintf(
|
|
readRecipientQueryFmtStr,
|
|
sqlf.Join(recipientColumns, ","),
|
|
args.Conds(),
|
|
args.Limit(),
|
|
)
|
|
rows, err := s.Query(ctx, q)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
return scanRecipients(rows)
|
|
}
|
|
|
|
const totalCountRecipientsFmtStr = `
|
|
SELECT COUNT(*)
|
|
FROM cm_recipients
|
|
WHERE email = %s
|
|
`
|
|
|
|
func (s *codeMonitorStore) CountRecipients(ctx context.Context, emailID int64) (int32, error) {
|
|
var count int32
|
|
err := s.QueryRow(ctx, sqlf.Sprintf(totalCountRecipientsFmtStr, emailID)).Scan(&count)
|
|
return count, err
|
|
}
|
|
|
|
func scanRecipients(rows *sql.Rows) ([]*Recipient, error) {
|
|
var rs []*Recipient
|
|
for rows.Next() {
|
|
r, err := scanRecipient(rows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rs = append(rs, r)
|
|
}
|
|
return rs, rows.Err()
|
|
}
|
|
|
|
func scanRecipient(scanner dbutil.Scanner) (*Recipient, error) {
|
|
var r Recipient
|
|
err := scanner.Scan(
|
|
&r.ID,
|
|
&r.Email,
|
|
&r.NamespaceUserID,
|
|
&r.NamespaceOrgID,
|
|
)
|
|
return &r, err
|
|
}
|