lance-csr 0.3.1

Graph query engine for Lance datasets with Cypher support
Documentation 
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

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: Copyright The Lance Authors

//! Shared test fixtures and utilities for datafusion_planner tests

use crate::config::GraphConfig;
use crate::logical_plan::LogicalOperator;
use arrow_schema::{DataType, Field, Schema};
use lance_csr_catalog::{InMemoryCatalog, SimpleTableSource};
use std::sync::Arc;

pub fn person_schema() -> Arc<Schema> {
    Arc::new(Schema::new(vec![
        Field::new("id", DataType::Int64, false),
        Field::new("name", DataType::Utf8, true),
        Field::new("age", DataType::Int64, true),
    ]))
}

pub fn make_catalog() -> Arc<dyn lance_csr_catalog::GraphSourceCatalog> {
    let person_src = Arc::new(SimpleTableSource::new(person_schema()));
    let knows_schema = Arc::new(Schema::new(vec![
        Field::new("src_person_id", DataType::Int64, false),
        Field::new("dst_person_id", DataType::Int64, false),
    ]));
    let knows_src = Arc::new(SimpleTableSource::new(knows_schema));
    Arc::new(
        InMemoryCatalog::new()
            .with_node_source("Person", person_src)
            .with_relationship_source("KNOWS", knows_src),
    )
}

pub fn person_config() -> GraphConfig {
    GraphConfig::builder()
        .node_table("Person", "id")
        .build()
        .unwrap()
}

pub fn person_knows_config() -> GraphConfig {
    GraphConfig::builder()
        .node_table("Person", "id")
        .rel_table(
            "KNOWS",
            "Person",
            "Person",
            "src_person_id",
            "dst_person_id",
        )
        .build()
        .unwrap()
}

pub fn person_scan(variable: &str) -> LogicalOperator {
    LogicalOperator::ScanByLabel {
        variable: variable.to_string(),
        label: "Person".to_string(),
        properties: Default::default(),
    }
}