diff --git a/internal/slurm/client.go b/internal/slurm/client.go index c0025d2..bc5c996 100644 --- a/internal/slurm/client.go +++ b/internal/slurm/client.go @@ -35,6 +35,18 @@ type Client struct { Licenses *LicensesService Reconfigure *ReconfigureService Shares *SharesService + + SlurmdbDiag *SlurmdbDiagService + SlurmdbConfig *SlurmdbConfigService + SlurmdbTres *SlurmdbTresService + SlurmdbQos *SlurmdbQosService + SlurmdbAssocs *SlurmdbAssocsService + SlurmdbInstances *SlurmdbInstancesService + SlurmdbUsers *SlurmdbUsersService + SlurmdbClusters *SlurmdbClustersService + SlurmdbWckeys *SlurmdbWckeysService + SlurmdbAccounts *SlurmdbAccountsService + SlurmdbJobs *SlurmdbJobsService } // service is the base struct for all API services. @@ -51,6 +63,17 @@ type PingService service type LicensesService service type ReconfigureService service type SharesService service +type SlurmdbDiagService service +type SlurmdbConfigService service +type SlurmdbTresService service +type SlurmdbQosService service +type SlurmdbAssocsService service +type SlurmdbInstancesService service +type SlurmdbUsersService service +type SlurmdbClustersService service +type SlurmdbWckeysService service +type SlurmdbAccountsService service +type SlurmdbJobsService service // Response wraps an http.Response and will later hold Meta/Errors/Warnings // fields parsed from Slurm API responses. @@ -90,6 +113,18 @@ func NewClient(baseURL string, httpClient *http.Client) (*Client, error) { c.Reconfigure = (*ReconfigureService)(&c.common) c.Shares = (*SharesService)(&c.common) + c.SlurmdbDiag = (*SlurmdbDiagService)(&c.common) + c.SlurmdbConfig = (*SlurmdbConfigService)(&c.common) + c.SlurmdbTres = (*SlurmdbTresService)(&c.common) + c.SlurmdbQos = (*SlurmdbQosService)(&c.common) + c.SlurmdbAssocs = (*SlurmdbAssocsService)(&c.common) + c.SlurmdbInstances = (*SlurmdbInstancesService)(&c.common) + c.SlurmdbUsers = (*SlurmdbUsersService)(&c.common) + c.SlurmdbClusters = (*SlurmdbClustersService)(&c.common) + c.SlurmdbWckeys = (*SlurmdbWckeysService)(&c.common) + c.SlurmdbAccounts = (*SlurmdbAccountsService)(&c.common) + c.SlurmdbJobs = (*SlurmdbJobsService)(&c.common) + return c, nil } diff --git a/internal/slurm/client_test.go b/internal/slurm/client_test.go index af55b5a..02efe33 100644 --- a/internal/slurm/client_test.go +++ b/internal/slurm/client_test.go @@ -56,6 +56,17 @@ func TestNewClient_ServicesInitialized(t *testing.T) { {"Licenses", client.Licenses}, {"Reconfigure", client.Reconfigure}, {"Shares", client.Shares}, + {"SlurmdbDiag", client.SlurmdbDiag}, + {"SlurmdbConfig", client.SlurmdbConfig}, + {"SlurmdbTres", client.SlurmdbTres}, + {"SlurmdbQos", client.SlurmdbQos}, + {"SlurmdbAssocs", client.SlurmdbAssocs}, + {"SlurmdbInstances", client.SlurmdbInstances}, + {"SlurmdbUsers", client.SlurmdbUsers}, + {"SlurmdbClusters", client.SlurmdbClusters}, + {"SlurmdbWckeys", client.SlurmdbWckeys}, + {"SlurmdbAccounts", client.SlurmdbAccounts}, + {"SlurmdbJobs", client.SlurmdbJobs}, } for _, s := range services {