mu: lib: update store API. update some dependents

Implement add_message / remove_message.

Rename path_tstamp ->  dirstamp.

Rename maildir -> root_maildir

Update dependents.
This commit is contained in:
Dirk-Jan C. Binnema 2020-01-31 00:20:34 +02:00
parent ad5f7d8dad
commit e51240f43f
6 changed files with 192 additions and 83 deletions

View File

@ -41,7 +41,7 @@ AM_CFLAGS= \
-DMU_TESTMAILDIR3=\"${abs_srcdir}/testdir3\" \ -DMU_TESTMAILDIR3=\"${abs_srcdir}/testdir3\" \
-DMU_TESTMAILDIR4=\"${abs_srcdir}/testdir4\" \ -DMU_TESTMAILDIR4=\"${abs_srcdir}/testdir4\" \
-DABS_CURDIR=\"${abs_builddir}\" \ -DABS_CURDIR=\"${abs_builddir}\" \
-DABS_SRCDIR=\"${abs_srcdir}\" -DABS_SRCDIR=\"${abs_srcdir}\" \
-Wno-format-nonliteral \ -Wno-format-nonliteral \
-Wno-switch-enum \ -Wno-switch-enum \
-Wno-deprecated-declarations \ -Wno-deprecated-declarations \
@ -56,7 +56,8 @@ AM_CXXFLAGS= \
$(WARN_CXXFLAGS) \ $(WARN_CXXFLAGS) \
$(XAPIAN_CXXFLAGS) \ $(XAPIAN_CXXFLAGS) \
$(ASAN_CXXFLAGS) \ $(ASAN_CXXFLAGS) \
$(CODE_COVERAGE_CFLAGS) $(CODE_COVERAGE_CFLAGS) \
-DMU_TESTMAILDIR=\"${abs_srcdir}/testdir\"
AM_CPPFLAGS= \ AM_CPPFLAGS= \
$(CODE_COVERAGE_CPPFLAGS) $(CODE_COVERAGE_CPPFLAGS)

View File

