mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-12 01:27:32 +00:00
chore: move the go code into a subfolder
This commit is contained in:
178
apps/api/internal/query/org_domain.go
Normal file
178
apps/api/internal/query/org_domain.go
Normal file
@@ -0,0 +1,178 @@
|
||||
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/query/projection"
|
||||
"github.com/zitadel/zitadel/internal/telemetry/tracing"
|
||||
"github.com/zitadel/zitadel/internal/zerrors"
|
||||
)
|
||||
|
||||
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, zerrors.ThrowInvalidArgument(err, "QUERY-ZRfj1", "Errors.Query.SQLStatement")
|
||||
}
|
||||
|
||||
err = q.client.QueryContext(ctx, func(rows *sql.Rows) error {
|
||||
domains, err = scan(rows)
|
||||
return err
|
||||
}, stmt, args...)
|
||||
if err != nil {
|
||||
return nil, zerrors.ThrowInternal(err, "QUERY-M6mYN", "Errors.Internal")
|
||||
}
|
||||
|
||||
domains.State, err = q.latestState(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, zerrors.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,
|
||||
}
|
||||
)
|
Reference in New Issue
Block a user