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

1// Code generated by sqlc. DO NOT EDIT. 2// versions: 3// sqlc v1.31.1 4// source: awards.sql 5 6package dbgen 7 8import ( 9 "context" 10 11 "github.com/jackc/pgx/v5/pgtype" 12 "github.com/shopspring/decimal" 13) 14 15const awardsSyncMeta = `-- name: AwardsSyncMeta :one 16SELECT COUNT(*)::bigint AS total, 17 COUNT(*) FILTER (WHERE is_galapagos)::bigint AS galapagos, 18 MAX(synced_at) AS last_synced 19FROM usaspending_awards 20` 21 22type AwardsSyncMetaRow struct { 23 Total int64 `json:"total"` 24 Galapagos int64 `json:"galapagos"` 25 LastSynced interface{} `json:"last_synced"` 26} 27 28func (q *Queries) AwardsSyncMeta(ctx context.Context) (AwardsSyncMetaRow, error) { 29 row := q.db.QueryRow(ctx, awardsSyncMeta) 30 var i AwardsSyncMetaRow 31 err := row.Scan(&i.Total, &i.Galapagos, &i.LastSynced) 32 return i, err 33} 34 35const listAwardsByNaics = `-- name: ListAwardsByNaics :many 36SELECT generated_internal_id, award_id, recipient_name, recipient_uei, recipient_id, award_amount, awarding_agency, awarding_sub_agency, pop_start_date, pop_end_date, naics_code, naics_description, description, is_galapagos, raw, source, synced_at FROM usaspending_awards 37WHERE naics_code = ANY($1::text[]) 38 AND NOT is_galapagos 39ORDER BY award_amount DESC NULLS LAST, generated_internal_id 40LIMIT $2::int 41` 42 43type ListAwardsByNaicsParams struct { 44 Naics []string `json:"naics"` 45 Lim int32 `json:"lim"` 46} 47 48// Incumbents and similar awards for a NAICS, largest first (excludes Galapagos own awards). 49func (q *Queries) ListAwardsByNaics(ctx context.Context, arg ListAwardsByNaicsParams) ([]UsaspendingAward, error) { 50 rows, err := q.db.Query(ctx, listAwardsByNaics, arg.Naics, arg.Lim) 51 if err != nil { 52 return nil, err 53 } 54 defer rows.Close() 55 items := []UsaspendingAward{} 56 for rows.Next() { 57 var i UsaspendingAward 58 if err := rows.Scan( 59 &i.GeneratedInternalID, 60 &i.AwardID, 61 &i.RecipientName, 62 &i.RecipientUei, 63 &i.RecipientID, 64 &i.AwardAmount, 65 &i.AwardingAgency, 66 &i.AwardingSubAgency, 67 &i.PopStartDate, 68 &i.PopEndDate, 69 &i.NaicsCode, 70 &i.NaicsDescription, 71 &i.Description, 72 &i.IsGalapagos, 73 &i.Raw, 74 &i.Source, 75 &i.SyncedAt, 76 ); err != nil { 77 return nil, err 78 } 79 items = append(items, i) 80 } 81 if err := rows.Err(); err != nil { 82 return nil, err 83 } 84 return items, nil 85} 86 87const listGalapagosAwards = `-- name: ListGalapagosAwards :many 88SELECT generated_internal_id, award_id, recipient_name, recipient_uei, recipient_id, award_amount, awarding_agency, awarding_sub_agency, pop_start_date, pop_end_date, naics_code, naics_description, description, is_galapagos, raw, source, synced_at FROM usaspending_awards 89WHERE is_galapagos 90ORDER BY award_amount DESC NULLS LAST, generated_internal_id 91LIMIT $1::int 92` 93 94// Galapagos own past performance, largest first. 95func (q *Queries) ListGalapagosAwards(ctx context.Context, lim int32) ([]UsaspendingAward, error) { 96 rows, err := q.db.Query(ctx, listGalapagosAwards, lim) 97 if err != nil { 98 return nil, err 99 } 100 defer rows.Close() 101 items := []UsaspendingAward{} 102 for rows.Next() { 103 var i UsaspendingAward 104 if err := rows.Scan( 105 &i.GeneratedInternalID, 106 &i.AwardID, 107 &i.RecipientName, 108 &i.RecipientUei, 109 &i.RecipientID, 110 &i.AwardAmount, 111 &i.AwardingAgency, 112 &i.AwardingSubAgency, 113 &i.PopStartDate, 114 &i.PopEndDate, 115 &i.NaicsCode, 116 &i.NaicsDescription, 117 &i.Description, 118 &i.IsGalapagos, 119 &i.Raw, 120 &i.Source, 121 &i.SyncedAt, 122 ); err != nil { 123 return nil, err 124 } 125 items = append(items, i) 126 } 127 if err := rows.Err(); err != nil { 128 return nil, err 129 } 130 return items, nil 131} 132 133const upsertAward = `-- name: UpsertAward :exec 134 135INSERT INTO usaspending_awards ( 136 generated_internal_id, award_id, recipient_name, recipient_uei, recipient_id, award_amount, 137 awarding_agency, awarding_sub_agency, pop_start_date, pop_end_date, naics_code, naics_description, 138 description, is_galapagos, raw, source, synced_at 139) VALUES ( 140 $1, $2, $3, $4, $5, $6, 141 $7, $8, $9, $10, $11, $12, 142 $13, $14, $15, $16, now() 143) 144ON CONFLICT (generated_internal_id) DO UPDATE SET 145 award_id = EXCLUDED.award_id, 146 recipient_name = EXCLUDED.recipient_name, 147 recipient_uei = EXCLUDED.recipient_uei, 148 recipient_id = EXCLUDED.recipient_id, 149 award_amount = EXCLUDED.award_amount, 150 awarding_agency = EXCLUDED.awarding_agency, 151 awarding_sub_agency = EXCLUDED.awarding_sub_agency, 152 pop_start_date = EXCLUDED.pop_start_date, 153 pop_end_date = EXCLUDED.pop_end_date, 154 naics_code = EXCLUDED.naics_code, 155 naics_description = EXCLUDED.naics_description, 156 description = EXCLUDED.description, 157 is_galapagos = EXCLUDED.is_galapagos OR usaspending_awards.is_galapagos, 158 raw = EXCLUDED.raw, 159 source = EXCLUDED.source, 160 synced_at = now() 161` 162 163type UpsertAwardParams struct { 164 GeneratedInternalID string `json:"generated_internal_id"` 165 AwardID string `json:"award_id"` 166 RecipientName string `json:"recipient_name"` 167 RecipientUei string `json:"recipient_uei"` 168 RecipientID string `json:"recipient_id"` 169 AwardAmount *decimal.Decimal `json:"award_amount"` 170 AwardingAgency string `json:"awarding_agency"` 171 AwardingSubAgency string `json:"awarding_sub_agency"` 172 PopStartDate pgtype.Date `json:"pop_start_date"` 173 PopEndDate pgtype.Date `json:"pop_end_date"` 174 NaicsCode string `json:"naics_code"` 175 NaicsDescription string `json:"naics_description"` 176 Description string `json:"description"` 177 IsGalapagos bool `json:"is_galapagos"` 178 Raw []byte `json:"raw"` 179 Source string `json:"source"` 180} 181 182// USASpending awards cache. Global public catalog (no organization_id, outside RLS): reads run off 183// the pool, writes come from the daily sync (cmd/awards-sync) as the migrator role. 184func (q *Queries) UpsertAward(ctx context.Context, arg UpsertAwardParams) error { 185 _, err := q.db.Exec(ctx, upsertAward, 186 arg.GeneratedInternalID, 187 arg.AwardID, 188 arg.RecipientName, 189 arg.RecipientUei, 190 arg.RecipientID, 191 arg.AwardAmount, 192 arg.AwardingAgency, 193 arg.AwardingSubAgency, 194 arg.PopStartDate, 195 arg.PopEndDate, 196 arg.NaicsCode, 197 arg.NaicsDescription, 198 arg.Description, 199 arg.IsGalapagos, 200 arg.Raw, 201 arg.Source, 202 ) 203 return err 204} 205