X-Git-Url: https://git.notmuchmail.org/git?p=notmuch;a=blobdiff_plain;f=lib%2Fquery.cc;h=686d75f9657cf4e142a1983c41e0ab29ee3ff960;hp=1777e47dc044daa339c7ec0988e3d0fbef3de5a9;hb=53f8cc565126db4a003dbfc02850d2bf3b260636;hpb=93dcc3b695e19dd36cc8f638c6e01ecbbd9a447d diff --git a/lib/query.cc b/lib/query.cc index 1777e47d..686d75f9 100644 --- a/lib/query.cc +++ b/lib/query.cc @@ -61,7 +61,7 @@ notmuch_query_create (notmuch_database_t *notmuch, query->query_string = talloc_strdup (query, query_string); - query->sort = NOTMUCH_SORT_DATE_OLDEST_FIRST; + query->sort = NOTMUCH_SORT_NEWEST_FIRST; return query; } @@ -72,21 +72,6 @@ notmuch_query_set_sort (notmuch_query_t *query, notmuch_sort_t sort) query->sort = sort; } -/* We end up having to call the destructors explicitly because we had - * to use "placement new" in order to initialize C++ objects within a - * block that we allocated with talloc. So C++ is making talloc - * slightly less simple to use, (we wouldn't need - * talloc_set_destructor at all otherwise). - */ -static int -_notmuch_messages_destructor (notmuch_messages_t *messages) -{ - messages->iterator.~MSetIterator (); - messages->iterator_end.~MSetIterator (); - - return 0; -} - notmuch_messages_t * notmuch_query_search_messages (notmuch_query_t *query, int first, @@ -94,10 +79,11 @@ notmuch_query_search_messages (notmuch_query_t *query, { notmuch_database_t *notmuch = query->notmuch; const char *query_string = query->query_string; - notmuch_messages_t *messages; + notmuch_message_list_t *message_list; + Xapian::MSetIterator i; - messages = talloc (query, notmuch_messages_t); - if (unlikely (messages == NULL)) + message_list = _notmuch_message_list_create (query); + if (unlikely (message_list == NULL)) return NULL; try { @@ -111,7 +97,8 @@ notmuch_query_search_messages (notmuch_query_t *query, Xapian::QueryParser::FLAG_PHRASE | Xapian::QueryParser::FLAG_LOVEHATE | Xapian::QueryParser::FLAG_BOOLEAN_ANY_CASE | - Xapian::QueryParser::FLAG_WILDCARD); + Xapian::QueryParser::FLAG_WILDCARD | + Xapian::QueryParser::FLAG_PURE_NOT); if (strcmp (query_string, "") == 0) { final_query = mail_query; @@ -123,10 +110,10 @@ notmuch_query_search_messages (notmuch_query_t *query, } switch (query->sort) { - case NOTMUCH_SORT_DATE_OLDEST_FIRST: + case NOTMUCH_SORT_OLDEST_FIRST: enquire.set_sort_by_value (NOTMUCH_VALUE_TIMESTAMP, FALSE); break; - case NOTMUCH_SORT_DATE_NEWEST_FIRST: + case NOTMUCH_SORT_NEWEST_FIRST: enquire.set_sort_by_value (NOTMUCH_VALUE_TIMESTAMP, TRUE); break; case NOTMUCH_SORT_MESSAGE_ID: @@ -144,22 +131,31 @@ notmuch_query_search_messages (notmuch_query_t *query, max_messages = notmuch->xapian_db->get_doccount (); mset = enquire.get_mset (first, max_messages); - messages->notmuch = notmuch; + for (i = mset.begin (); i != mset.end (); i++) { + notmuch_message_t *message; + notmuch_private_status_t status; - new (&messages->iterator) Xapian::MSetIterator (); - new (&messages->iterator_end) Xapian::MSetIterator (); - - talloc_set_destructor (messages, _notmuch_messages_destructor); + message = _notmuch_message_create (message_list, notmuch, + *i, &status); + if (message == NULL) + { + if (status == NOTMUCH_PRIVATE_STATUS_NO_DOCUMENT_FOUND) + INTERNAL_ERROR ("A message iterator contains a " + "non-existent document ID.\n"); + break; + } - messages->iterator = mset.begin (); - messages->iterator_end = mset.end (); + _notmuch_message_list_add_message (message_list, message); + } } catch (const Xapian::Error &error) { - fprintf (stderr, "A Xapian exception occurred: %s\n", + fprintf (stderr, "A Xapian exception occurred performing query: %s\n", error.get_msg().c_str()); + fprintf (stderr, "Query string was: %s\n", query->query_string); + notmuch->exception_reported = TRUE; } - return messages; + return _notmuch_messages_create (message_list); } /* Glib objects force use to use a talloc destructor as well, (but not @@ -200,7 +196,7 @@ notmuch_query_search_threads (notmuch_query_t *query, seen = g_hash_table_new_full (g_str_hash, g_str_equal, free, NULL); - while (threads_seen < first + max_threads) + while (max_threads < 0 || threads_seen < first + max_threads) { int messages_seen_previously = messages_seen; @@ -218,9 +214,10 @@ notmuch_query_search_threads (notmuch_query_t *query, thread_id, NULL, (void **) &thread)) { - if (threads_seen > first) { + if (threads_seen >= first) { thread = _notmuch_thread_create (query, query->notmuch, - thread_id); + thread_id, + query->query_string); g_ptr_array_add (threads->threads, thread); } else { thread = NULL; @@ -231,12 +228,12 @@ notmuch_query_search_threads (notmuch_query_t *query, threads_seen++; } - if (thread) - _notmuch_thread_add_message (thread, message); - notmuch_message_destroy (message); messages_seen++; + + if (max_threads >= 0 && threads_seen >= first + max_threads) + break; } /* Stop if we're not seeing any more messages. */ @@ -255,49 +252,6 @@ notmuch_query_destroy (notmuch_query_t *query) talloc_free (query); } -notmuch_bool_t -notmuch_messages_has_more (notmuch_messages_t *messages) -{ - return (messages->iterator != messages->iterator_end); -} - -notmuch_message_t * -notmuch_messages_get (notmuch_messages_t *messages) -{ - notmuch_message_t *message; - Xapian::docid doc_id; - notmuch_private_status_t status; - - if (! notmuch_messages_has_more (messages)) - return NULL; - - doc_id = *messages->iterator; - - message = _notmuch_message_create (messages, - messages->notmuch, doc_id, - &status); - - if (message == NULL && - status == NOTMUCH_PRIVATE_STATUS_NO_DOCUMENT_FOUND) - { - INTERNAL_ERROR ("a messages iterator contains a non-existent document ID.\n"); - } - - return message; -} - -void -notmuch_messages_advance (notmuch_messages_t *messages) -{ - messages->iterator++; -} - -void -notmuch_messages_destroy (notmuch_messages_t *messages) -{ - talloc_free (messages); -} - notmuch_bool_t notmuch_threads_has_more (notmuch_threads_t *threads) { @@ -325,3 +279,55 @@ notmuch_threads_destroy (notmuch_threads_t *threads) { talloc_free (threads); } + +unsigned +notmuch_query_count_messages (notmuch_query_t *query) +{ + notmuch_database_t *notmuch = query->notmuch; + const char *query_string = query->query_string; + Xapian::doccount count; + + try { + Xapian::Enquire enquire (*notmuch->xapian_db); + Xapian::Query mail_query (talloc_asprintf (query, "%s%s", + _find_prefix ("type"), + "mail")); + Xapian::Query string_query, final_query; + Xapian::MSet mset; + unsigned int flags = (Xapian::QueryParser::FLAG_BOOLEAN | + Xapian::QueryParser::FLAG_PHRASE | + Xapian::QueryParser::FLAG_LOVEHATE | + Xapian::QueryParser::FLAG_BOOLEAN_ANY_CASE | + Xapian::QueryParser::FLAG_WILDCARD | + Xapian::QueryParser::FLAG_PURE_NOT); + + if (strcmp (query_string, "") == 0) { + final_query = mail_query; + } else { + string_query = notmuch->query_parser-> + parse_query (query_string, flags); + final_query = Xapian::Query (Xapian::Query::OP_AND, + mail_query, string_query); + } + + enquire.set_weighting_scheme(Xapian::BoolWeight()); + enquire.set_docid_order(Xapian::Enquire::ASCENDING); + +#if DEBUG_QUERY + fprintf (stderr, "Final query is:\n%s\n", final_query.get_description().c_str()); +#endif + + enquire.set_query (final_query); + + mset = enquire.get_mset (0, notmuch->xapian_db->get_doccount ()); + + count = mset.get_matches_estimated(); + + } catch (const Xapian::Error &error) { + fprintf (stderr, "A Xapian exception occurred: %s\n", + error.get_msg().c_str()); + fprintf (stderr, "Query string was: %s\n", query->query_string); + } + + return count; +}