diff options
Diffstat (limited to 'src/storage')
-rw-r--r-- | src/storage/flexible_ngram.h | 16 | ||||
-rw-r--r-- | src/storage/ngram.h | 6 |
2 files changed, 12 insertions, 10 deletions
diff --git a/src/storage/flexible_ngram.h b/src/storage/flexible_ngram.h index 6f3c778..e5a086b 100644 --- a/src/storage/flexible_ngram.h +++ b/src/storage/flexible_ngram.h @@ -275,18 +275,26 @@ public: } /* attach berkeley db on filesystem for training purpose. */ - bool attach(const char * dbfile){ + bool attach(const char * dbfile, guint32 flags){ reset(); + u_int32_t db_flags = 0; + + if ( flags & ATTACH_READONLY ) + db_flags |= DB_RDONLY; + if ( flags & ATTACH_READWRITE ) + assert( !(flags & ATTACH_READONLY ) ); + if ( !dbfile ) return false; int ret = db_create(&m_db, NULL, 0); if ( ret != 0 ) assert(false); - ret = m_db->open(m_db, NULL, dbfile, NULL, DB_HASH, 0, 0644); - if ( ret != 0 ) { + ret = m_db->open(m_db, NULL, dbfile, NULL, DB_HASH, db_flags, 0644); + if ( ret != 0 && (flags & ATTACH_CREATE) ) { + db_flags |= DB_CREATE; /* Create database file here, and write the signature. */ - ret = m_db->open(m_db, NULL, dbfile, NULL, DB_HASH, DB_CREATE, 0644); + ret = m_db->open(m_db, NULL, dbfile, NULL, DB_HASH, db_flags, 0644); if ( ret != 0 ) return false; diff --git a/src/storage/ngram.h b/src/storage/ngram.h index 93e6ad7..a2bc7b6 100644 --- a/src/storage/ngram.h +++ b/src/storage/ngram.h @@ -26,12 +26,6 @@ namespace pinyin{ -enum { - ATTACH_READONLY = 1, - ATTACH_READWRITE = 0x1 << 1, - ATTACH_CREATE = 0x1 << 2, -}; - class Bigram; /* Note: |