Skip to content

Commit b6af81a

Browse files
committed
feat: implement InMemoryCatalog as a subclass of SqlCatalog
1 parent 052a9cd commit b6af81a

File tree

3 files changed

+128
-271
lines changed

3 files changed

+128
-271
lines changed

pyiceberg/catalog/memory.py

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
# Licensed to the Apache Software Foundation (ASF) under one
2+
# or more contributor license agreements. See the NOTICE file
3+
# distributed with this work for additional information
4+
# regarding copyright ownership. The ASF licenses this file
5+
# to you under the Apache License, Version 2.0 (the
6+
# "License"); you may not use this file except in compliance
7+
# with the License. You may obtain a copy of the License at
8+
#
9+
# http://www.apache.org/licenses/LICENSE-2.0
10+
#
11+
# Unless required by applicable law or agreed to in writing,
12+
# software distributed under the License is distributed on an
13+
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14+
# KIND, either express or implied. See the License for the
15+
# specific language governing permissions and limitations
16+
# under the License.
17+
18+
from pyiceberg.catalog.sql import SqlCatalog
19+
20+
21+
class InMemoryCatalog(SqlCatalog):
22+
"""
23+
An in-memory catalog implementation that uses SqlCatalog with SQLite in-memory database.
24+
25+
This is useful for test, demo, and playground but not in production as data is not persisted.
26+
"""
27+
28+
def __init__(self, name: str, warehouse: str = "file:///tmp/warehouse", **kwargs: str) -> None:
29+
self._warehouse_location = warehouse
30+
if "uri" not in kwargs:
31+
kwargs["uri"] = "sqlite:///:memory:"
32+
super().__init__(name=name, warehouse=warehouse, **kwargs)

0 commit comments

Comments
 (0)