Add dummy jobs client

Author: renaudhartert-db

databricks/internal/internal.go

@@ -14,7 +14,6 @@ type ClientOptions struct {
 	Credentials auth.Credentials
 	Timeout     time.Duration
 	Logger      *slog.Logger
-	Debug       bool
 }
 
 // Initialize initializes the client options and validates its configuration.

databricks/options/unstable/unstable.go

@@ -0,0 +1,33 @@
+// Package unstable contains functionalities that need to be shared
+// between SDK packages but are not considered part of the public API
+// of the Databricks SDKs.
+//
+// IMPORTANT: Clients should not directly depend on functionalities in
+// this package. All functionalities in this package are considered as
+// UNSTABLE and may change without notice.
+package unstable
+
+import (
+	"log/slog"
+
+	"github.com/databricks/sdk-go/databricks/internal"
+	"github.com/databricks/sdk-go/databricks/options"
+)
+
+// ResolvedOptions contains the resolved configuration values from ClientOptions.
+type ResolvedOptions struct {
+	Logger *slog.Logger
+}
+
+// Resolve applies all ClientOptions and returns the resolved configuration.
+func Resolve(opts ...options.ClientOption) (ResolvedOptions, error) {
+	o := &internal.ClientOptions{}
+	for _, opt := range opts {
+		if err := opt(o); err != nil {
+			return ResolvedOptions{}, err
+		}
+	}
+	return ResolvedOptions{
+		Logger: o.Logger,
+	}, nil
+}

go.work

@@ -1,6 +1,7 @@
 go 1.25.4
 
 use (
-    ./auth
-    ./databricks
+	./auth
+	./databricks
+	./jobs
 )

jobs/go.mod

@@ -0,0 +1,3 @@
+module github.com/databricks/sdk-go/jobs
+
+go 1.25.4

jobs/v2/genhelper.go

@@ -0,0 +1,61 @@
+package jobs
+
+import (
+	"io"
+	"log/slog"
+	"net/http"
+
+	"github.com/databricks/sdk-go/databricks/apierr"
+)
+
+type httpCallOptions struct {
+	req    *http.Request
+	client *http.Client
+	logger *slog.Logger
+}
+
+// lazyRequest implements slog.LogValuer for lazy evaluation of request
+// logging.
+type lazyRequest struct{ req *http.Request }
+
+func (r lazyRequest) LogValue() slog.Value {
+	return slog.GroupValue(
+		slog.String("method", r.req.Method),
+		slog.String("url", r.req.URL.String()),
+	)
+}
+
+// lazyResponse implements slog.LogValuer for lazy evaluation of response
+// logging.
+type lazyResponse struct {
+	resp *http.Response
+	body []byte
+}
+
+func (r lazyResponse) LogValue() slog.Value {
+	return slog.GroupValue(
+		slog.Int("status", r.resp.StatusCode),
+		slog.String("body", string(r.body)),
+	)
+}
+
+// executeHTTPCall executes an HTTP call and returns the response body or
+// and error. This function takes care of logging the request and response.
+func executeHTTPCall(opts httpCallOptions) ([]byte, error) {
+	opts.logger.Debug("HTTP request", "request", lazyRequest{opts.req})
+	resp, err := opts.client.Do(opts.req)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		opts.logger.Debug("HTTP response", "response", lazyResponse{resp, nil})
+		return nil, err
+	}
+	opts.logger.Debug("HTTP response", "response", lazyResponse{resp, body})
+	if err := apierr.FromHTTPError(resp.StatusCode, resp.Header, body); err != nil {
+		return nil, err
+	}
+	return body, nil
+}

jobs/v2/jobs.go

@@ -0,0 +1,112 @@
+package jobs
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"iter"
+	"log/slog"
+	"net/http"
+	"net/url"
+
+	"github.com/databricks/sdk-go/databricks/api"
+	"github.com/databricks/sdk-go/databricks/options"
+	"github.com/databricks/sdk-go/databricks/options/unstable"
+	"github.com/databricks/sdk-go/databricks/transport"
+)
+
+type Client struct {
+	httpClient *http.Client
+	logger     *slog.Logger
+	host       string
+}
+
+func NewClient(ctx context.Context, opts ...options.ClientOption) (*Client, error) {
+	resolved, err := unstable.Resolve(opts...)
+	if err != nil {
+		return nil, err
+	}
+	httpClient, err := transport.NewHTTPClient(ctx, opts...)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Client{
+		httpClient: httpClient,
+		logger:     resolved.Logger,
+	}, nil
+}
+
+func (c *Client) CreateJob(ctx context.Context, req *CreateJobRequest, opts ...api.Option) (*Job, error) {
+	body, err := json.Marshal(req)
+	if err != nil {
+		return nil, err
+	}
+
+	headers := http.Header{}
+	headers.Set("Content-Type", "application/json")
+
+	baseURL, err := url.Parse(c.host)
+	if err != nil {
+		return nil, err
+	}
+	queryParams := url.Values{}
+	baseURL.RawQuery = queryParams.Encode()
+
+	resp := &Job{}
+	call := func(ctx context.Context) error {
+		httpReq, err := http.NewRequest("POST", baseURL.String(), bytes.NewBuffer(body))
+		if err != nil {
+			return err
+		}
+		httpReq = httpReq.WithContext(ctx)
+		httpReq.Header = headers
+
+		respBody, err := executeHTTPCall(httpCallOptions{
+			req:    httpReq,
+			client: c.httpClient,
+			logger: c.logger,
+		})
+		if err != nil {
+			return err
+		}
+		if err := json.Unmarshal(respBody, resp); err != nil {
+			return err
+		}
+		return nil
+	}
+
+	if err := api.Execute(ctx, call, opts...); err != nil {
+		return nil, err
+	}
+	return resp, nil
+}
+
+func ListJobs(ctx context.Context, req *ListJobsRequest, opts ...api.Option) (*ListJobsResponse, error) {
+	return nil, nil
+}
+
+func ListJobsIter(ctx context.Context, req *ListJobsRequest, opts ...api.Option) iter.Seq2[*Job, error] {
+	return func(yield func(*Job, error) bool) {
+		pageReq := &ListJobsRequest{
+			PageSize:  req.PageSize,
+			PageToken: req.PageToken,
+		}
+		for {
+			resp, err := ListJobs(ctx, pageReq, opts...)
+			if err != nil {
+				yield(nil, err)
+				return
+			}
+			for i := range resp.Jobs {
+				if !yield(&resp.Jobs[i], nil) {
+					return
+				}
+			}
+			if resp.NextPageToken == "" {
+				return
+			}
+			pageReq.PageToken = resp.NextPageToken
+		}
+	}
+}

jobs/v2/model.go

@@ -0,0 +1,25 @@
+package jobs
+
+import "time"
+
+type Job struct {
+	ID          string    `json:"id"`
+	Name        string    `json:"name"`
+	Description string    `json:"description"`
+	CreatedAt   time.Time `json:"created_at"`
+	UpdatedAt   time.Time `json:"updated_at"`
+}
+
+type CreateJobRequest struct {
+	Job Job `json:"job"`
+}
+
+type ListJobsRequest struct {
+	PageSize  int    `json:"page_size"`
+	PageToken string `json:"page_token"`
+}
+
+type ListJobsResponse struct {
+	Jobs          []Job  `json:"jobs"`
+	NextPageToken string `json:"next_page_token"`
+}