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 #include "nsIWeakReference.h"
  30 #include "nsIWeakReferenceUtils.h"
  31
  32 /* Unfrozen API */
  33 #include "unstable/mozIStorageConnection.h"
  34 #include "unstable/mozIStorageStatement.h"
  35
  36 /* Project includes */
  37 #include <abstract/base/aaIDataNode.h>
  38 #include <abstract/storage/aaStorageUtils.h>
  39 #include <abstract/storage/aaIFilter.h>
  40 #include <abstract/storage/aaILoadQuery.h>
  41 #include "aaLoadQuery.h"
  42
  43 aaLoadQuery::aaLoadQuery(nsISupports *aOuter)
  44   :mNewFilter(PR_FALSE)
  45 {
  46   mConnection = do_QueryInterface(aOuter);
  47   mData = do_CreateInstance(NS_ARRAY_CONTRACTID);
  48   if (!mData)
  49     mConnection = nsnull;
  50 }
  51
  52 aaLoadQuery::~aaLoadQuery()
  53 {
  54 }
  55
  56 NS_IMPL_ISUPPORTS3(aaLoadQuery,
  57                    nsIArray,
  58                    aaILoadQuery,
  59                    aaILoadQueryInternal)
  60
  61 /* aaILoadQuery */
  62 NS_IMETHODIMP
  63 aaLoadQuery::Load()
  64 {
  65   nsresult rv;
  66
  67   rv = mData->Clear();
  68   NS_ENSURE_SUCCESS(rv, rv);
  69
  70   if (! mStatement || mNewFilter) {
  71     nsEmbedCString sql(getSelectQuery());
  72     if (mFilter) {
  73       nsEmbedCString filter;
  74       rv = mFilter->GetExpression(filter);
  75       NS_ENSURE_SUCCESS(rv, rv);
  76
  77       sql.Append(filter);
  78     }
  79     rv = mConnection->CreateStatement(sql, getter_AddRefs(mStatement));
  80     NS_ENSURE_SUCCESS(rv, rv);
  81     mNewFilter = PR_FALSE;
  82   }
  83
  84   PRBool hasMore;
  85   PRInt32 state;
  86   nsCOMPtr<aaIDataNode> row;
  87   nsCOMPtr<aaILoadObserver> observer;
  88   if (mObserver) {
  89     observer = do_QueryReferent(mObserver, &rv);
  90     NS_ENSURE_SUCCESS(rv, rv);
  91   }
  92   do {
  93     rv = mStatement->ExecuteStep(&hasMore);
  94     if (NS_FAILED(rv))
  95       break;
  96     rv = mStatement->GetState(&state);
  97     if (NS_FAILED(rv))
  98       continue;
  99     if (state != mozIStorageStatement::MOZ_STORAGE_STATEMENT_EXECUTING)
 100       continue;
 101     row = getter_AddRefs(fetchRow());
 102     if (row) {
 103       mData->AppendElement(row, PR_FALSE);
 104       if (observer) observer->ObserveLoad(row);
 105     }
 106   } while (hasMore);
 107   mStatement->Reset();
 108   NS_ENSURE_SUCCESS(rv, rv);
 109   return NS_OK;
 110 }
 111
 112 NS_IMETHODIMP
 113 aaLoadQuery::GetFilter(aaIFilter * * aFilter)
 114 {
 115   NS_ENSURE_ARG_POINTER(aFilter);
 116   NS_IF_ADDREF(*aFilter = mFilter);
 117   return NS_OK;
 118 }
 119 NS_IMETHODIMP
 120 aaLoadQuery::SetFilter(aaIFilter * aFilter)
 121 {
 122   mFilter = aFilter;
 123   mNewFilter = PR_TRUE;
 124   return NS_OK;
 125 }
 126 aaIFilter *
 127 aaLoadQuery::PickFilter()
 128 {
 129   return mFilter;
 130 }
 131
 132 /* aaILoadQueryInternal */
 133 NS_IMETHODIMP
 134 aaLoadQuery::SetLoadObserver(aaILoadObserver *aObserver)
 135 {
 136   nsresult rv;
 137   nsCOMPtr<nsISupportsWeakReference> swr = do_QueryInterface(aObserver, &rv);
 138   NS_ENSURE_SUCCESS(rv, rv);
 139   rv = swr->GetWeakReference(getter_AddRefs( mObserver ));
 140   NS_ENSURE_SUCCESS(rv, rv);
 141   return NS_OK;
 142 }