kitedb 0.2.18

High-performance embedded graph database
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
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
103
104

//! Export and import operations for Python bindings

use pyo3::exceptions::PyRuntimeError;
use pyo3::prelude::*;

use crate::core::single_file::SingleFileDB as RustSingleFileDB;
use crate::export as ray_export;

use crate::pyo3_bindings::options::{ExportOptions, ExportResult, ImportOptions, ImportResult};

/// Trait for export/import operations
pub trait ExportImportOps {
  /// Export to JSON file
  fn export_to_json_impl(&self, path: String, options: ExportOptions) -> PyResult<ExportResult>;

  /// Export to JSONL file
  fn export_to_jsonl_impl(&self, path: String, options: ExportOptions) -> PyResult<ExportResult>;

  /// Import from JSON file
  fn import_from_json_impl(&self, path: String, options: ImportOptions) -> PyResult<ImportResult>;
}

// ============================================================================
// Single-file database operations
// ============================================================================

pub fn export_to_object_single(
  db: &RustSingleFileDB,
  options: ray_export::ExportOptions,
) -> PyResult<ray_export::ExportedDatabase> {
  ray_export::export_to_object_single(db, options)
    .map_err(|e| PyRuntimeError::new_err(e.to_string()))
}

pub fn export_to_json_single(
  db: &RustSingleFileDB,
  path: String,
  options: ExportOptions,
) -> PyResult<ExportResult> {
  let opts = options.to_rust();
  let data = ray_export::export_to_object_single(db, opts.clone())
    .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;

  let result = ray_export::export_to_json(&data, path, opts.pretty)
    .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;

  Ok(ExportResult {
    node_count: result.node_count as i64,
    edge_count: result.edge_count as i64,
  })
}

pub fn export_to_jsonl_single(
  db: &RustSingleFileDB,
  path: String,
  options: ExportOptions,
) -> PyResult<ExportResult> {
  let opts = options.to_rust();
  let data = ray_export::export_to_object_single(db, opts)
    .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;

  let result =
    ray_export::export_to_jsonl(&data, path).map_err(|e| PyRuntimeError::new_err(e.to_string()))?;

  Ok(ExportResult {
    node_count: result.node_count as i64,
    edge_count: result.edge_count as i64,
  })
}

pub fn import_from_object_single(
  db: &RustSingleFileDB,
  data: &ray_export::ExportedDatabase,
  options: ImportOptions,
) -> PyResult<ImportResult> {
  let opts = options.to_rust();
  let result = ray_export::import_from_object_single(db, data, opts)
    .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;

  Ok(ImportResult {
    node_count: result.node_count as i64,
    edge_count: result.edge_count as i64,
    skipped: result.skipped as i64,
  })
}

pub fn import_from_json_single(
  db: &RustSingleFileDB,
  path: String,
  options: ImportOptions,
) -> PyResult<ImportResult> {
  let opts = options.to_rust();
  let parsed =
    ray_export::import_from_json(path).map_err(|e| PyRuntimeError::new_err(e.to_string()))?;

  let result = ray_export::import_from_object_single(db, &parsed, opts)
    .map_err(|e| PyRuntimeError::new_err(e.to_string()))?;

  Ok(ImportResult {
    node_count: result.node_count as i64,
    edge_count: result.edge_count as i64,
    skipped: result.skipped as i64,
  })
}