steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

cleanup

This commit is contained in:
Shreyas Goenka 2025-01-02 23:56:58 +05:30
parent ac37ca0d98
commit f4623ebbb9
No known key found for this signature in database
GPG Key ID: 92A07DF49CCB0622
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/filer/dbfs_client.go
View File

@ -69,11 +69,9 @@ func (info dbfsFileInfo) Sys() any {
} }
// Interface to allow mocking of the Databricks API client. // Interface to allow mocking of the Databricks API client.
//
//nolint:gofumpt
type databricksClient interface { type databricksClient interface {
Do(ctx context.Context, method, path string, headers map[string]string, Do(ctx context.Context, method, path string, headers map[string]string,
requestBody any, responseBody any, visitors ...func(*http.Request) error) error requestBody, responseBody any, visitors ...func(*http.Request) error) error
} }
// DbfsClient implements the [Filer] interface for the DBFS backend. // DbfsClient implements the [Filer] interface for the DBFS backend.
@ -102,7 +100,7 @@ func NewDbfsClient(w *databricks.WorkspaceClient, root string) (Filer, error) {
// The PUT API for DBFS requires setting the content length header beforehand in the HTTP // The PUT API for DBFS requires setting the content length header beforehand in the HTTP
// request. // request.
func putContentLength(path, overwriteField string, file *os.File) (int64, error) { func contentLength(path, overwriteField string, file *os.File) (int64, error) {
buf := &bytes.Buffer{} buf := &bytes.Buffer{}
writer := multipart.NewWriter(buf) writer := multipart.NewWriter(buf)
err := writer.WriteField("path", path) err := writer.WriteField("path", path)
@ -132,7 +130,7 @@ func putContentLength(path, overwriteField string, file *os.File) (int64, error)
func contentLengthVisitor(path, overwriteField string, file *os.File) func(*http.Request) error { func contentLengthVisitor(path, overwriteField string, file *os.File) func(*http.Request) error {
return func(r *http.Request) error { return func(r *http.Request) error {
cl, err := putContentLength(path, overwriteField, file) cl, err := contentLength(path, overwriteField, file)
if err != nil { if err != nil {
return fmt.Errorf("failed to calculate content length: %w", err) return fmt.Errorf("failed to calculate content length: %w", err)
} }