mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-13 03:24:26 +00:00
28760ab4b3
* fix(query): add tracing for each method
178 lines
4.6 KiB
Go
178 lines
4.6 KiB
Go
package query
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"time"
|
|
|
|
sq "github.com/Masterminds/squirrel"
|
|
|
|
"github.com/zitadel/zitadel/internal/api/authz"
|
|
"github.com/zitadel/zitadel/internal/domain"
|
|
"github.com/zitadel/zitadel/internal/errors"
|
|
"github.com/zitadel/zitadel/internal/query/projection"
|
|
"github.com/zitadel/zitadel/internal/telemetry/tracing"
|
|
)
|
|
|
|
type Domain struct {
|
|
CreationDate time.Time
|
|
ChangeDate time.Time
|
|
Sequence uint64
|
|
Domain string
|
|
OrgID string
|
|
IsVerified bool
|
|
IsPrimary bool
|
|
ValidationType domain.OrgDomainValidationType
|
|
}
|
|
|
|
type Domains struct {
|
|
SearchResponse
|
|
Domains []*Domain
|
|
}
|
|
|
|
type OrgDomainSearchQueries struct {
|
|
SearchRequest
|
|
Queries []SearchQuery
|
|
}
|
|
|
|
func (q *OrgDomainSearchQueries) toQuery(query sq.SelectBuilder) sq.SelectBuilder {
|
|
query = q.SearchRequest.toQuery(query)
|
|
for _, q := range q.Queries {
|
|
query = q.toQuery(query)
|
|
}
|
|
return query
|
|
}
|
|
|
|
func NewOrgDomainDomainSearchQuery(method TextComparison, value string) (SearchQuery, error) {
|
|
return NewTextQuery(OrgDomainDomainCol, value, method)
|
|
}
|
|
|
|
func NewOrgDomainOrgIDSearchQuery(value string) (SearchQuery, error) {
|
|
return NewTextQuery(OrgDomainOrgIDCol, value, TextEquals)
|
|
}
|
|
|
|
func NewOrgDomainVerifiedSearchQuery(verified bool) (SearchQuery, error) {
|
|
return NewBoolQuery(OrgDomainIsVerifiedCol, verified)
|
|
}
|
|
|
|
func (q *Queries) SearchOrgDomains(ctx context.Context, queries *OrgDomainSearchQueries, withOwnerRemoved bool) (domains *Domains, err error) {
|
|
ctx, span := tracing.NewSpan(ctx)
|
|
defer func() { span.EndWithError(err) }()
|
|
|
|
query, scan := prepareDomainsQuery()
|
|
eq := sq.Eq{OrgDomainInstanceIDCol.identifier(): authz.GetInstance(ctx).InstanceID()}
|
|
if !withOwnerRemoved {
|
|
eq[OrgDomainOwnerRemovedCol.identifier()] = false
|
|
}
|
|
stmt, args, err := queries.toQuery(query).Where(eq).ToSql()
|
|
if err != nil {
|
|
return nil, errors.ThrowInvalidArgument(err, "QUERY-ZRfj1", "Errors.Query.SQLStatement")
|
|
}
|
|
|
|
rows, err := q.client.QueryContext(ctx, stmt, args...)
|
|
if err != nil {
|
|
return nil, errors.ThrowInternal(err, "QUERY-M6mYN", "Errors.Internal")
|
|
}
|
|
domains, err = scan(rows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
domains.LatestSequence, err = q.latestSequence(ctx, orgDomainsTable)
|
|
return domains, err
|
|
}
|
|
|
|
func prepareDomainsQuery() (sq.SelectBuilder, func(*sql.Rows) (*Domains, error)) {
|
|
return sq.Select(
|
|
OrgDomainCreationDateCol.identifier(),
|
|
OrgDomainChangeDateCol.identifier(),
|
|
OrgDomainSequenceCol.identifier(),
|
|
OrgDomainDomainCol.identifier(),
|
|
OrgDomainOrgIDCol.identifier(),
|
|
OrgDomainIsVerifiedCol.identifier(),
|
|
OrgDomainIsPrimaryCol.identifier(),
|
|
OrgDomainValidationTypeCol.identifier(),
|
|
countColumn.identifier(),
|
|
).From(orgDomainsTable.identifier()).PlaceholderFormat(sq.Dollar),
|
|
func(rows *sql.Rows) (*Domains, error) {
|
|
domains := make([]*Domain, 0)
|
|
var count uint64
|
|
for rows.Next() {
|
|
domain := new(Domain)
|
|
err := rows.Scan(
|
|
&domain.CreationDate,
|
|
&domain.ChangeDate,
|
|
&domain.Sequence,
|
|
&domain.Domain,
|
|
&domain.OrgID,
|
|
&domain.IsVerified,
|
|
&domain.IsPrimary,
|
|
&domain.ValidationType,
|
|
&count,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
domains = append(domains, domain)
|
|
}
|
|
|
|
if err := rows.Close(); err != nil {
|
|
return nil, errors.ThrowInternal(err, "QUERY-rKd6k", "Errors.Query.CloseRows")
|
|
}
|
|
|
|
return &Domains{
|
|
Domains: domains,
|
|
SearchResponse: SearchResponse{
|
|
Count: count,
|
|
},
|
|
}, nil
|
|
}
|
|
}
|
|
|
|
var (
|
|
orgDomainsTable = table{
|
|
name: projection.OrgDomainTable,
|
|
instanceIDCol: projection.OrgDomainInstanceIDCol,
|
|
}
|
|
|
|
OrgDomainCreationDateCol = Column{
|
|
name: projection.OrgDomainCreationDateCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
OrgDomainChangeDateCol = Column{
|
|
name: projection.OrgDomainChangeDateCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
OrgDomainSequenceCol = Column{
|
|
name: projection.OrgDomainSequenceCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
OrgDomainDomainCol = Column{
|
|
name: projection.OrgDomainDomainCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
OrgDomainOrgIDCol = Column{
|
|
name: projection.OrgDomainOrgIDCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
OrgDomainInstanceIDCol = Column{
|
|
name: projection.OrgDomainInstanceIDCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
OrgDomainIsVerifiedCol = Column{
|
|
name: projection.OrgDomainIsVerifiedCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
OrgDomainIsPrimaryCol = Column{
|
|
name: projection.OrgDomainIsPrimaryCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
OrgDomainValidationTypeCol = Column{
|
|
name: projection.OrgDomainValidationTypeCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
OrgDomainOwnerRemovedCol = Column{
|
|
name: projection.OrgDomainOwnerRemovedCol,
|
|
table: orgDomainsTable,
|
|
}
|
|
)
|