storage/base/aaBalanceUpdate.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) 2008 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 "xpcom-config.h"
  23
  24 #include "nsCOMPtr.h"
  25 #include "nsComponentManagerUtils.h"
  26 #include "nsXPCOMCID.h"
  27 #include "nsIMutableArray.h"
  28 #include "nsISupportsPrimitives.h"
  29
  30 /* Unfrozen API */
  31
  32 /* Project includes */
  33 #include "aaAmountUtils.h"
  34 #include "aaIBalance.h"
  35
  36 #include "aaBalanceUpdate.h"
  37
  38 static const char *sql =
  39 "UPDATE balance SET side = ?, amount = ?, value = ? WHERE id == ?";
  40
  41 aaBalanceUpdate::aaBalanceUpdate()
  42 {
  43   mSql = sql;
  44 }
  45
  46 NS_IMPL_ISUPPORTS_INHERITED0(aaBalanceUpdate,
  47                              aaSaveRequest)
  48
  49 /* aaSaveRequest */
  50 NS_IMETHODIMP
  51 aaBalanceUpdate::SetFilter(nsISupports * aFilter)
  52 {
  53   nsresult rv;
  54   mBalance = do_QueryInterface(aFilter, &rv);
  55   NS_ENSURE_SUCCESS(rv, rv);
  56
  57   return NS_OK;
  58 }
  59
  60 NS_IMETHODIMP
  61 aaBalanceUpdate::GetParams(nsIArray * * aParams)
  62 {
  63   NS_ENSURE_TRUE(mStatement, NS_ERROR_NOT_INITIALIZED);
  64   NS_ENSURE_TRUE(mBalance, NS_ERROR_NOT_INITIALIZED);
  65   nsresult rv;
  66
  67   PRInt64 id;
  68   rv = mBalance->GetId(&id);
  69   NS_ENSURE_SUCCESS(rv, rv);
  70
  71   nsCOMPtr<nsIMutableArray> result
  72     = do_CreateInstance(NS_ARRAY_CONTRACTID, &rv);
  73   NS_ENSURE_SUCCESS(rv, rv);
  74
  75   nsCOMPtr<nsISupportsPRInt64> valInt64;
  76   nsCOMPtr<nsISupportsPRInt32> valInt32;
  77   nsCOMPtr<nsISupportsDouble> valDouble;
  78
  79   valInt32 = do_CreateInstance(NS_SUPPORTS_PRINT32_CONTRACTID, &rv);
  80   NS_ENSURE_SUCCESS(rv, rv);
  81   rv = valInt32->SetData(mBalance->PickSide());
  82   NS_ENSURE_SUCCESS(rv, rv);
  83   rv = result->InsertElementAt(valInt32, 0, PR_FALSE);
  84   NS_ENSURE_SUCCESS(rv, rv);
  85
  86   valDouble = do_CreateInstance(NS_SUPPORTS_DOUBLE_CONTRACTID, &rv);
  87   NS_ENSURE_SUCCESS(rv, rv);
  88   rv = valDouble->SetData(mBalance->PickAmount());
  89   NS_ENSURE_SUCCESS(rv, rv);
  90   rv = result->InsertElementAt(valDouble, 1, PR_FALSE);
  91   NS_ENSURE_SUCCESS(rv, rv);
  92
  93   valDouble = do_CreateInstance(NS_SUPPORTS_DOUBLE_CONTRACTID, &rv);
  94   NS_ENSURE_SUCCESS(rv, rv);
  95   rv = valDouble->SetData(mBalance->PickValue());
  96   NS_ENSURE_SUCCESS(rv, rv);
  97   rv = result->InsertElementAt(valDouble, 2, PR_FALSE);
  98   NS_ENSURE_SUCCESS(rv, rv);
  99
 100   valInt64 = do_CreateInstance(NS_SUPPORTS_PRINT64_CONTRACTID, &rv);
 101   NS_ENSURE_SUCCESS(rv, rv);
 102   rv = valInt64->SetData(id);
 103   NS_ENSURE_SUCCESS(rv, rv);
 104   rv = result->InsertElementAt(valInt64, 3, PR_FALSE);
 105   NS_ENSURE_SUCCESS(rv, rv);
 106
 107   *aParams = result;
 108   result.forget();
 109
 110   return NS_OK;
 111 }