Janady13/missionmatch / internal/platform/database/dbgen/opportunity_records.sql.go

1// Code generated by sqlc. DO NOT EDIT. 2// versions: 3// sqlc v1.31.1 4// source: opportunity_records.sql 5 6package dbgen 7 8import ( 9 "context" 10 "time" 11 12 "github.com/google/uuid" 13 "github.com/jackc/pgx/v5/pgtype" 14 "github.com/shopspring/decimal" 15) 16 17const getOpportunityRecord = `-- name: GetOpportunityRecord :one 18SELECT id, source_name, source_record_id, dedupe_key, record_kind, title, agency, sub_agency, naics, psc, pop_city, pop_state, pop_country, pop_zip, set_aside, set_aside_code, contract_vehicle, estimated_value, value_source, incumbent_name, incumbent_uei, posted_date, response_deadline, award_date, estimated_award_date, pop_start_date, pop_end_date, ui_link, is_planning, raw_object_key, raw_sha256, fetched_at, synced_at, embedding, search_tsv, provenance FROM opportunity_records WHERE id = $1 19` 20 21func (q *Queries) GetOpportunityRecord(ctx context.Context, id uuid.UUID) (OpportunityRecord, error) { 22 row := q.db.QueryRow(ctx, getOpportunityRecord, id) 23 var i OpportunityRecord 24 err := row.Scan( 25 &i.ID, 26 &i.SourceName, 27 &i.SourceRecordID, 28 &i.DedupeKey, 29 &i.RecordKind, 30 &i.Title, 31 &i.Agency, 32 &i.SubAgency, 33 &i.Naics, 34 &i.Psc, 35 &i.PopCity, 36 &i.PopState, 37 &i.PopCountry, 38 &i.PopZip, 39 &i.SetAside, 40 &i.SetAsideCode, 41 &i.ContractVehicle, 42 &i.EstimatedValue, 43 &i.ValueSource, 44 &i.IncumbentName, 45 &i.IncumbentUei, 46 &i.PostedDate, 47 &i.ResponseDeadline, 48 &i.AwardDate, 49 &i.EstimatedAwardDate, 50 &i.PopStartDate, 51 &i.PopEndDate, 52 &i.UiLink, 53 &i.IsPlanning, 54 &i.RawObjectKey, 55 &i.RawSha256, 56 &i.FetchedAt, 57 &i.SyncedAt, 58 &i.Embedding, 59 &i.SearchTsv, 60 &i.Provenance, 61 ) 62 return i, err 63} 64 65const listOpportunityRecords = `-- name: ListOpportunityRecords :many 66SELECT id, source_name, source_record_id, dedupe_key, record_kind, title, agency, sub_agency, naics, psc, pop_city, pop_state, pop_country, pop_zip, set_aside, set_aside_code, contract_vehicle, estimated_value, value_source, incumbent_name, incumbent_uei, posted_date, response_deadline, award_date, estimated_award_date, pop_start_date, pop_end_date, ui_link, is_planning, raw_object_key, raw_sha256, fetched_at, synced_at, embedding, search_tsv, provenance FROM opportunity_records 67WHERE ($1::text[] IS NULL OR record_kind = ANY($1::text[])) 68 AND ($2::text[] IS NULL OR naics = ANY($2::text[])) 69 AND ($3::text[] IS NULL OR upper(agency) = ANY($3::text[])) 70 AND ($4::text[] IS NULL OR upper(pop_state) = ANY($4::text[])) 71 AND ($5::text[] IS NULL OR EXISTS ( 72 SELECT 1 FROM federal_mission_tags t 73 WHERE t.record_table = 'opportunity_records' 74 AND t.record_id = opportunity_records.id::text 75 AND t.mission_code = ANY($5::text[]))) 76ORDER BY (response_deadline IS NULL OR response_deadline < now()), response_deadline ASC, posted_date DESC NULLS LAST, id 77LIMIT $6::int 78` 79 80type ListOpportunityRecordsParams struct { 81 Kinds []string `json:"kinds"` 82 Naics []string `json:"naics"` 83 Agencies []string `json:"agencies"` 84 States []string `json:"states"` 85 Missions []string `json:"missions"` 86 Lim int32 `json:"lim"` 87} 88 89// Optional filters (NULL array = no filter). Soonest closing first, then most recently posted, then id 90// as a unique tiebreaker so the order is deterministic. Text equality filters compare upper() on both 91// sides; the service upper cases the params so casing drift cannot silently zero out results. The 92// optional missions filter keeps a record only when a federal_mission_tag links it to one of the named 93// mission codes; a NULL missions array means no mission filter, so the default path is unchanged. 94func (q *Queries) ListOpportunityRecords(ctx context.Context, arg ListOpportunityRecordsParams) ([]OpportunityRecord, error) { 95 rows, err := q.db.Query(ctx, listOpportunityRecords, 96 arg.Kinds, 97 arg.Naics, 98 arg.Agencies, 99 arg.States, 100 arg.Missions, 101 arg.Lim, 102 ) 103 if err != nil { 104 return nil, err 105 } 106 defer rows.Close() 107 items := []OpportunityRecord{} 108 for rows.Next() { 109 var i OpportunityRecord 110 if err := rows.Scan( 111 &i.ID, 112 &i.SourceName, 113 &i.SourceRecordID, 114 &i.DedupeKey, 115 &i.RecordKind, 116 &i.Title, 117 &i.Agency, 118 &i.SubAgency, 119 &i.Naics, 120 &i.Psc, 121 &i.PopCity, 122 &i.PopState, 123 &i.PopCountry, 124 &i.PopZip, 125 &i.SetAside, 126 &i.SetAsideCode, 127 &i.ContractVehicle, 128 &i.EstimatedValue, 129 &i.ValueSource, 130 &i.IncumbentName, 131 &i.IncumbentUei, 132 &i.PostedDate, 133 &i.ResponseDeadline, 134 &i.AwardDate, 135 &i.EstimatedAwardDate, 136 &i.PopStartDate, 137 &i.PopEndDate, 138 &i.UiLink, 139 &i.IsPlanning, 140 &i.RawObjectKey, 141 &i.RawSha256, 142 &i.FetchedAt, 143 &i.SyncedAt, 144 &i.Embedding, 145 &i.SearchTsv, 146 &i.Provenance, 147 ); err != nil { 148 return nil, err 149 } 150 items = append(items, i) 151 } 152 if err := rows.Err(); err != nil { 153 return nil, err 154 } 155 return items, nil 156} 157 158const opportunityRecordMeta = `-- name: OpportunityRecordMeta :one 159SELECT COUNT(*)::bigint AS total, MAX(synced_at) AS last_synced FROM opportunity_records 160` 161 162type OpportunityRecordMetaRow struct { 163 Total int64 `json:"total"` 164 LastSynced interface{} `json:"last_synced"` 165} 166 167func (q *Queries) OpportunityRecordMeta(ctx context.Context) (OpportunityRecordMetaRow, error) { 168 row := q.db.QueryRow(ctx, opportunityRecordMeta) 169 var i OpportunityRecordMetaRow 170 err := row.Scan(&i.Total, &i.LastSynced) 171 return i, err 172} 173 174const upsertOpportunityRecord = `-- name: UpsertOpportunityRecord :exec 175 176INSERT INTO opportunity_records ( 177 source_name, source_record_id, dedupe_key, record_kind, title, agency, sub_agency, 178 naics, psc, pop_city, pop_state, pop_country, pop_zip, set_aside, set_aside_code, 179 contract_vehicle, estimated_value, value_source, incumbent_name, incumbent_uei, 180 posted_date, response_deadline, award_date, estimated_award_date, pop_start_date, pop_end_date, 181 ui_link, is_planning, raw_object_key, raw_sha256, fetched_at, synced_at 182) VALUES ( 183 $1, $2, $3, $4, $5, $6, $7, 184 $8, $9, $10, $11, $12, $13, $14, $15, 185 $16, $17, $18, $19, $20, 186 $21, $22, $23, $24, $25, $26, 187 $27, $28, $29, $30, $31, now() 188) 189ON CONFLICT (source_name, source_record_id) DO UPDATE SET 190 dedupe_key = EXCLUDED.dedupe_key, 191 record_kind = EXCLUDED.record_kind, 192 title = EXCLUDED.title, 193 agency = EXCLUDED.agency, 194 sub_agency = EXCLUDED.sub_agency, 195 naics = EXCLUDED.naics, 196 psc = EXCLUDED.psc, 197 pop_city = EXCLUDED.pop_city, 198 pop_state = EXCLUDED.pop_state, 199 pop_country = EXCLUDED.pop_country, 200 pop_zip = EXCLUDED.pop_zip, 201 set_aside = EXCLUDED.set_aside, 202 set_aside_code = EXCLUDED.set_aside_code, 203 contract_vehicle = EXCLUDED.contract_vehicle, 204 estimated_value = EXCLUDED.estimated_value, 205 value_source = EXCLUDED.value_source, 206 incumbent_name = EXCLUDED.incumbent_name, 207 incumbent_uei = EXCLUDED.incumbent_uei, 208 posted_date = EXCLUDED.posted_date, 209 response_deadline = EXCLUDED.response_deadline, 210 award_date = EXCLUDED.award_date, 211 estimated_award_date = EXCLUDED.estimated_award_date, 212 pop_start_date = EXCLUDED.pop_start_date, 213 pop_end_date = EXCLUDED.pop_end_date, 214 ui_link = EXCLUDED.ui_link, 215 is_planning = EXCLUDED.is_planning, 216 raw_object_key = EXCLUDED.raw_object_key, 217 raw_sha256 = EXCLUDED.raw_sha256, 218 fetched_at = EXCLUDED.fetched_at, 219 synced_at = now() 220` 221 222type UpsertOpportunityRecordParams struct { 223 SourceName string `json:"source_name"` 224 SourceRecordID string `json:"source_record_id"` 225 DedupeKey string `json:"dedupe_key"` 226 RecordKind string `json:"record_kind"` 227 Title string `json:"title"` 228 Agency string `json:"agency"` 229 SubAgency string `json:"sub_agency"` 230 Naics string `json:"naics"` 231 Psc string `json:"psc"` 232 PopCity string `json:"pop_city"` 233 PopState string `json:"pop_state"` 234 PopCountry string `json:"pop_country"` 235 PopZip string `json:"pop_zip"` 236 SetAside string `json:"set_aside"` 237 SetAsideCode string `json:"set_aside_code"` 238 ContractVehicle string `json:"contract_vehicle"` 239 EstimatedValue *decimal.Decimal `json:"estimated_value"` 240 ValueSource string `json:"value_source"` 241 IncumbentName string `json:"incumbent_name"` 242 IncumbentUei string `json:"incumbent_uei"` 243 PostedDate pgtype.Date `json:"posted_date"` 244 ResponseDeadline *time.Time `json:"response_deadline"` 245 AwardDate pgtype.Date `json:"award_date"` 246 EstimatedAwardDate pgtype.Date `json:"estimated_award_date"` 247 PopStartDate pgtype.Date `json:"pop_start_date"` 248 PopEndDate pgtype.Date `json:"pop_end_date"` 249 UiLink string `json:"ui_link"` 250 IsPlanning bool `json:"is_planning"` 251 RawObjectKey string `json:"raw_object_key"` 252 RawSha256 string `json:"raw_sha256"` 253 FetchedAt time.Time `json:"fetched_at"` 254} 255 256// Opportunity Intelligence unified records. Global public catalog (no organization_id, outside RLS): 257// reads run straight off the pool, writes come from cmd/opp-sync as the migrator role. 258func (q *Queries) UpsertOpportunityRecord(ctx context.Context, arg UpsertOpportunityRecordParams) error { 259 _, err := q.db.Exec(ctx, upsertOpportunityRecord, 260 arg.SourceName, 261 arg.SourceRecordID, 262 arg.DedupeKey, 263 arg.RecordKind, 264 arg.Title, 265 arg.Agency, 266 arg.SubAgency, 267 arg.Naics, 268 arg.Psc, 269 arg.PopCity, 270 arg.PopState, 271 arg.PopCountry, 272 arg.PopZip, 273 arg.SetAside, 274 arg.SetAsideCode, 275 arg.ContractVehicle, 276 arg.EstimatedValue, 277 arg.ValueSource, 278 arg.IncumbentName, 279 arg.IncumbentUei, 280 arg.PostedDate, 281 arg.ResponseDeadline, 282 arg.AwardDate, 283 arg.EstimatedAwardDate, 284 arg.PopStartDate, 285 arg.PopEndDate, 286 arg.UiLink, 287 arg.IsPlanning, 288 arg.RawObjectKey, 289 arg.RawSha256, 290 arg.FetchedAt, 291 ) 292 return err 293} 294