storage/aaLoadPendingFacts.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/aaIFlow.h>
  36 #include <abstract/base/aaIEvent.h>
  37 #include <abstract/base/aaIFact.h>
  38 #include <abstract/storage/aaILoadQuery.h>
  39 #include "aaLoadPendingFacts.h"
  40
  41 #define AA_LOADPENDINGFACTS_SELECT_QUERY "SELECT strftime('%s',transfer.day),\
  42   transfer.event_id, transfer.id, fromF.flow_id, toF.flow_id, transfer.amount\
  43   FROM transfer\
  44   LEFT JOIN txn ON transfer.day == txn.day AND transfer.event_id == txn.event_id\
  45     AND transfer.id == txn.transfer_id\
  46   LEFT JOIN fact AS fromF ON transfer.day == fromF.day AND transfer.event_id\
  47     == fromF.event_id AND transfer.id == fromF.transfer_id AND\
  48     fromF.side == 1\
  49   LEFT JOIN fact AS toF ON transfer.day == toF.day AND transfer.event_id ==\
  50     toF.event_id AND transfer.id == toF.transfer_id AND toF.side == 0\
  51   WHERE txn.value IS NULL"
  52
  53 aaLoadPendingFacts::aaLoadPendingFacts(nsISupports *aOuter)
  54   :aaLoadQuery(aOuter)
  55 {
  56 }
  57
  58 aaLoadPendingFacts::~aaLoadPendingFacts()
  59 {
  60 }
  61
  62 NS_IMPL_ISUPPORTS_INHERITED0(aaLoadPendingFacts,
  63                              aaLoadQuery)
  64
  65 /* Virtual methods */
  66 const char*
  67 aaLoadPendingFacts::getSelectQuery() const
  68 {
  69   return AA_LOADPENDINGFACTS_SELECT_QUERY;
  70 }
  71
  72 aaIDataNode*
  73 aaLoadPendingFacts::fetchRow()
  74 {
  75   nsresult rv;
  76   PRInt64 id, eventId, fromId, toId;
  77   PRTime time;
  78   double sum;
  79
  80   rv = mStatement->GetInt64(0, &time);
  81   if (NS_FAILED(rv))
  82     return nsnull;
  83   time *= 1000000;
  84
  85   rv = mStatement->GetInt64(1, &eventId);
  86   if (NS_FAILED(rv))
  87     return nsnull;
  88
  89   nsCOMPtr<aaIEvent> event = do_CreateInstance("@aasii.org/base/event;1");
  90   if (NS_UNLIKELY( ! event ))
  91     return nsnull;
  92   event->SetTime( time );
  93   event->SetId( eventId );
  94
  95   rv = mStatement->GetInt64(2, &id);
  96   NS_ENSURE_SUCCESS(rv, nsnull);
  97
  98   nsCOMPtr<aaIFlow> fromFlow(do_CreateInstance("@aasii.org/base/flow;1"));
  99   nsCOMPtr<aaIFlow> toFlow(do_CreateInstance("@aasii.org/base/flow;1"));
 100   NS_ENSURE_TRUE(fromFlow && toFlow, nsnull);
 101
 102   rv = mStatement->GetInt64(3, &fromId);
 103   NS_ENSURE_SUCCESS(rv, nsnull);
 104   fromFlow->SetId(fromId);
 105
 106   rv = mStatement->GetInt64(4, &toId);
 107   NS_ENSURE_SUCCESS(rv, nsnull);
 108   toFlow->SetId(toId);
 109
 110   rv = mStatement->GetDouble(5, &sum);
 111   NS_ENSURE_SUCCESS(rv, nsnull);
 112
 113   aaIFact *fact;
 114   CallCreateInstance("@aasii.org/base/fact;1", nsnull, &fact);
 115   if (! fact)
 116     return nsnull;
 117
 118   fact->SetId(id);
 119   fact->SetEvent(event);
 120   fact->SetTakeFrom(fromFlow);
 121   fact->SetGiveTo(toFlow);
 122   fact->SetAmount(sum);
 123   return fact;
 124 }
 125