Function FetchSpecStore

utils/spec_store.go:31–55

FetchSpecStore loads GPU specs and availability from the API concurrently and merges them into a SpecStore. Specs are required; availability is best-effort — a failure leaves configurations unmarked rather than aborting.

(client *api.Client)

// merges them into a SpecStore. Specs are required; availability is best-effort
// — a failure leaves configurations unmarked rather than aborting.
func FetchSpecStore(client *api.Client) (*SpecStore, error) {
	var (
		specsMap     map[string]api.GpuSpecConfig
		specsErr     error
		availability map[string]string
		wg           sync.WaitGroup
	)
	wg.Add(2)
	go func() {
		defer wg.Done()
		specsMap, specsErr = client.GetSpecs()
	}()
	go func() {
		defer wg.Done()
		if avail, err := client.GetAvailability(); err == nil && avail != nil {
			availability = avail.Specs
		}
	}()
	wg.Wait()

	if specsErr != nil {
		return nil, fmt.Errorf("failed to fetch GPU specs: %w", specsErr)
	}
	return NewSpecStoreWithAvailability(specsMap, availability), nil
}

func publicConfigKey(gpuType string, gpuCount int) string {
	return fmt.Sprintf("%s_x%d", gpuType, gpuCount)