All files / src/views/home-view/datasources-card datasources-card.tsx

70.27% Statements 26/37
40% Branches 4/10
62.5% Functions 5/8
67.86% Lines 19/28

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102                                    2x 2x 2x   2x   2x                       2x             1x 1x         1x                               1x 1x                 2x 1x   1x     2x       2x 2x 2x                         2x  
/*
 * 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 { IconNames } from '@blueprintjs/icons';
import axios from 'axios';
import React from 'react';
 
import { pluralIfNeeded, queryDruidSql, QueryManager } from '../../../utils';
import { Capabilities } from '../../../utils/capabilities';
import { HomeViewCard } from '../home-view-card/home-view-card';
 
export interface DatasourcesCardProps {
  capabilities: Capabilities;
}
 
export interface DatasourcesCardState {
  datasourceCountLoading: boolean;
  datasourceCount: number;
  datasourceCountError?: string;
}
 
export class DatasourcesCard extends React.PureComponent<
  DatasourcesCardProps,
  DatasourcesCardState
> {
  private datasourceQueryManager: QueryManager<Capabilities, any>;
 
  constructor(props: DatasourcesCardProps, context: any) {
    super(props, context);
    this.state = {
      datasourceCountLoading: false,
      datasourceCount: 0,
    };
 
    this.datasourceQueryManager = new QueryManager({
      processQuery: async capabilities => {
        let datasources: string[];
        if (capabilities.hasSql()) {
          datasources = await queryDruidSql({
            query: `SELECT datasource FROM sys.segments GROUP BY 1`,
          });
        } else if (capabilities.hasCoordinatorAccess()) {
          const datasourcesResp = await axios.get('/druid/coordinator/v1/datasources');
          datasources = datasourcesResp.data;
        } else {
          throw new Error(`must have SQL or coordinator access`);
        }
 
        return datasources.length;
      },
      onStateChange: ({ result, loading, error }) => {
        this.setState({
          datasourceCountLoading: loading,
          datasourceCount: result,
          datasourceCountError: error || undefined,
        });
      },
    });
  }
 
  componentDidMount(): void {
    const { capabilities } = this.props;
 
    this.datasourceQueryManager.runQuery(capabilities);
  }
 
  componentWillUnmount(): void {
    this.datasourceQueryManager.terminate();
  }
 
  render(): JSX.Element {
    const { datasourceCountLoading, datasourceCountError, datasourceCount } = this.state;
    return (
      <HomeViewCard
        className="datasources-card"
        href={'#datasources'}
        icon={IconNames.MULTI_SELECT}
        title={'Datasources'}
        loading={datasourceCountLoading}
        error={datasourceCountError}
      >
        {pluralIfNeeded(datasourceCount, 'datasource')}
      </HomeViewCard>
    );
  }
}