arangodb/js/apps/system/_admin/aardvark/APP/cluster.js

'use strict';

////////////////////////////////////////////////////////////////////////////////
/// DISCLAIMER
///
/// Copyright 2010-2013 triAGENS GmbH, Cologne, Germany
/// Copyright 2016 ArangoDB GmbH, Cologne, Germany
///
/// Licensed 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.
///
/// Copyright holder is ArangoDB GmbH, Cologne, Germany
///
/// @author Michael Hackstein
/// @author Alan Plum
////////////////////////////////////////////////////////////////////////////////

const _ = require('lodash');
const dd = require('dedent');
const httperr = require('http-errors');
const cluster = require('@arangodb/cluster');
const createRouter = require('@arangodb/foxx/router');
const internal = require('internal');
const plans = require('./repositories/plans.js');

const router = createRouter();
module.exports = router;

router.use((req, res, next) => {
  if (!internal.options()['server.disable-authentication'] && !req.user) {
    throw new httperr.Unauthorized();
  }
  next();
});

router.get('/amICoordinator', function(req, res) {
  res.json(cluster.isCoordinator());
})
.summary('Plan and start a new cluster')
.description('This will plan a new cluster with the information given in the body');

if (cluster.isCluster()) {
  // only make these functions available in cluster mode!
  var Communication = require("@arangodb/cluster/agency-communication");
  var comm = new Communication.Communication();
  var beats = comm.sync.Heartbeats();
  var diff = comm.diff.current;
  var servers = comm.current.DBServers();
  var dbs = comm.current.Databases();
  var coords = comm.current.Coordinators();

  router.get("/ClusterType", function(req, res) {
    // Not yet implemented
    res.json({
      type: "symmetricSetup"
    });
  })
  .summary('Get the type of the cluster')
  .description(dd`
     Returns a string containing the cluster type
     Possible anwers:
     - testSetup
     - symmetricalSetup
     - asymmetricalSetup
  `);

  router.get("/DBServers", function(req, res) {
    var resList = [],
      list = servers.getList(),
      diffList = diff.DBServers(),
      didBeat = beats.didBeat(),
      serving = beats.getServing();

    _.each(list, function(v, k) {
      v.name = k;
      resList.push(v);
      if (!_.contains(didBeat, k)) {
        v.status = "critical";
        return;
      }
      if (v.role === "primary" && !_.contains(serving, k)) {
        v.status = "warning";
        return;
      }
      v.status = "ok";
    });
    _.each(diffList.missing, function(v) {
      v.status = "missing";
      resList.push(v);
    });
    res.json(resList);
  })
  .summary('Get all DBServers')
  .description('Get a list of all running and expected DBServers within the cluster');

  router.get("/Coordinators", function(req, res) {
    var resList = [],
      list = coords.getList(),
      diffList = diff.Coordinators(),
      didBeat = beats.didBeat();

    _.each(list, function(v, k) {
      v.name = k;
      resList.push(v);
      if (!_.contains(didBeat, k)) {
        v.status = "critical";
        return;
      }
      v.status = "ok";
    });
    _.each(diffList.missing, function(v) {
      v.status = "missing";
      resList.push(v);
    });
    res.json(resList);
  });

  router.get("/Databases", function(req, res) {
    var list = dbs.getList();
    res.json(_.map(list, (name) => ({name})));
  });

  router.get("/:dbname/Collections", function(req, res) {
    var dbname = req.params("dbname"),
      selected = dbs.select(dbname);
    try {
      res.json(_.map(
        selected.getCollections(),
        (name) => ({name})
      ));
    } catch(e) {
      res.json([]);
    }
  });

  router.get("/:dbname/:colname/Shards", function(req, res) {
    var dbname = req.params("dbname");
    var colname = req.params("colname");
    var selected = dbs.select(dbname).collection(colname);
    res.json(selected.getShardsByServers());
  });

  router.get("/:dbname/:colname/Shards/:servername", function(req, res) {
    var dbname = req.params("dbname");
    var colname = req.params("colname");
    var servername = req.params("servername");
    var selected = dbs.select(dbname).collection(colname);
    res.json(_.map(
      selected.getShardsForServer(servername),
      (c) => ({id: c})
    ));
  });
}