feat: 添加 Node 领域类型和 NodesService

包含 Node、UpdateNodeMsg、AcctGatherEnergy、ExtSensorsData、PowerMgmtData 等类型。NodesService 提供 GetNodes、GetNode、PostNode、DeleteNode 4 个方法，并定义 NodeFlag* 常量。 Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
2026-04-08 18:29:26 +08:00
parent 2c84930983
commit f9234b2167
4 changed files with 706 additions and 0 deletions
--- a/internal/slurm/slurm_nodes.go
+++ b/internal/slurm/slurm_nodes.go
@@ -0,0 +1,126 @@
+package slurm
+
+import (
+	"context"
+	"fmt"
+	"net/url"
+)
+
+// Node query flags for GetNodesOptions.Flags and GetNodeOptions.Flags.
+const (
+	NodeFlagAll        = "ALL"
+	NodeFlagDetail     = "DETAIL"
+	NodeFlagMixed      = "MIXED"
+	NodeFlagLocal      = "LOCAL"
+	NodeFlagSibling    = "SIBLING"
+	NodeFlagFederation = "FEDERATION"
+	NodeFlagFuture     = "FUTURE"
+)
+
+// GetNodesOptions specifies optional parameters for GetNodes.
+type GetNodesOptions struct {
+	UpdateTime *string `url:"update_time,omitempty"`
+	Flags      *string `url:"flags,omitempty"` // Use NodeFlag* constants (e.g. NodeFlagDetail)
+}
+
+// GetNodeOptions specifies optional parameters for GetNode.
+type GetNodeOptions struct {
+	UpdateTime *string `url:"update_time,omitempty"`
+	Flags      *string `url:"flags,omitempty"` // Use NodeFlag* constants (e.g. NodeFlagDetail)
+}
+
+// GetNodes lists all nodes.
+func (s *NodesService) GetNodes(ctx context.Context, opts *GetNodesOptions) (*OpenapiNodesResp, *Response, error) {
+	path := "slurm/v0.0.40/nodes"
+	req, err := s.client.NewRequest("GET", path, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	if opts != nil {
+		u, parseErr := url.Parse(req.URL.String())
+		if parseErr != nil {
+			return nil, nil, parseErr
+		}
+		q := u.Query()
+		if opts.UpdateTime != nil {
+			q.Set("update_time", *opts.UpdateTime)
+		}
+		if opts.Flags != nil {
+			q.Set("flags", *opts.Flags)
+		}
+		u.RawQuery = q.Encode()
+		req.URL = u
+	}
+
+	var result OpenapiNodesResp
+	resp, err := s.client.Do(ctx, req, &result)
+	if err != nil {
+		return nil, resp, err
+	}
+	return &result, resp, nil
+}
+
+// GetNode gets a single node by name.
+func (s *NodesService) GetNode(ctx context.Context, nodeName string, opts *GetNodeOptions) (*OpenapiNodesResp, *Response, error) {
+	path := fmt.Sprintf("slurm/v0.0.40/node/%s", nodeName)
+	req, err := s.client.NewRequest("GET", path, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	if opts != nil {
+		u, parseErr := url.Parse(req.URL.String())
+		if parseErr != nil {
+			return nil, nil, parseErr
+		}
+		q := u.Query()
+		if opts.UpdateTime != nil {
+			q.Set("update_time", *opts.UpdateTime)
+		}
+		if opts.Flags != nil {
+			q.Set("flags", *opts.Flags)
+		}
+		u.RawQuery = q.Encode()
+		req.URL = u
+	}
+
+	var result OpenapiNodesResp
+	resp, err := s.client.Do(ctx, req, &result)
+	if err != nil {
+		return nil, resp, err
+	}
+	return &result, resp, nil
+}
+
+// PostNode updates a node.
+func (s *NodesService) PostNode(ctx context.Context, nodeName string, update *UpdateNodeMsg) (*OpenapiResp, *Response, error) {
+	path := fmt.Sprintf("slurm/v0.0.40/node/%s", nodeName)
+	req, err := s.client.NewRequest("POST", path, update)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	var result OpenapiResp
+	resp, err := s.client.Do(ctx, req, &result)
+	if err != nil {
+		return nil, resp, err
+	}
+	return &result, resp, nil
+}
+
+// DeleteNode deletes a node.
+func (s *NodesService) DeleteNode(ctx context.Context, nodeName string) (*OpenapiResp, *Response, error) {
+	path := fmt.Sprintf("slurm/v0.0.40/node/%s", nodeName)
+	req, err := s.client.NewRequest("DELETE", path, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	var result OpenapiResp
+	resp, err := s.client.Do(ctx, req, &result)
+	if err != nil {
+		return nil, resp, err
+	}
+	return &result, resp, nil
+}