repo

type Databases struct {
	Databases map[string]uuid.UUID `json:"databases,omitempty"` // the key is the UUIDv5 sha of the database
}

type Queries struct {
	Queries             map[string]*Query                         `json:"queries,omitempty"`
	FunctionsInQueries  map[string]*extractor.FunctionsInQueries  `json:"functions_in_queries,omitempty"`
	ColumnsInQueries    map[string]*extractor.ColumnsInQueries    `json:"columns_in_queries,omitempty"`
	TablesInQueries     map[string]*extractor.TablesInQueries     `json:"tables_in_queries,omitempty"`
	TableJoinsInQueries map[string]*extractor.TableJoinsInQueries `json:"table_joins_in_queries,omitempty"`
}

type Query struct {
	UID           uuid.UUID       `json:"uid,omitempty"`            // unique sha of the query
	DatabaseUID   uuid.UUID       `json:"database_uid,omitempty"`   // the dataset the query belongs to
	SourceUID     uuid.UUID       `json:"source_uid,omitempty"`     // the source the query belongs to
	Command       token.TokenType `json:"command,omitempty"`        // the type of query
	TotalCount    int64           `json:"total_count,omitempty"`    // the number of times the query was executed
	TotalDuration int64           `json:"total_duration,omitempty"` // the total duration of all executions of the query in microseconds
	MaskedQuery   string          `json:"masked_query,omitempty"`   // the query with parameters masked
	UnmaskedQuery string          `json:"unmasked_query,omitempty"` // the query with parameters unmasked
	SourceQuery   string          `json:"source,omitempty"`         // the original query from the source
}

type QueryWorker struct {
	Databases   *Databases
	Source      *Source
	SourceUID   uuid.UUID
	Database    string
	DatabaseUID uuid.UUID
	Input       string // Original query. This may contain many queries
	Duration    int64
	MustExtract bool
	Command     token.TokenType
	Masked      string // Masked query. This is the query with all values replaced with ?
	Unmasked    string // Unmasked query. This is the query with all values left alone
}

type Source struct {
	UID  uuid.UUID `json:"uid,omitempty"`  // unique UUID of the source
	Name string    `json:"name,omitempty"` // the name of the source
	URL  string    `json:"url,omitempty"`  // the url of the source
}

type Sources struct {
	Sources map[string]*Source `json:"sources,omitempty"` // the key is the sha of the database
}