class Multiwoven::Integrations::Source::DatabricksModel::Client

def check_connection(connection_config)
  connection_config = connection_config.with_indifferent_access
  url = build_url(DATABRICKS_HEALTH_URL, connection_config)
  response = Multiwoven::Integrations::Core::HttpClient.request(
    url,
    HTTP_GET,
    headers: auth_headers(connection_config[:token])
  )
  if success?(response)
    success_status
  else
    failure_status(nil)
  end
rescue StandardError => e
  ConnectionStatus.new(status: ConnectionStatusType["failed"], message: e.message).to_multiwoven_message
end