mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-12 03:37:34 +00:00
chore: move the go code into a subfolder
This commit is contained in:
138
apps/api/internal/query/milestone.go
Normal file
138
apps/api/internal/query/milestone.go
Normal file
@@ -0,0 +1,138 @@
|
||||
package query
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"time"
|
||||
|
||||
sq "github.com/Masterminds/squirrel"
|
||||
|
||||
"github.com/zitadel/zitadel/internal/api/authz"
|
||||
"github.com/zitadel/zitadel/internal/query/projection"
|
||||
"github.com/zitadel/zitadel/internal/repository/milestone"
|
||||
"github.com/zitadel/zitadel/internal/telemetry/tracing"
|
||||
"github.com/zitadel/zitadel/internal/zerrors"
|
||||
)
|
||||
|
||||
type Milestones struct {
|
||||
SearchResponse
|
||||
Milestones []*Milestone
|
||||
}
|
||||
|
||||
type Milestone struct {
|
||||
InstanceID string
|
||||
Type milestone.Type
|
||||
ReachedDate time.Time
|
||||
PushedDate time.Time
|
||||
PrimaryDomain string
|
||||
}
|
||||
|
||||
type MilestonesSearchQueries struct {
|
||||
SearchRequest
|
||||
Queries []SearchQuery
|
||||
}
|
||||
|
||||
func (q *MilestonesSearchQueries) toQuery(query sq.SelectBuilder) sq.SelectBuilder {
|
||||
query = q.SearchRequest.toQuery(query)
|
||||
for _, q := range q.Queries {
|
||||
query = q.toQuery(query)
|
||||
}
|
||||
return query
|
||||
}
|
||||
|
||||
var (
|
||||
milestonesTable = table{
|
||||
name: projection.MilestonesProjectionTable,
|
||||
instanceIDCol: projection.MilestoneColumnInstanceID,
|
||||
}
|
||||
MilestoneInstanceIDColID = Column{
|
||||
name: projection.MilestoneColumnInstanceID,
|
||||
table: milestonesTable,
|
||||
}
|
||||
MilestoneTypeColID = Column{
|
||||
name: projection.MilestoneColumnType,
|
||||
table: milestonesTable,
|
||||
}
|
||||
MilestoneReachedDateColID = Column{
|
||||
name: projection.MilestoneColumnReachedDate,
|
||||
table: milestonesTable,
|
||||
}
|
||||
MilestonePushedDateColID = Column{
|
||||
name: projection.MilestoneColumnPushedDate,
|
||||
table: milestonesTable,
|
||||
}
|
||||
)
|
||||
|
||||
// SearchMilestones tries to defer the instanceID from the passed context if no instanceIDs are passed
|
||||
func (q *Queries) SearchMilestones(ctx context.Context, instanceIDs []string, queries *MilestonesSearchQueries) (milestones *Milestones, err error) {
|
||||
ctx, span := tracing.NewSpan(ctx)
|
||||
defer func() { span.EndWithError(err) }()
|
||||
query, scan := prepareMilestonesQuery()
|
||||
if len(instanceIDs) == 0 {
|
||||
instanceIDs = []string{authz.GetInstance(ctx).InstanceID()}
|
||||
}
|
||||
stmt, args, err := queries.toQuery(query).Where(
|
||||
sq.Eq{
|
||||
MilestoneInstanceIDColID.identifier(): instanceIDs,
|
||||
InstanceDomainIsPrimaryCol.identifier(): true,
|
||||
},
|
||||
).ToSql()
|
||||
if err != nil {
|
||||
return nil, zerrors.ThrowInternal(err, "QUERY-A9i5k", "Errors.Query.SQLStatement")
|
||||
}
|
||||
err = q.client.QueryContext(ctx, func(rows *sql.Rows) error {
|
||||
milestones, err = scan(rows)
|
||||
return err
|
||||
}, stmt, args...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
milestones.State, err = q.latestState(ctx, milestonesTable)
|
||||
return milestones, err
|
||||
}
|
||||
|
||||
func prepareMilestonesQuery() (sq.SelectBuilder, func(*sql.Rows) (*Milestones, error)) {
|
||||
return sq.Select(
|
||||
MilestoneInstanceIDColID.identifier(),
|
||||
InstanceDomainDomainCol.identifier(),
|
||||
MilestoneReachedDateColID.identifier(),
|
||||
MilestonePushedDateColID.identifier(),
|
||||
MilestoneTypeColID.identifier(),
|
||||
countColumn.identifier(),
|
||||
).
|
||||
From(milestonesTable.identifier()).
|
||||
LeftJoin(join(InstanceDomainInstanceIDCol, MilestoneInstanceIDColID)).
|
||||
PlaceholderFormat(sq.Dollar),
|
||||
func(rows *sql.Rows) (*Milestones, error) {
|
||||
milestones := make([]*Milestone, 0)
|
||||
var count uint64
|
||||
for rows.Next() {
|
||||
m := new(Milestone)
|
||||
reachedDate := sql.NullTime{}
|
||||
pushedDate := sql.NullTime{}
|
||||
primaryDomain := sql.NullString{}
|
||||
err := rows.Scan(
|
||||
&m.InstanceID,
|
||||
&primaryDomain,
|
||||
&reachedDate,
|
||||
&pushedDate,
|
||||
&m.Type,
|
||||
&count,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
m.PrimaryDomain = primaryDomain.String
|
||||
m.ReachedDate = reachedDate.Time
|
||||
m.PushedDate = pushedDate.Time
|
||||
milestones = append(milestones, m)
|
||||
}
|
||||
return &Milestones{
|
||||
Milestones: milestones,
|
||||
SearchResponse: SearchResponse{
|
||||
Count: count,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user