storage/aaLoadIncome.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/aaIEntity.h>
  36 #include <abstract/base/aaIResource.h>
  37 #include <abstract/base/aaIFlow.h>
  38 #include <abstract/base/aaIState.h>
  39 #include <abstract/base/aaIBalance.h>
  40 #include <abstract/storage/aaIFilter.h>
  41 #include <abstract/storage/aaILoadQuery.h>
  42 #include "aaLoadIncome.h"
  43
  44 #define AA_LOADINCOME_SELECT_QUERY "SELECT income.side, income.value\
  45   FROM income"
  46 #define AA_LI_DEFAULT_FILTER \
  47   " WHERE income.paid IS NULL"
  48
  49 aaLoadIncome::aaLoadIncome(nsISupports *aOuter)
  50   :aaLoadQuery(aOuter)
  51 {
  52   mFilter = do_CreateInstance(AA_FILTER_CONTRACT_ID);
  53   nsCOMPtr<aaIStringFilter> strFilter(do_QueryInterface(mFilter));
  54   strFilter->SetExpression(NS_LITERAL_CSTRING(AA_LI_DEFAULT_FILTER));
  55   mNewFilter = PR_TRUE;
  56 }
  57
  58 aaLoadIncome::~aaLoadIncome()
  59 {
  60 }
  61
  62 NS_IMPL_ISUPPORTS_INHERITED0(aaLoadIncome,
  63                              aaLoadQuery)
  64
  65 /* Virtual methods */
  66 const char*
  67 aaLoadIncome::getSelectQuery() const
  68 {
  69   return AA_LOADINCOME_SELECT_QUERY;
  70 }
  71
  72 aaIDataNode*
  73 aaLoadIncome::fetchRow()
  74 {
  75   nsresult rv;
  76   PRBool side;
  77   double value;
  78
  79   rv = mStatement->GetInt32(0, &side);
  80   NS_ENSURE_SUCCESS(rv, nsnull);
  81
  82   rv = mStatement->GetDouble(1, &value);
  83   NS_ENSURE_SUCCESS(rv, nsnull);
  84
  85   nsCOMPtr<aaIFlow> finres = do_CreateInstance(
  86       "@aasii.org/base/income-flow;1", &rv);
  87   NS_ENSURE_SUCCESS(rv, nsnull);
  88
  89   aaIBalance *balance;
  90   CallCreateInstance("@aasii.org/base/state;1", nsnull, &balance);
  91   NS_ENSURE_TRUE(balance, nsnull);
  92
  93   balance->SetSide(side);
  94   balance->SetValue(value);
  95   balance->SetAmount(0.0);
  96   balance->SetFlow(finres);
  97   return balance;
  98 }
  99