dremio 的InformationSchemaCatalog 服务一

InformationSchemaCatalog 服务实现了模仿information_schema的能力,让我们可以更好的在bi 工具以及其他系统使用

接口定义

/**
 * Facet of the catalog service that provides metadata with an information_schema like API.
 * <p>
 * The metadata provided may not satisfy the configured metadata policy requirement of various sources; the reads
 * are directly from the KVStore without an inline refresh of the metadata.
 */
interface InformationSchemaCatalog {
 
  /**
   * List catalogs that satisfy the given search query.
   *
   * @param searchQuery        search query, may be null
   * @return iterator of catalogs
   */
  Iterator<Catalog> listCatalogs(SearchQuery searchQuery);
 
  /**
   * List schemata that satisfy the given search query.
   *
   * @param searchQuery        search query, may be null
   * @return iterator of schemata
   */
  Iterator<Schema> listSchemata(SearchQuery searchQuery);
 
  /**
   * List tables (datasets) that satisfy the given search query.
   *
   * @param searchQuery        search query, may be null
   * @return iterator of tables
   */
  Iterator<Table> listTables(SearchQuery searchQuery);
 
  /**
   * List views that satisfy the given search query.
   *
   * @param searchQuery        search query, may be null
   * @return iterator of views
   */
  Iterator<View> listViews(SearchQuery searchQuery);
 
  /**
   * List columns of all datasets that satisfy the given search query.
   *
   * @param searchQuery        search query, may be null
   * @return iterator of columns
   */
  Iterator<TableSchema> listTableSchemata(SearchQuery searchQuery);
 
}

子类

 

 


从上图可以看出InformationSchemaCatalog 的直接实现是没有包含PrivilegeCatalog,catalog 的子类需要实现此接口,具体的实现InformationSchemaCatalogImpl
核心依赖NamespaceService(此服务比较重要,dremio 利用了key,value 存储解决数据状态管理)
比如数据表获取的

 
 @Override
  public Iterator<Table> listTables(SearchQuery searchQuery) {
    // 基于NamespaceService进行处理
    final Iterable<Map.Entry<NamespaceKey, NameSpaceContainer>> searchResults =
      userNamespace.find(getCondition(searchQuery));
 
    return StreamSupport.stream(searchResults.spliterator(), false)
      .filter(IS_NOT_INTERNAL)
      .filter(IS_DATASET)
      .map(input -> {
        final String sourceName = input.getKey().getRoot();
 
        final TableType tableType;
        if (input.getValue().getDataset().getType() == DatasetType.VIRTUAL_DATASET) {
          tableType = TableType.VIEW;
        } else if ("sys".equals(sourceName) || "INFORMATION_SCHEMA".equals(sourceName)) {
         // 包含了sys 以及INFORMATION_SCHEMA 的处理,sys 数据dremio 内部的,INFORMATION_SCHEMA 属于标准的INFORMATION_SCHEMA
          tableType = TableType.SYSTEM_TABLE;
        } else {
          tableType = TableType.TABLE;
        }
 
        return Table.newBuilder()
          .setCatalogName(DEFAULT_CATALOG_NAME)
          .setSchemaName(input.getKey().getParent().toUnescapedString())
          .setTableName(input.getKey().getName())
          .setTableType(tableType)
          .build();
      })
      .iterator();
  }

说明

InformationSchemaCatalog 的实现,完善了dremio 对于bi 以及jdbc 周边工具的集成,实现上不难,主要是利用了NamespaceService 服务

参考资料

sabot/kernel/src/main/java/com/dremio/exec/catalog/InformationSchemaCatalog.java
https://docs.dremio.com/software/sql-reference/information-schema/tables/

posted on 2022-07-17 12:06  荣锋亮  阅读(31)  评论(0编辑  收藏  举报

导航