SQLite:如何将memvfs扩展嵌入到合并?

时间:2018-11-23 14:21:00

标签: sqlite vfs

我需要在内存缓冲区中加载/保存SQLite数据库。为此,我想将memvfs扩展嵌入sqlite3代码中,并将其完全编译为sqlite3.dll。

怎么做?

更新1:

我想使用memvfs作为临时内存缓冲区。我的程序将数据从net加载到缓冲区,连接到该内存缓冲区,然后将数据还原到空的内存db中。我认为将memvfs包含在sqlite合并中会提高性能。

Update2:

如果要使用memvfs扩展,请注意源代码中自述注释中的错误。使用“ PRAGMA journal_mode = 关闭”代替“ journal_mode = NONE”

Update3:

memvfs.c中的另一个错误-URI中的maxsz参数使用'max'代替'maxsz'。 sqlite开发人员精心设置了耙:(

1 个答案:

答案 0 :(得分:1)

测试程序以演示如何使用memvfs:

#include <fcntl.h>
#include <sqlite3.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>

int main(void) {
  sqlite3 *db;
  char *err;

  // Open an in-memory database to use as a handle for loading the memvfs extension
  if (sqlite3_open(":memory:", &db) != SQLITE_OK) {
    fprintf(stderr, "open :memory: %s\n", sqlite3_errmsg(db));
    return EXIT_FAILURE;
  }

  sqlite3_enable_load_extension(db, 1);
  if (sqlite3_load_extension(db, "./memvfs", NULL, &err) != SQLITE_OK) {
    fprintf(stderr, "load extension: %s\n", err);
    return EXIT_FAILURE;
  }

  // Done with this database
  sqlite3_close(db);

  // Read the real database into memory
  int fd = open("foo.db", O_RDONLY);
  if (fd < 0) {
    perror("open");
    return EXIT_FAILURE;
  }
  struct stat s;
  if (fstat(fd, &s) < 0) {
    perror("fstat");
    return EXIT_FAILURE;
  }
  void *memdb = sqlite3_malloc64(s.st_size);
  if (read(fd, memdb, s.st_size) != s.st_size) {
    perror("read");
    return EXIT_FAILURE;
  }
  close(fd);

  // And open that memory with memvfs now that it holds a valid database
  char *memuri = sqlite3_mprintf("file:whatever?ptr=0x%p&sz=%lld&freeonclose=1",
                                 memdb, (long long)s.st_size);
  printf("Trying to open '%s'\n", memuri);
  if (sqlite3_open_v2(memuri, &db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_URI,
                      "memvfs") != SQLITE_OK) {
    fprintf(stderr, "open memvfs: %s\n", sqlite3_errmsg(db));
    return EXIT_FAILURE;
  }
  sqlite3_free(memuri);

  // Try querying the database to show it works.
  sqlite3_stmt *stmt;
  if (sqlite3_prepare_v2(db, "SELECT b FROM test", -1, &stmt, NULL) !=
      SQLITE_OK) {
    fprintf(stderr, "prepare: %s\n", sqlite3_errmsg(db));
    sqlite3_close(db);
    return EXIT_FAILURE;
  }

  for (int rc = sqlite3_step(stmt); rc == SQLITE_ROW; rc = sqlite3_step(stmt)) {
    printf("%d\n", sqlite3_column_int(stmt, 0));
  }

  sqlite3_finalize(stmt);
  sqlite3_close(db);
  return 0;
}

用法:

# Create a test database to use with memvfs
$ sqlite3 foo.db
sqlite> CREATE TABLE test(b INTEGER);
sqlite> INSERT INTO test VALUES (1), (2);
sqlite> .quit
# Compile the memvfs module and test program
$ gcc -O -fPIC -shared -o memvfs.so memvfs.c
$ gcc -O -Wall -Wextra testmem.c -lsqlite3
# And run it.
$ ./a.out
Trying to open 'file:whatever?ptr=0x56653FE2B940&sz=8192&freeonclose=1'
1
2

相同的工作流程,如果直接将其编译到程序中而不是使用可加载模块;您只需使用正确的参数调用sqlite3_memvfs_init()即可,而不必使用sqlite3_load_extension()