Gleipnir/nidus-sync
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Move address list func to types so it can be shared with csv

Browse source 
And stop double-geocoding all the rows.
This commit is contained in:
Eli Ribble 2026-04-16 03:06:18 +00:00
parent 5bf93c3dfd
commit b2d8e3ba27
No known key found for this signature in database
4 changed files with 73 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

36
platform/types/address.go
View file

@ -1,10 +1,17 @@
package types
import (
"context"
"fmt"
"github.com/Gleipnir-Technology/bob"
"github.com/Gleipnir-Technology/bob/dialect/psql"
"github.com/Gleipnir-Technology/bob/dialect/psql/sm"
"github.com/Gleipnir-Technology/nidus-sync/db"
"github.com/Gleipnir-Technology/nidus-sync/db/models"
//"github.com/rs/zerolog/log"
"github.com/stephenafamo/scan"
)
type Address struct {
@ -43,3 +50,32 @@ func AddressFromModel(m *models.Address) Address {
Unit: m.Unit,
}
}
func AddressList(ctx context.Context, ids []int32) (map[int32]*Address, error) {
rows, err := bob.All(ctx, db.PGInstance.BobDB, psql.Select(
sm.Columns(
"COALESCE(address.country, 'usa') AS \"country\"",
"COALESCE(address.gid, '') AS \"gid\"",
"COALESCE(address.id, '') AS \"id\"",
"COALESCE(address.locality, '') AS \"locality\"",
"COALESCE(address.number_, '') AS \"number\"",
"COALESCE(address.postal_code, '') AS \"postal_code\"",
"COALESCE(address.region, '') AS \"region\"",
"COALESCE(address.street, '') AS \"street\"",
"COALESCE(address.unit, '') AS \"unit\"",
// This will work great, up until we add polygons to signal
"COALESCE(address.location_latitude, 0) AS \"location.latitude\"",
"COALESCE(address.location_longitude, 0) AS \"location.longitude\"",
),
sm.From("address"),
sm.Where(psql.Quote("address", "id").EQ(psql.Any(ids))),
), scan.StructMapper[*Address]())
if err != nil {
return nil, fmt.Errorf("query addresses: %w", err)
}
addresses_by_id := make(map[int32]*Address, len(rows))
for _, a := range rows {
addresses_by_id[*a.ID] = a
}
return addresses_by_id, err
}