/pgrocks-fdw

Bring RocksDB to PostgreSQL as an extension. It is the first foreign data wrapper (FDW) that introduces LSM-tree into PostgreSQL. The underneath storage engine can be RocksDB. The FDW also serves for VidarDB engine, a versatile storage engine for various workloads. See the link for more info about VidarDB engine.

Primary LanguageCApache License 2.0Apache-2.0

pgrocks-fdw

Build Status

This PostgreSQL extension implements a Foreign Data Wrapper (FDW) for RocksDB. This repo has been listed in PostgreSQL wiki. We are also building an extension to supercharge PostgreSQL analytics. It's not product ready yet, but if you are interested, please contact us at info@vidardb.com to gain access.

RocksDB is a high performance key-value store based on a log-structured merge-tree (LSM tree). RocksDB can efficiently use many CPU cores and fast storage. This is the first foreign data wrapper that connects a LSM-tree-based storage engine to PostgreSQL. Because RocksDB is an embeddable key-value store, you do not need to run another server to use this extension.

This extension can also be used for other systems that have RocksDB-like APIs, but please check the compatibility before you use this extension for other systems.

This extension is developed and maintained by the VidarDB team. Currently only works for PG13. Feel free to report bugs or issues via Github.

Building

We test this foreign data wrapper on Ubuntu Server 20.04 using PostgreSQL-13 together with RocksDB-6.11.4 (built with GCC-9.3.0).

  • Install PostgreSQL and the dev library which is required by extensions:

    # add the repository
    sudo tee /etc/apt/sources.list.d/pgdg.list << END
    deb http://apt.postgresql.org/pub/repos/apt/ bionic-pgdg main
    END
    
    # get the signing key and import it
    wget https://www.postgresql.org/media/keys/ACCC4CF8.asc
    sudo apt-key add ACCC4CF8.asc
    
    # fetch the metadata from the new repo
    sudo apt-get update
    
    # install postgresql and the dev library
    sudo apt-get install postgresql-13
    sudo apt-get install postgresql-server-dev-13
  • Install RocksDB from source code:

    git clone -b v6.11.4 https://github.com/facebook/rocksdb.git
    
    cd rocksdb
    
    sudo DEBUG_LEVEL=0 make shared_lib install-shared
    
    sudo sh -c "echo /usr/local/lib >> /etc/ld.so.conf"  
    
    sudo ldconfig
  • Build this foreign data wrapper:

    git clone https://github.com/vidardb/pgrocks-fdw.git
    
    cd pgrocks-fdw 
    
    make
    
    sudo make install

    To build the foreign data wrapper for VidarDB [ version <= 1.0.0 ], add flag VIDARDB=true to the above make command.

  • Before using this foreign data wrapper, we need to add it to shared_preload_libraries in the postgresql.conf:

    sudo bash -c 'echo "shared_preload_libraries = 'kv_fdw'" >> /etc/postgresql/13/main/postgresql.conf'

    and restart PostgreSQL:

    sudo service postgresql restart
  • When uninstall this extension, first issue the following commands, and then delete the data by locating PostgreSQL data folder via show data_directory; in PostgreSQL terminal.

    cd pgrocks-fdw 
    
    sudo make uninstall

Limitations

  • The first attribute in the table definition must be the primary key. Composite primary key can be achieved via creating a new data type CREATE TYPE ... AS.

  • Currently no rollback, abort.

  • Currently once the table is created, cannot drop or add columns.

  • Do not support secondary index.

Usage

This extension does not have any parameter. After creating the extension and corresponding server, you can use RocksDB as a foreign storage engine for your PostgreSQL.

A simple example is as follows (you can run 'sudo -u postgres psql -U postgres' to connect the local postgresql server):

    CREATE DATABASE example;  
    \c example  

    CREATE EXTENSION kv_fdw;  
    CREATE SERVER kv_server FOREIGN DATA WRAPPER kv_fdw;  

    CREATE FOREIGN TABLE student(id INTEGER, name TEXT) SERVER kv_server;  

    INSERT INTO student VALUES(20757123, 'Rafferty');  
    SELECT * FROM student;  

    INSERT INTO student VALUES(20767234, 'Jones');  
    SELECT * FROM student;  

    DELETE FROM student WHERE name='Jones';  
    SELECT * FROM student;  

    UPDATE student SET name='Tom' WHERE id=20757123;  
    SELECT * FROM student;  

    DROP FOREIGN TABLE student;  

    DROP SERVER kv_server;  
    DROP EXTENSION kv_fdw;  
  
    \c postgres  
    DROP DATABASE example;  

Testing

We have tested certain typical SQL statements and will add more test cases later. The test scripts are in the sql folder which are recommended to be placed in a non-root directory. The corresponding results can be found in the expected folder. You can run the tests in the following way:

    sudo service postgresql restart  

    cd pgrocks-fdw 

    sudo -u postgres psql -U postgres -a -f sql/create.sql 

    sudo -u postgres psql -U postgres -d kvtest -a -f sql/test.sql 

    sudo -u postgres psql -U postgres -d kvtest -a -f sql/clear.sql  

Debug

If you want to debug the source code, you may need to start PostgreSQL in the debug mode:

    sudo service postgresql stop  

    sudo -u postgres /usr/lib/postgresql/13/bin/postgres -d 0 -D /var/lib/postgresql/13/main -c config_file=/etc/postgresql/13/main/postgresql.conf

Docker

We can also run PostgreSQL with RocksDB in Docker container and you can refer to here.