storage/aaLoadQuery.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim:set ts=2 sw=2 sts=2 et cindent tw=79 ft=cpp: */
   3 /*
   4  * Copyright (C) 2007 Sergey Yanovich <ynvich@gmail.com>
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License as
   8  * published by the Free Software Foundation; either version 2 of the
   9  * License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public
  17  * License along with this program; if not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 #include <abstract/aacore.h>
  23
  24 #include "nsCOMPtr.h"
  25 #include "nsIMutableArray.h"
  26 #include "nsComponentManagerUtils.h"
  27 #include "nsStringAPI.h"
  28 #include "nsEmbedString.h"
  29
  30 /* Unfrozen API */
  31 #include "unstable/mozIStorageConnection.h"
  32 #include "unstable/mozIStorageStatement.h"
  33
  34 /* Project includes */
  35 #include <abstract/base/aaIDataNode.h>
  36 #include <abstract/storage/aaStorageUtils.h>
  37 #include <abstract/storage/aaIFilter.h>
  38 #include <abstract/storage/aaILoadQuery.h>
  39 #include "aaLoadQuery.h"
  40
  41 aaLoadQuery::aaLoadQuery(nsISupports *aOuter)
  42   :mNewFilter(PR_FALSE)
  43 {
  44   mConnection = do_QueryInterface(aOuter);
  45   mData = do_CreateInstance(NS_ARRAY_CONTRACTID);
  46   if (!mData)
  47     mConnection = nsnull;
  48 }
  49
  50 aaLoadQuery::~aaLoadQuery()
  51 {
  52 }
  53
  54 NS_IMPL_ISUPPORTS2(aaLoadQuery,
  55                    nsIArray,
  56                    aaILoadQuery)
  57
  58 /* aaILoadQuery */
  59 NS_IMETHODIMP
  60 aaLoadQuery::Load()
  61 {
  62   nsresult rv;
  63
  64   rv = mData->Clear();
  65   NS_ENSURE_SUCCESS(rv, rv);
  66
  67   if (! mStatement || mNewFilter) {
  68     nsEmbedCString sql(getSelectQuery());
  69     if (mFilter) {
  70       nsEmbedCString filter;
  71       rv = mFilter->GetExpression(filter);
  72       NS_ENSURE_SUCCESS(rv, rv);
  73
  74       sql.Append(filter);
  75     }
  76     rv = mConnection->CreateStatement(sql, getter_AddRefs(mStatement));
  77     NS_ENSURE_SUCCESS(rv, rv);
  78     mNewFilter = PR_FALSE;
  79   }
  80
  81   PRBool hasMore;
  82   PRInt32 state;
  83   nsCOMPtr<aaIDataNode> row;
  84   do {
  85     rv = mStatement->ExecuteStep(&hasMore);
  86     if (NS_FAILED(rv))
  87       break;
  88     rv = mStatement->GetState(&state);
  89     if (NS_FAILED(rv))
  90       continue;
  91     if (state != mozIStorageStatement::MOZ_STORAGE_STATEMENT_EXECUTING)
  92       continue;
  93     row = getter_AddRefs(fetchRow());
  94     if (row)
  95       mData->AppendElement(row, PR_FALSE);
  96   } while (hasMore);
  97   mStatement->Reset();
  98   NS_ENSURE_SUCCESS(rv, rv);
  99   return NS_OK;
 100 }
 101
 102 NS_IMETHODIMP
 103 aaLoadQuery::GetFilter(aaIFilter * * aFilter)
 104 {
 105   NS_ENSURE_ARG_POINTER(aFilter);
 106   NS_IF_ADDREF(*aFilter = mFilter);
 107   return NS_OK;
 108 }
 109 NS_IMETHODIMP
 110 aaLoadQuery::SetFilter(aaIFilter * aFilter)
 111 {
 112   mFilter = aFilter;
 113   mNewFilter = PR_TRUE;
 114   return NS_OK;
 115 }
 116 aaIFilter *
 117 aaLoadQuery::PickFilter()
 118 {
 119   return mFilter;
 120 }