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: */
4 * Copyright (C) 2007 Sergey Yanovich <ynvich@gmail.com>
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.
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.
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.
22 #include <abstract/aacore.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"
33 #include "unstable/mozIStorageConnection.h"
34 #include "unstable/mozIStorageStatement.h"
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"
43 aaLoadQuery::aaLoadQuery(nsISupports
*aOuter
)
46 mConnection
= do_QueryInterface(aOuter
);
47 mData
= do_CreateInstance(NS_ARRAY_CONTRACTID
);
52 aaLoadQuery::~aaLoadQuery()
56 NS_IMPL_ISUPPORTS3(aaLoadQuery
,
68 NS_ENSURE_SUCCESS(rv
, rv
);
70 if (! mStatement
|| mNewFilter
) {
71 nsEmbedCString
sql(getSelectQuery());
73 nsEmbedCString filter
;
74 rv
= mFilter
->GetExpression(filter
);
75 NS_ENSURE_SUCCESS(rv
, rv
);
79 rv
= mConnection
->CreateStatement(sql
, getter_AddRefs(mStatement
));
80 NS_ENSURE_SUCCESS(rv
, rv
);
81 mNewFilter
= PR_FALSE
;
86 nsCOMPtr
<aaIDataNode
> row
;
87 nsCOMPtr
<aaILoadObserver
> observer
;
89 observer
= do_QueryReferent(mObserver
, &rv
);
90 NS_ENSURE_SUCCESS(rv
, rv
);
93 rv
= mStatement
->ExecuteStep(&hasMore
);
96 rv
= mStatement
->GetState(&state
);
99 if (state
!= mozIStorageStatement::MOZ_STORAGE_STATEMENT_EXECUTING
)
101 row
= getter_AddRefs(fetchRow());
103 mData
->AppendElement(row
, PR_FALSE
);
104 if (observer
) observer
->ObserveLoad(row
);
108 NS_ENSURE_SUCCESS(rv
, rv
);
113 aaLoadQuery::GetFilter(aaIFilter
* * aFilter
)
115 NS_ENSURE_ARG_POINTER(aFilter
);
116 NS_IF_ADDREF(*aFilter
= mFilter
);
120 aaLoadQuery::SetFilter(aaIFilter
* aFilter
)
123 mNewFilter
= PR_TRUE
;
127 aaLoadQuery::PickFilter()
132 /* aaILoadQueryInternal */
134 aaLoadQuery::SetLoadObserver(aaILoadObserver
*aObserver
)
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
);