cloud/bmaas/bmdb: Filter for machines based on provider
Currently the equinix shepherd tries to initialize our nodes from lumen,
which of course is not correct. This change adds another parameter to
the queries and prevents that.
Change-Id: Ib3f65e68403cb1b1282b80c1d494fb030a6d17b1
Reviewed-on: https://review.monogon.dev/c/monogon/+/2039
Reviewed-by: Serge Bazanski <serge@monogon.tech>
Tested-by: Jenkins CI
diff --git a/cloud/shepherd/equinix/manager/initializer_test.go b/cloud/shepherd/equinix/manager/initializer_test.go
index 20c2f16..82e1f90 100644
--- a/cloud/shepherd/equinix/manager/initializer_test.go
+++ b/cloud/shepherd/equinix/manager/initializer_test.go
@@ -14,6 +14,7 @@
"google.golang.org/protobuf/proto"
apb "source.monogon.dev/cloud/agent/api"
+
"source.monogon.dev/cloud/bmaas/bmdb"
"source.monogon.dev/cloud/bmaas/bmdb/model"
"source.monogon.dev/cloud/lib/component"
@@ -193,7 +194,10 @@
var machines []model.MachineProvided
err = sess.Transact(ctx, func(q *model.Queries) error {
var err error
- machines, err = q.GetMachinesForAgentStart(ctx, 100)
+ machines, err = q.GetMachinesForAgentStart(ctx, model.GetMachinesForAgentStartParams{
+ Limit: 100,
+ Provider: model.ProviderEquinix,
+ })
return err
})
if err != nil {