com.databricks.sdk.service.files.FilesImpl Maven / Gradle / Ivy

Go to download
// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT.
package com.databricks.sdk.service.files;

import com.databricks.sdk.core.ApiClient;
import com.databricks.sdk.support.Generated;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

/** Package-local implementation of Files */
@Generated
class FilesImpl implements FilesService {
  private final ApiClient apiClient;

  public FilesImpl(ApiClient apiClient) {
    this.apiClient = apiClient;
  }

  @Override
  public void delete(DeleteFileRequest request) {
    String path = String.format("/api/2.0/fs/files/%s", request.getFilePath());
    Map headers = new HashMap<>();
    apiClient.DELETE(path, request, Void.class, headers);
  }

  @Override
  public DownloadResponse download(DownloadRequest request) {
    String path = String.format("/api/2.0/fs/files/%s", request.getFilePath());
    Map headers = new HashMap<>();
    headers.put("Accept", "application/octet-stream");
    InputStream response = apiClient.GET(path, request, InputStream.class, headers);
    return new DownloadResponse().setContents(response);
  }

  @Override
  public FileInfo getStatus(GetStatusRequest request) {
    String path = "/api/2.0/fs/get-status";
    Map headers = new HashMap<>();
    headers.put("Accept", "application/json");
    return apiClient.GET(path, request, FileInfo.class, headers);
  }

  @Override
  public void upload(UploadRequest request) {
    String path = String.format("/api/2.0/fs/files/%s", request.getFilePath());
    Map headers = new HashMap<>();
    headers.put("Content-Type", "application/octet-stream");
    apiClient.PUT(path, request.getContents(), Void.class, headers);
  }
}