feat: list clients by ID

GitOrigin-RevId: 7fbfc60678beca387f854a96b55e908632508388

Author: zepatrik

client/manager.go

@@ -15,6 +15,7 @@ type Filter struct {
 	PageOpts []keysetpagination.Option
 	Name     string
 	Owner    string
+	IDs      []string
 }
 
 type Manager interface {

client/manager_test_helpers.go

@@ -313,6 +313,31 @@ func TestHelperCreateGetUpdateDeleteClient(t1, t2 Storage) func(t *testing.T) {
 			assert.True(t, nextPage.IsLast())
 		})
 
+		t.Run("list by ids", func(t *testing.T) {
+			cs, nextPage, err := t1.GetClients(ctx, Filter{IDs: []string{t1c1.ID, c2Template.ID}})
+			require.NoError(t, err)
+			require.Len(t, cs, 2)
+			assert.ElementsMatch(t, []string{t1c1.ID, c2Template.ID}, []string{cs[0].GetID(), cs[1].GetID()})
+			assert.True(t, nextPage.IsLast())
+
+			cs, nextPage, err = t1.GetClients(ctx, Filter{IDs: []string{t1c1.ID}})
+			require.NoError(t, err)
+			require.Len(t, cs, 1)
+			assert.Equal(t, t1c1.ID, cs[0].GetID())
+			assert.True(t, nextPage.IsLast())
+
+			cs, nextPage, err = t1.GetClients(ctx, Filter{IDs: []string{c2Template.ID}})
+			require.NoError(t, err)
+			require.Len(t, cs, 1)
+			assert.Equal(t, c2Template.ID, cs[0].GetID())
+			assert.True(t, nextPage.IsLast())
+
+			cs, nextPage, err = t1.GetClients(ctx, Filter{IDs: []string{uuidx.NewV4().String()}})
+			require.NoError(t, err)
+			require.Len(t, cs, 0)
+			assert.True(t, nextPage.IsLast())
+		})
+
 		testHelperUpdateClient(t, ctx, t1, t1c1)
 		testHelperUpdateClient(t, ctx, t2, &t2c1)
 

persistence/sql/persister_client.go

@@ -160,6 +160,9 @@ func (p *Persister) GetClients(ctx context.Context, filters client.Filter) (cs [
 	if filters.Owner != "" {
 		query.Where("owner = ?", filters.Owner)
 	}
+	if len(filters.IDs) > 0 {
+		query.Where("id IN (?)", filters.IDs)
+	}
 
 	if err := query.All(&cs); err != nil {
 		return nil, nil, sqlcon.HandleError(err)