@ -112,8 +112,8 @@ needs_index (MuIndexCallbackData *data, const char *fullpath,
if (data->_reindex) if (data->_reindex)
return TRUE; return TRUE;
/* it's not in the database yet (FIXME: GError)*/ /* it's not in the database yet */
if (!mu_store_contains_message (data->_store, fullpath, NULL)) if (!mu_store_contains_message (data->_store, fullpath))
return TRUE; return TRUE;
/* it's there, but it's not up to date */ /* it's there, but it's not up to date */
@ -242,7 +242,7 @@ on_run_maildir_dir (const char* fullpath, gboolean enter,
*/ */
if (enter) { if (enter) {
data->_dirstamp = data->_dirstamp =
mu_store_get_timestamp (data->_store, fullpath, &err); mu_store_get_dirstamp (data->_store, fullpath, &err);
/* in 'lazy' mode, we only check the dir timestamp, and if it's /* in 'lazy' mode, we only check the dir timestamp, and if it's
* up to date, we don't bother with this dir. This fails to * up to date, we don't bother with this dir. This fails to
* account for messages below this dir that have merely * account for messages below this dir that have merely
@ -257,7 +257,7 @@ on_run_maildir_dir (const char* fullpath, gboolean enter,
} }
g_debug ("entering %s", fullpath); g_debug ("entering %s", fullpath);
} else { } else {
mu_store_set_timestamp (data->_store, fullpath, mu_store_set_dirstamp (data->_store, fullpath,
time(NULL), &err); time(NULL), &err);
g_debug ("leaving %s", fullpath); g_debug ("leaving %s", fullpath);
} }

View File

@ -19,6 +19,7 @@
** **
*/ */
#if HAVE_CONFIG_H #if HAVE_CONFIG_H
#include "config.h" #include "config.h"
#endif /*HAVE_CONFIG_H*/ #endif /*HAVE_CONFIG_H*/

View File

@ -36,7 +36,7 @@
using namespace Mu; using namespace Mu;
constexpr auto SchemaVersionKey = "schema-version"; constexpr auto SchemaVersionKey = "schema-version";
constexpr auto MaildirKey = "maildir"; constexpr auto RootMaildirKey = "maildir"; // XXX: make this 'root-maildir'
constexpr auto ContactsKey = "contacts"; constexpr auto ContactsKey = "contacts";
constexpr auto PersonalAddressesKey = "personal-addresses"; constexpr auto PersonalAddressesKey = "personal-addresses";
constexpr auto CreatedKey = "created"; constexpr auto CreatedKey = "created";
@ -45,6 +45,10 @@ constexpr auto ExpectedSchemaVersion = MU_STORE_SCHEMA_VERSION;
using Addresses = Store::Addresses; using Addresses = Store::Addresses;
extern "C" {
static unsigned add_or_update_msg (MuStore *store, unsigned docid, MuMsg *msg, GError **err);
}
/* we cache these prefix strings, so we don't have to allocate them all /* we cache these prefix strings, so we don't have to allocate them all
* the time; this should save 10-20 string allocs per message */ * the time; this should save 10-20 string allocs per message */
G_GNUC_CONST static const std::string& G_GNUC_CONST static const std::string&
@ -97,23 +101,23 @@ struct Store::Private {
db_{readonly? db_{readonly?
std::make_shared<Xapian::Database>(db_path_) : std::make_shared<Xapian::Database>(db_path_) :
std::make_shared<Xapian::WritableDatabase>(db_path_, Xapian::DB_OPEN)}, std::make_shared<Xapian::WritableDatabase>(db_path_, Xapian::DB_OPEN)},
maildir_{db()->get_metadata(MaildirKey)}, root_maildir_{db()->get_metadata(RootMaildirKey)},
created_{atoll(db()->get_metadata(CreatedKey).c_str())}, created_{atoll(db()->get_metadata(CreatedKey).c_str())},
schema_version_{db()->get_metadata(SchemaVersionKey)}, schema_version_{db()->get_metadata(SchemaVersionKey)},
personal_addresses_{Mu::split(db()->get_metadata(PersonalAddressesKey),",")}, personal_addresses_{Mu::split(db()->get_metadata(PersonalAddressesKey),",")},
contacts_{db()->get_metadata(ContactsKey)} { contacts_{db()->get_metadata(ContactsKey)} {
} }
Private (const std::string& path, const std::string& maildir): Private (const std::string& path, const std::string& root_maildir):
db_path_{path}, db_path_{path},
db_{std::make_shared<Xapian::WritableDatabase>( db_{std::make_shared<Xapian::WritableDatabase>(
db_path_, Xapian::DB_CREATE_OR_OVERWRITE)}, db_path_, Xapian::DB_CREATE_OR_OVERWRITE)},
maildir_{maildir}, root_maildir_{root_maildir},
created_{time({})}, created_{time({})},
schema_version_{MU_STORE_SCHEMA_VERSION} { schema_version_{MU_STORE_SCHEMA_VERSION} {
writable_db()->set_metadata(SchemaVersionKey, schema_version_); writable_db()->set_metadata(SchemaVersionKey, schema_version_);
writable_db()->set_metadata(MaildirKey, maildir_); writable_db()->set_metadata(RootMaildirKey, root_maildir_);
writable_db()->set_metadata(CreatedKey, Mu::format("%" PRId64, (int64_t)created_)); writable_db()->set_metadata(CreatedKey, Mu::format("%" PRId64, (int64_t)created_));
} }
@ -174,7 +178,7 @@ struct Store::Private {
const std::string db_path_; const std::string db_path_;
std::shared_ptr<Xapian::Database> db_; std::shared_ptr<Xapian::Database> db_;
const std::string maildir_; const std::string root_maildir_;
const time_t created_{}; const time_t created_{};
const std::string schema_version_; const std::string schema_version_;
Addresses personal_addresses_; Addresses personal_addresses_;
@ -187,6 +191,24 @@ struct Store::Private {
}; };
static void
hash_str (char *buf, size_t buf_size, const char *data)
{
g_snprintf(buf, buf_size, "016%" PRIx64, mu_util_get_hash(data));
}
static std::string
get_uid_term (const char* path)
{
char uid_term[1 + 16 + 1] = {'\0'};
uid_term[0] = mu_msg_field_xapian_prefix(MU_MSG_FIELD_ID_UID);
hash_str(uid_term + 1, sizeof(uid_term)-1, path);
return std::string{uid_term, sizeof(uid_term)};
}
#undef LOCKED #undef LOCKED
#define LOCKED std::lock_guard<std::mutex> l(priv_->lock_); #define LOCKED std::lock_guard<std::mutex> l(priv_->lock_);
@ -196,15 +218,15 @@ Store::Store (const std::string& path, bool readonly):
if (ExpectedSchemaVersion == schema_version()) if (ExpectedSchemaVersion == schema_version())
return; // All is good; nothing further to do return; // All is good; nothing further to do
if (readonly || maildir().empty()) if (readonly || root_maildir().empty())
throw Mu::Error(Error::Code::SchemaMismatch, "database needs reindexing"); throw Mu::Error(Error::Code::SchemaMismatch, "database needs reindexing");
g_debug ("upgrading database"); g_debug ("upgrading database");
const auto addresses{personal_addresses()}; const auto addresses{personal_addresses()};
const auto mdir{maildir()}; const auto root_mdir{root_maildir()};
priv_.reset(); priv_.reset();
priv_ = std::make_unique<Private> (path, mdir); priv_ = std::make_unique<Private> (path, root_mdir);
set_personal_addresses (addresses); set_personal_addresses (addresses);
} }
@ -222,10 +244,10 @@ Store::read_only() const
} }
const std::string& const std::string&
Store::maildir () const Store::root_maildir () const
{ {
LOCKED; LOCKED;
return priv_->maildir_; return priv_->root_maildir_;
} }
void void
@ -287,8 +309,77 @@ Store::created() const
return priv_->created_; return priv_->created_;
} }
static std::string
maildir_from_path (const std::string& root, const std::string& path)
{
if (G_UNLIKELY(root.empty()) || root.length() >= path.length() ||
path.find(root) != 0)
throw Mu::Error{Error::Code::InvalidArgument,
"root '%s' is not a proper suffix of path '%s'",
root.c_str(), path.c_str()};
auto mdir{path.substr(root.length())};
auto slash{mdir.rfind('/')};
if (G_UNLIKELY(slash == std::string::npos) || slash < 4)
throw Mu::Error{Error::Code::InvalidArgument,
"invalid path: %s", path.c_str()};
mdir.erase(slash);
auto subdir=mdir.data()+slash-4;
if (G_UNLIKELY(strncmp(subdir, "/cur", 4) != 0 &&
strncmp(subdir, "/new", 4)))
throw Mu::Error{Error::Code::InvalidArgument,
"cannot find '/new' or '/cur' - invalid path: %s", path.c_str()};
if (mdir.length() == 4)
return "/";
mdir.erase(mdir.length()-4);
return mdir;
}
unsigned
Store::add_message (const std::string& path)
{
LOCKED;
GError *gerr{};
const auto maildir{ maildir_from_path(root_maildir(), path)};
auto msg{mu_msg_new_from_file (path.c_str(), maildir.c_str(), &gerr)};
if (G_UNLIKELY(!msg))
throw Error{Error::Code::Message, "failed to create message: %s",
gerr ? gerr->message : "something went wrong"};
auto store{reinterpret_cast<MuStore*>(this)}; // yuk.
const auto docid{add_or_update_msg (store, 0, msg, &gerr)};
mu_msg_unref (msg);
if (G_UNLIKELY(docid == MU_STORE_INVALID_DOCID))
throw Error{Error::Code::Message, "failed to store message: %s",
gerr ? gerr->message : "something went wrong"};
return docid;
}
bool
Store::remove_message (const std::string& path)
{
LOCKED;
try {
const std::string term{(get_uid_term(path.c_str()))};
auto wdb{priv()->wdb()};
wdb->delete_document (term);
} MU_XAPIAN_CATCH_BLOCK_RETURN (false);
return true;
}
time_t time_t
Store::path_tstamp (const std::string& path) const Store::dirstamp (const std::string& path) const
{ {
LOCKED; LOCKED;
@ -300,16 +391,29 @@ Store::path_tstamp (const std::string& path) const
} }
void void
Store::set_path_tstamp (const std::string& path, time_t tstamp) Store::set_dirstamp (const std::string& path, time_t tstamp)
{ {
LOCKED; LOCKED;
std::array<char, 2*sizeof(tstamp)+1> data{}; std::array<char, 2*sizeof(tstamp)+1> data{};
const std::size_t len = g_snprintf (data.data(), data.size(), "%x", tstamp); const std::size_t len = g_snprintf (data.data(), data.size(), "%zx", tstamp);
priv_->writable_db()->set_metadata(path, std::string{data.data(), len}); priv_->writable_db()->set_metadata(path, std::string{data.data(), len});
} }
bool
Store::contains_message (const std::string& path) const
{
LOCKED;
try {
const std::string term (get_uid_term(path.c_str()));
return priv_->db()->term_exists (term);
} MU_XAPIAN_CATCH_BLOCK_RETURN(false);
}
void void
Store::begin_transaction () try Store::begin_transaction () try
{ {
@ -386,24 +490,6 @@ mutable_self (MuStore *store)
} }
static void
hash_str (char *buf, size_t buf_size, const char *data)
{
g_snprintf(buf, buf_size, "016%" PRIx64, mu_util_get_hash(data));
}
static std::string
get_uid_term (const char* path)
{
char uid_term[1 + 16 + 1] = {'\0'};
uid_term[0] = mu_msg_field_xapian_prefix(MU_MSG_FIELD_ID_UID);
hash_str(uid_term + 1, sizeof(uid_term)-1, path);
return std::string{uid_term, sizeof(uid_term)};
}
MuStore* MuStore*
mu_store_new_readable (const char* xpath, GError **err) mu_store_new_readable (const char* xpath, GError **err)
{ {
@ -579,16 +665,15 @@ mu_store_get_read_only_database (MuStore *store)
} }
gboolean gboolean
mu_store_contains_message (const MuStore *store, const char* path, GError **err) mu_store_contains_message (const MuStore *store, const char* path)
{ {
g_return_val_if_fail (store, FALSE); g_return_val_if_fail (store, FALSE);
g_return_val_if_fail (path, FALSE); g_return_val_if_fail (path, FALSE);
try { try {
const std::string term (get_uid_term(path)); return self(store)->contains_message(path) ? TRUE : FALSE;
return self(store)->priv()->db()->term_exists (term) ? TRUE: FALSE;
} MU_XAPIAN_CATCH_BLOCK_G_ERROR_RETURN(err, MU_ERROR_XAPIAN, FALSE); } MU_XAPIAN_CATCH_BLOCK_RETURN(FALSE);
} }
unsigned unsigned
@ -679,7 +764,7 @@ mu_store_maildir (const MuStore *store)
{ {
g_return_val_if_fail (store, NULL); g_return_val_if_fail (store, NULL);
return self(store)->maildir().c_str(); return self(store)->root_maildir().c_str();
} }
@ -1222,7 +1307,7 @@ add_or_update_msg (MuStore *store, unsigned docid, MuMsg *msg, GError **err)
const std::string term (get_uid_term (mu_msg_get_path(msg))); const std::string term (get_uid_term (mu_msg_get_path(msg)));
auto self = mutable_self(store); auto self = mutable_self(store);
auto wdb = self->priv()->wdb(); auto wdb = self->priv()->wdb();
if (!self->in_transaction()) if (!self->in_transaction())
self->begin_transaction(); self->begin_transaction();
@ -1324,24 +1409,24 @@ mu_store_remove_path (MuStore *store, const char *msgpath)
gboolean gboolean
mu_store_set_timestamp (MuStore *store, const char* msgpath, mu_store_set_dirstamp (MuStore *store, const char* dirpath,
time_t stamp, GError **err) time_t stamp, GError **err)
{ {
g_return_val_if_fail (store, FALSE); g_return_val_if_fail (store, FALSE);
g_return_val_if_fail (msgpath, FALSE); g_return_val_if_fail (dirpath, FALSE);
mutable_self(store)->set_path_tstamp(msgpath, stamp); mutable_self(store)->set_dirstamp(dirpath, stamp);
return TRUE; return TRUE;
} }
time_t time_t
mu_store_get_timestamp (const MuStore *store, const char *msgpath, GError **err) mu_store_get_dirstamp (const MuStore *store, const char *dirpath, GError **err)
{ {
g_return_val_if_fail (store, 0); g_return_val_if_fail (store, 0);
g_return_val_if_fail (msgpath, 0); g_return_val_if_fail (dirpath, 0);
return self(store)->path_tstamp(msgpath); return self(store)->dirstamp(dirpath);
} }

