Function loadAIProviderKeysByProvider

coderd/ai_providers.go:567–577

loadAIProviderKeysByProvider fetches keys for every live provider in one query and buckets the rows by ProviderID, so the list handler can avoid an N+1 fetch. Soft-deleted providers' keys are excluded by the query.

(ctx context.Context, store database.Store)

// can avoid an N+1 fetch. Soft-deleted providers' keys are excluded
// by the query.
func loadAIProviderKeysByProvider(ctx context.Context, store database.Store) (map[uuid.UUID][]database.AIProviderKey, error) {
	rows, err := store.GetAIProviderKeys(ctx, false)
	if err != nil {
		return nil, err
	}
	out := make(map[uuid.UUID][]database.AIProviderKey, len(rows))
	for _, row := range rows {
		out[row.ProviderID] = append(out[row.ProviderID], row)
	}
	return out, nil
}

// insertAIProviderKeys writes a fresh set of key rows for a provider
// inside a transaction. It returns the inserted rows in insertion