lens-ui/app/stores/AdhocQueryStore.js

/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/

import assign from 'object-assign';
import { EventEmitter } from 'events';

import AppDispatcher from '../dispatcher/AppDispatcher';
import AdhocQueryConstants from '../constants/AdhocQueryConstants';
import Config from 'config.json';

var CHANGE_EVENT = 'change';
var adhocDetails = {

  queryHandle: null,
  queries: {},
  queryResults: {}, // map with handle being the key
  dbName: Config.dbName
};

function receiveQueryHandle (payload) {
  let id = payload.queryHandle.getElementsByTagName('handleId')[0].textContent;
  adhocDetails.queryHandle = id;
}

function receiveQueries (payload) {
  let queries = payload.queries;
  let queryObjects = {};

  queries.forEach((query) => {
    queryObjects[query.queryHandle.handleId] = query;
  });

  adhocDetails.queries = queryObjects;
}

function receiveQuery (payload) {
  let query = payload.query;
  adhocDetails.queries[query.queryHandle.handleId] = query;
}

function receiveQueryResult (payload) {
  let queryResult = {};
  queryResult.type = payload && payload.type;

  if (queryResult.type === 'INMEMORY') {
    let resultRows = payload.queryResult && payload.queryResult.rows &&
      payload.queryResult.rows.rows || [];
    let columns = payload.columns && payload.columns.columns &&
      payload.columns.columns.columns;

    adhocDetails.queryResults[payload.handle] = {};
    adhocDetails.queryResults[payload.handle].results = resultRows;
    adhocDetails.queryResults[payload.handle].columns = columns;
  } else {
    // persistent
    adhocDetails.queryResults[payload.handle] = {};
    adhocDetails.queryResults[payload.handle].downloadURL = payload.downloadURL;
  }
}

let AdhocQueryStore = assign({}, EventEmitter.prototype, {
  getQueries () {
    return adhocDetails.queries;
  },

  getQueryResult (handle) {
    return adhocDetails.queryResults[handle];
  },

  // always returns the last-run-query's handle
  getQueryHandle () {
    let handle = adhocDetails.queryHandle;
    adhocDetails.queryHandle = null;
    return handle;
  },

  emitChange (hash) {
    this.emit(CHANGE_EVENT, hash);
  },

  addChangeListener (callback) {
    this.on(CHANGE_EVENT, callback);
  },

  removeChangeListener (callback) {
    this.removeListener(CHANGE_EVENT, callback);
  }
});

AppDispatcher.register((action) => {
  switch (action.actionType) {

    case AdhocQueryConstants.RECEIVE_QUERY_HANDLE:
      receiveQueryHandle(action.payload);
      AdhocQueryStore.emitChange();
      break;

    case AdhocQueryConstants.RECEIVE_QUERIES:
      receiveQueries(action.payload);
      AdhocQueryStore.emitChange();
      break;

    case AdhocQueryConstants.RECEIVE_QUERY_RESULT:
      receiveQueryResult(action.payload);
      AdhocQueryStore.emitChange();
      break;

    case AdhocQueryConstants.RECEIVE_QUERY:
      receiveQuery(action.payload);
      AdhocQueryStore.emitChange();
      break;

    case AdhocQueryConstants.RECEIVE_QUERY_HANDLE_FAILED:
      AdhocQueryStore.emitChange(action.payload);
      break;
  }
});

export default AdhocQueryStore;