From 51c31190358d5e217b61e0c0e3a57275c7707760 Mon Sep 17 00:00:00 2001
From: sinev-valentine <37595780+sinev-valentine@users.noreply.github.com>
Date: Thu, 11 Nov 2021 20:02:10 +0300
Subject: [PATCH] #260 write cost to DB (#277)

* add cost calc

* test update

* update cost log

* fix merge

* gas_used added

* add trx details

* fix

* fix cost calc

* extra trx impl

* reason impl

* impl

* delete test_operator_spending.py

* apply comments

* add test

* apply comment

* log cost to db

* init

* impl

* fix import solana_utils

* comment operator cost

* add depends_on to docker-compose

* fix build

* apply comments
---
 proxy/docker-compose-test.yml         |  2 ++
 proxy/indexer/sql_dict.py             |  2 +-
 proxy/plugin/solana_rest_api_tools.py | 30 +++++++++++++++++----------
 3 files changed, 22 insertions(+), 12 deletions(-)

diff --git a/proxy/docker-compose-test.yml b/proxy/docker-compose-test.yml
index 55da74629..ba33fd034 100644
--- a/proxy/docker-compose-test.yml
+++ b/proxy/docker-compose-test.yml
@@ -46,6 +46,8 @@ services:
       - "9090"
     networks:
       - net
+    depends_on:
+      - postgres
 
 networks:
   net:
diff --git a/proxy/indexer/sql_dict.py b/proxy/indexer/sql_dict.py
index 945ce0a63..194b75a4c 100644
--- a/proxy/indexer/sql_dict.py
+++ b/proxy/indexer/sql_dict.py
@@ -5,7 +5,7 @@
 
 POSTGRES_DB = os.environ.get("POSTGRES_DB", "neon-db")
 POSTGRES_USER = os.environ.get("POSTGRES_USER", "neon-proxy")
-POSTGRES_PASSWORD = os.environ.get("POSTGRES_PASSWORD", "neon-proxy")
+POSTGRES_PASSWORD = os.environ.get("POSTGRES_PASSWORD", "neon-proxy-pass")
 POSTGRES_HOST = os.environ.get("POSTGRES_HOST", "localhost")
 
 try:
diff --git a/proxy/plugin/solana_rest_api_tools.py b/proxy/plugin/solana_rest_api_tools.py
index 087ca3393..80d225ee6 100644
--- a/proxy/plugin/solana_rest_api_tools.py
+++ b/proxy/plugin/solana_rest_api_tools.py
@@ -30,6 +30,7 @@
 from web3.auto import w3
 from proxy.environment import neon_cli, evm_loader_id, ETH_TOKEN_MINT_ID, COLLATERAL_POOL_BASE, read_elf_params
 from .eth_proto import Trx
+from ..indexer.sql_dict import SQLDict
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
@@ -806,6 +807,14 @@ def simulate_continue(signer, client, perm_accs, trx_accs, step_count):
     return (continue_count, step_count)
 
 
+class CostSingleton(object):
+    def __new__(cls):
+        if not hasattr(cls, 'instance'):
+            cls.instance = super(CostSingleton, cls).__new__(cls)
+            cls.instance.operator_cost = SQLDict(tablename="operator_cost")
+        return cls.instance
+
+
 def update_transaction_cost(receipt, eth_trx, extra_sol_trx=False, reason=None):
     cost = receipt['result']['meta']['preBalances'][0] - receipt['result']['meta']['postBalances'][0]
     if eth_trx:
@@ -835,17 +844,16 @@ def update_transaction_cost(receipt, eth_trx, extra_sol_trx=False, reason=None):
                     used_gas = base58.b58decode(event['data'])[2:10]
                     used_gas = int().from_bytes(used_gas, "little")
 
-    logger.debug("COST %s %d %d %s %s %s %s %s",
-                 hash,
-                 cost,
-                 used_gas if used_gas else 0,
-                 sender,
-                 to_address,
-                 sig,
-                 "extra" if extra_sol_trx else "ok",
-                 reason if reason else "None",
-                 )
-
+    table = CostSingleton()
+    table.operator_cost[hash] = {
+        'cost': cost,
+        'used_gas': used_gas if used_gas else 0,
+        'sender': sender,
+        'to_address': to_address,
+        'sig': sig,
+        'status': 'extra' if extra_sol_trx else 'ok',
+        'reason':  reason if reason else ''
+    }
 
 def create_account_list_by_emulate(signer, client, eth_trx):