Create BlockChain_base.py

BlockChain_base.py

@@ -0,0 +1,101 @@
+import hashlib
+import json
+from time import time
+
+class Blockchain:
+    def __init__(self):
+        self.chain = []
+        self.current_transactions = []
+        self.user_wallets = {}
+        self.user_gpus = {}
+
+        # Genesis 블록 생성
+        self.new_block(previous_hash="1")
+
+    def new_block(self, previous_hash=None):
+        """
+        블록체인에 새로운 블록 추가
+        :param previous_hash: 이전 블록의 해시 값
+        :return: 새로 생성된 블록
+        """
+        block = {
+            'index': len(self.chain) + 1,
+            'timestamp': time(),
+            'transactions': self.current_transactions,
+            'previous_hash': previous_hash or self.hash(self.chain[-1]),
+        }
+        
+        # 현재 트랜잭션 초기화
+        self.current_transactions = []
+        
+        # 블록을 체인에 추가
+        self.chain.append(block)
+        return block
+
+    def new_transaction(self, id, kind, data):
+        """
+        새로운 트랜잭션 생성 및 추가
+        id: 요청자
+        kind: 요청 종류 (inference, add, out)
+        data: inference 시 [입력 prompt, output], peer add 시 gpu mem, out 시 또한 gpu mem
+        
+        return: 해당 트랜잭션을 포함한 블록의 인덱스
+        """
+        transaction = {
+            'id': id,
+            'kind': kind,
+            'data': data,
+        }
+        self.current_transactions.append(transaction)
+
+        # 각 유저의 행동력 업데이트
+        if kind == "inference":
+            if id not in self.user_wallets:
+                # 새로운 유저인 경우 초기 행동력 설정
+                self.user_wallets[id] = 10  # 초기 행동력을 10으로 설정
+            else:
+                # inference 요청 시 차감
+                self.user_wallets[id] -= 1
+        elif kind == "add":
+            if id not in self.user_gpus:
+                self.user_gpus[id] = int(data)
+            else:
+                self.user_gpus[id] += int(data)
+        elif kind == "out":
+            if id in self.user_gpus:
+                self.user_gpus[id] -= int(data)
+                if self.user_gpus[id] < 0:
+                    del(self.user_gpus[id])
+
+        return self.last_block['index'] + 1
+
+    def get_user_balance(self, id):
+        # 특정 유저의 행동력을 조회
+        return self.user_wallets.get(id, 0)
+
+    def get_user_gpu_mem(self, id):
+        # 특정 유저의 기여 gpu를 조회
+        return self.user_gpus.get(id, 0)
+
+    def get_total_gpu_mem(self):
+        # 전체 메모리 반환
+        result = 0
+        for mem in self.user_gpus.values():
+            result += int(mem)
+        return result
+    @property
+    def last_block(self):
+        """
+        체인의 마지막 블록 반환
+        """
+        return self.chain[-1]
+    
+    @staticmethod
+    def hash(block):
+        """
+        블록을 SHA-256으로 해싱
+        :param block: 블록
+        :return: 해시 값
+        """
+        block_string = json.dumps(block, sort_keys=True).encode()
+        return hashlib.sha256(block_string).hexdigest()