View source
<?php
namespace Drupal\elasticsearch_connector\Controller;
use Drupal\Core\Config\Entity\ConfigEntityListBuilder;
use Drupal\Core\Entity\EntityInterface;
use Drupal\Core\Entity\EntityStorageInterface;
use Drupal\Core\Entity\EntityTypeInterface;
use Drupal\Core\Url;
use Drupal\elasticsearch_connector\ElasticSearch\ClientManagerInterface;
use Drupal\elasticsearch_connector\Entity\Cluster;
use Drupal\elasticsearch_connector\Entity\Index;
use Symfony\Component\DependencyInjection\ContainerInterface;
use Symfony\Component\HttpKernel\Exception\NotFoundHttpException;
class ClusterListBuilder extends ConfigEntityListBuilder {
protected $indexStorage;
private $clientManager;
public function __construct(EntityTypeInterface $entity_type, EntityStorageInterface $storage, EntityStorageInterface $index_storage, ClientManagerInterface $client_manager) {
parent::__construct($entity_type, $storage);
$this->indexStorage = $index_storage;
$this->clientManager = $client_manager;
}
public static function createInstance(ContainerInterface $container, EntityTypeInterface $entity_type) {
return new static($entity_type, $container
->get('entity.manager')
->getStorage($entity_type
->id()), $container
->get('entity.manager')
->getStorage('elasticsearch_index'), $container
->get('elasticsearch_connector.client_manager'));
}
public function group() {
$clusters = $this->storage
->loadMultiple();
$indices = $this->indexStorage
->loadMultiple();
$cluster_groups = array();
$lone_indices = array();
foreach ($clusters as $cluster) {
$cluster_group = array(
'cluster.' . $cluster->cluster_id => $cluster,
);
foreach ($indices as $index) {
if ($index->server == $cluster->cluster_id) {
$cluster_group['index.' . $index->index_id] = $index;
}
elseif ($index->server == NULL) {
$lone_indices['index.' . $index->index_id] = $index;
}
}
$cluster_groups['cluster.' . $cluster->cluster_id] = $cluster_group;
}
return array(
'clusters' => $cluster_groups,
'lone_indexes' => $lone_indices,
);
}
public function buildHeader() {
return array(
'type' => $this
->t('Type'),
'title' => $this
->t('Name'),
'machine_name' => $this
->t('Machine Name'),
'status' => $this
->t('Status'),
'cluster_status' => $this
->t('Cluster Status'),
) + parent::buildHeader();
}
public function buildRow(EntityInterface $entity) {
if ($entity instanceof Cluster) {
$client_connector = $this->clientManager
->getClientForCluster($entity);
}
elseif ($entity instanceof Index) {
$cluster = Cluster::load($entity->server);
$client_connector = $this->clientManager
->getClientForCluster($cluster);
}
else {
throw new NotFoundHttpException();
}
$row = parent::buildRow($entity);
$result = array();
$status = NULL;
if (isset($entity->cluster_id)) {
$cluster = Cluster::load($entity->cluster_id);
if ($client_connector
->isClusterOk()) {
$cluster_health = $client_connector
->cluster()
->health();
$version_number = $client_connector
->getServerVersion();
$status = $cluster_health['status'];
}
else {
$status = $this
->t('Not available');
$version_number = $this
->t('Unknown version');
}
$result = array(
'data' => array(
'type' => array(
'data' => $this
->t('Cluster'),
),
'title' => array(
'data' => array(
'#type' => 'link',
'#title' => $entity
->label() . ' (' . $version_number . ')',
'#url' => new Url('entity.elasticsearch_cluster.edit_form', array(
'elasticsearch_cluster' => $entity
->id(),
)),
),
),
'machine_name' => array(
'data' => $entity
->id(),
),
'status' => array(
'data' => $cluster->status ? 'Active' : 'Inactive',
),
'clusterStatus' => array(
'data' => $status,
),
'operations' => $row['operations'],
),
'title' => $this
->t('Machine name: @name', array(
'@name' => $entity
->id(),
)),
);
}
elseif (isset($entity->index_id)) {
$result = array(
'data' => array(
'type' => array(
'data' => $this
->t('Index'),
'class' => array(
'es-list-index',
),
),
'title' => array(
'data' => $entity
->label(),
),
'machine_name' => array(
'data' => $entity
->id(),
),
'status' => array(
'data' => '',
),
'clusterStatus' => array(
'data' => '-',
),
'operations' => $row['operations'],
),
'title' => $this
->t('Machine name: @name', array(
'@name' => $entity
->id(),
)),
);
}
return $result;
}
public function getDefaultOperations(EntityInterface $entity) {
$operations = array();
if (isset($entity->cluster_id)) {
$operations['info'] = array(
'title' => $this
->t('Info'),
'weight' => 19,
'url' => new Url('entity.elasticsearch_cluster.canonical', array(
'elasticsearch_cluster' => $entity
->id(),
)),
);
$operations['edit'] = array(
'title' => $this
->t('Edit'),
'weight' => 20,
'url' => new Url('entity.elasticsearch_cluster.edit_form', array(
'elasticsearch_cluster' => $entity
->id(),
)),
);
$operations['delete'] = array(
'title' => $this
->t('Delete'),
'weight' => 21,
'url' => new Url('entity.elasticsearch_cluster.delete_form', array(
'elasticsearch_cluster' => $entity
->id(),
)),
);
}
elseif (isset($entity->index_id)) {
$operations['delete'] = array(
'title' => $this
->t('Delete'),
'weight' => 20,
'url' => new Url('entity.elasticsearch_index.delete_form', array(
'elasticsearch_index' => $entity
->id(),
)),
);
}
return $operations;
}
public function render() {
$entity_groups = $this
->group();
$rows = array();
foreach ($entity_groups['clusters'] as $cluster_group) {
foreach ($cluster_group as $entity) {
$rows[$entity
->id()] = $this
->buildRow($entity);
}
}
$list['#type'] = 'container';
$list['#attached']['library'][] = 'elasticsearch_connector/drupal.elasticsearch_connector.ec_index';
$list['clusters'] = array(
'#type' => 'table',
'#header' => $this
->buildHeader(),
'#rows' => $rows,
'#empty' => $this
->t('No clusters available. <a href="@link">Add new cluster</a>.', array(
'@link' => \Drupal::urlGenerator()
->generate('entity.elasticsearch_cluster.add_form'),
)),
);
return $list;
}
}