View File

@ -75,7 +75,7 @@ public:
* *
* @return the maildir * @return the maildir
*/ */
const std::string& maildir() const; const std::string& root_maildir() const;
/** /**
* Version of the database-schema * Version of the database-schema
@ -117,6 +117,51 @@ public:
*/ */
const Contacts& contacts() const; const Contacts& contacts() const;
/**
* Add a message to the store.
*
* @param path the message path.
*
* @return the doc id of the added message
*/
unsigned add_message (const std::string& path);
/**
* Add a message to the store.
*
* @param path the message path.
*
* @return true if removing happened; false otherwise.
*/
bool remove_message (const std::string& path);
/**
* does a certain message exist in the store already?
*
* @param path the message path
*
* @return true if the message exists in the store, false otherwise
*/
bool contains_message (const std::string& path) const;
/**
* Get the timestamp for some directory
*
* @param path the path
*
* @return the timestamp, or 0 if not found
*/
time_t dirstamp (const std::string& path) const;
/**
* Set the timestamp for some directory
*
* @param path a filesystem path
* @param tstamp the timestamp for that path
*/
void set_dirstamp (const std::string& path, time_t tstamp);
/** /**
* Get the number of documents in the document database * Get the number of documents in the document database
* *
@ -131,24 +176,6 @@ public:
*/ */
bool empty() const; bool empty() const;
/**
* Get the timestamp for a given path.
*
* @param path the path
*
* @return the timestamp, or 0 if not found
*/
time_t path_tstamp (const std::string& path) const;
/**
* Set the timestamp for some path
*
* @param path a filesystem path
* @param tstamp the timestamp for that path
*/
void set_path_tstamp (const std::string& path, time_t tstamp);
/** /**
* Begin a database transaction * Begin a database transaction
*/ */
@ -451,18 +478,15 @@ unsigned mu_store_add_path (MuStore *store, const char *path,
*/ */
gboolean mu_store_remove_path (MuStore *store, const char* msgpath); gboolean mu_store_remove_path (MuStore *store, const char* msgpath);
/** /**
* does a certain message exist in the database already? * does a certain message exist in the database already?
* *
* @param store a store * @param store a store
* @param path the message path * @param path the message path
* @param err to receive error info or NULL. err->code is MuError value
* *
* @return TRUE if the message exists, FALSE otherwise * @return TRUE if the message exists, FALSE otherwise
*/ */
gboolean mu_store_contains_message (const MuStore *store, const char* path, gboolean mu_store_contains_message (const MuStore *store, const char* path);
GError **err);
/** /**
* get the docid for message at path * get the docid for message at path
@ -480,25 +504,25 @@ unsigned mu_store_get_docid_for_path (const MuStore *store, const char* path,
* store a timestamp for a directory * store a timestamp for a directory
* *
* @param store a valid store * @param store a valid store
* @param msgpath path to a maildir * @param dirpath path to some directory
* @param stamp a timestamp * @param stamp a timestamp
* @param err to receive error info or NULL. err->code is MuError value * @param err to receive error info or NULL. err->code is MuError value
* *
* @return TRUE if setting the timestamp succeeded, FALSE otherwise * @return TRUE if setting the timestamp succeeded, FALSE otherwise
*/ */
gboolean mu_store_set_timestamp (MuStore *store, const char* msgpath, gboolean mu_store_set_dirstamp (MuStore *store, const char* dirpath,
time_t stamp, GError **err); time_t stamp, GError **err);
/** /**
* get the timestamp for a directory * get the timestamp for a directory
* *
* @param store a valid store * @param store a valid store
* @param msgpath path to a maildir * @param msgpath path to some directory
* @param err to receive error info or NULL. err->code is MuError value * @param err to receive error info or NULL. err->code is MuError value
* *
* @return the timestamp, or 0 in case of error * @return the timestamp, or 0 in case of error
*/ */
time_t mu_store_get_timestamp (const MuStore *store, const char* msgpath, time_t mu_store_get_dirstamp (const MuStore *store, const char* dirpath,
GError **err); GError **err);
/** /**

View File

@ -112,8 +112,7 @@ test_mu_store_store_msg_and_count (void)
g_assert_cmpuint (1,==,mu_store_count (store, NULL)); g_assert_cmpuint (1,==,mu_store_count (store, NULL));
g_assert_cmpuint (TRUE,==,mu_store_contains_message g_assert_cmpuint (TRUE,==,mu_store_contains_message
(store, (store,
MU_TESTMAILDIR "/cur/1283599333.1840_11.cthulhu!2,", MU_TESTMAILDIR "/cur/1283599333.1840_11.cthulhu!2,"));
NULL));
mu_msg_unref (msg); mu_msg_unref (msg);
/* add another one */ /* add another one */
@ -125,7 +124,7 @@ test_mu_store_store_msg_and_count (void)
g_assert_cmpuint (2,==,mu_store_count (store, NULL)); g_assert_cmpuint (2,==,mu_store_count (store, NULL));
g_assert_cmpuint (TRUE,==, g_assert_cmpuint (TRUE,==,
mu_store_contains_message (store, MU_TESTMAILDIR2 mu_store_contains_message (store, MU_TESTMAILDIR2
"/bar/cur/mail3", NULL)); "/bar/cur/mail3"));
mu_msg_unref (msg); mu_msg_unref (msg);
/* try to add the first one again. count should be 2 still */ /* try to add the first one again. count should be 2 still */
@ -175,8 +174,7 @@ test_mu_store_store_msg_remove_and_count (void)
g_assert_cmpuint (0,==,mu_store_count (store, NULL)); g_assert_cmpuint (0,==,mu_store_count (store, NULL));
g_assert_cmpuint (FALSE,==,mu_store_contains_message g_assert_cmpuint (FALSE,==,mu_store_contains_message
(store, (store,
MU_TESTMAILDIR "/cur/1283599333.1840_11.cthulhu!2,", MU_TESTMAILDIR "/cur/1283599333.1840_11.cthulhu!2,"));
NULL));
g_free (tmpdir); g_free (tmpdir);
mu_store_unref (store); mu_store_unref (store);
} }