python里pos_区块链中Python实现基于POS算法

区块链中的共识算法

在比特币公链结构解析中，提到了去中心化设计，它设计了一套共识协议并且通过协议来保证系统的稳定和放攻击性。目前接受最广的就是POW工作量证明算法。

POW共识算法取得了成功，但是质疑声也从未停过，最主要的就是电力消耗。据不完整统计，其消耗的电量是巨大的，甚至比绝大多数国家耗电还多，这就造成了资源的极大浪费，造成了算力的高度集中。

基于以上种种原因，更多的共识算法被提出来 POS、DPOS、BPFT等等。 今天我们就来认识POS算法。

POS算法，简称权益证明，简单理解就是拥有更多token的人，由更大的概率获得记账权力，获取奖励。

代码实战

生成一个Block

既然要实现POS算法，就要生成一条链，链又是一个个块生成的，那么先看看如何生成块。由于今天我们的核心是实现POS，所以关于块的生成，我们就用最简单的方式，好让大家聚焦在核心的内容上面。

我们用三个方法来实现生成一个合法的区块

calculate_hash 计算区块的hash值

is_block_valid 校验区块是否合法

generate_block 生成一个区块

from hashlib import sha256from datetime import datetimedef generate_block(oldblock, bpm, address)::param oldblock::param bpm::param address::return:newblock = {"Index": oldblock["Index"] + 1,"BPM": bpm,"Timestamp": str(datetime.now()),"PrevHash": oldblock["Hash"],"Validator": address}newblock["Hash"] = calculate_hash(newblock)return newblockdef calculate_hash(block):record = "".join([str(block["Index"]),str(block["BPM"]),block["Timestamp"],block["PrevHash"]])return sha256(record.encode()).hexdigest()def is_block_valid(newblock, oldblock)::param newblock::param oldblock::return:if oldblock["Index"] + 1 != newblock["Index"]:return Falseif oldblock["Hash"] != newblock["PrevHash"]:return Falseif calculate_hash(newblock) != newblock["Hash"]:return Falsereturn True

这是为了更灵活直接采用函数来实现块生成。

创建一个TCP服务器

由于需要权益证明算法来算则记账人，就需要在很多节点中选择记账人，也就是要一个server让节点连接，同时同步信息给节点，就需要一个TCP长连接。

Fromsocketserver import BaseRequestHandler, ThreadingTCPServerdef run():# start a tcp serverserv = ThreadingTCPServer(('', 9090), HandleConn)serv.serve_forever()

在这里用Python内库socketserver来创建一个TCPServer，当然注意的一定就是采用的多线程额创建方式。这样可以保证客户端同时连接，不至于堵塞。在这里server也有问题，就是由多少个客户端连接就会创建多少个线程，当然更好的方式就是创建一个线程池。

在我们创建TCPServer的时候，使用到了HandleConn，但是我们还没有定义，所以接下来我们就来定义一个HandleConn

消息处理器下面我们来实现Handler函数，Handler函数在跟Client Node通信的时候，需要我们的Node实现下面的功能

Node可以输入balance(token数量) 也就是股权数目

Node需要能够接收广播，方便Server同步区块以及记账人信息

添加自己到候选人名单 (候选人为持有token的人)

输入BPM生成Block

验证一个区块的合法性

感觉任务还是蛮多的，接下来我们看代码实现

import threadingfrom queue import Queue, Empty# 定义变量block_chain = []temp_blocks = []candidate_blocks = Queue()  # 创建队列，用于线程间通信announcements = Queue()validators = {}My_Lock = threading.Lock()class HandleConn(BaseRequestHandler):def handle(self):print("Got connection from", self.client_address)# validator addressself.request.send(b"Enter token balance:")balance = self.request.recv(8192)try:balance = int(balance)except Exception as e:print(e)t = str(datetime.now())address = sha256(t.encode()).hexdigest()validators[address] = balanceprint(validators)while True:announce_winner_t= threading.Thread(target=annouce_winner, args=(announcements, self.request,),daemon=True)announce_winner_t.start()self.request.send(b"\nEnter a new BPM:")bpm = self.request.recv(8192)try:bpm = int(bpm)except Exception as e:print(e)del validators[address]break# with My_Lock:last_block = block_chain[-1]new_block = generate_block(last_block, bpm, address)if is_block_valid(new_block, last_block):print("new block is valid!")candidate_blocks.put(new_block)self.request.send(b"\nEnter a new BPM:\n")annouce_blockchain_t= threading.Thread(target=annouce_blockchain, args=(self.request,), daemon=True)annouce_blockchain_t.start()

这段代码，可能有些难度，在这里我们采用了多线程的方式，同时为了能够让消息在线程间通信，我们使用了队列。 这里使用队列，为了更好的拓展。将多线程里面处理的任务拆分出去成独立的服务，然后用消息队列进行通信，就是一个简单的分布式系统啦。

由于这里有难度，所以代码还是讲一讲吧。

# validator addressself.request.send(b"Enter token balance:")balance = self.request.recv(8192)try:balance = int(balance)except Exception as e:print(e)t = str(datetime.now())address = sha256(t.encode()).hexdigest()validators[address] = balanceprint(validators)

这一段就是我们提到的Node 客户端添加自己到候选人的代码，每链接一个客户端，就会添加一个候选人。 这里我们用添加的时间戳的hash来记录候选人。 当然也可以用其他的方式，比如我们代码里面的client_address

announce_winner_t = threading.Thread(target=annouce_winner, args=(announcements, self.request,),daemon=True)announce_winner_t.start()def annouce_winner(announcements, request)::param announcements::param request::return:while True:try:msg = announcements.get(block=False)request.send(msg.encode())request.send(b'\n')except Empty:time.sleep(3)continue

然后接下来我们起了一个线程去广播获得记账权的节点信息到所有节点。

self.request.send(b"\nEnter a new BPM:")bpm = self.request.recv(8192)try:bpm = int(bpm)except Exception as e:print(e)del validators[address]break# with My_Lock:last_block = block_chain[-1]new_block = generate_block(last_block, bpm, address)if is_block_valid(new_block, last_block):print("new block is valid!")candidate_blocks.put(new_block)

根据节点输入的BPM值生成一个区块，并校验区块的有效性。 将有效的区块放到候选区块当中，等待记账人将区块添加到链上。

annouce_blockchain_t = threading.Thread(target=annouce_blockchain, args=(self.request,), daemon=True)annouce_blockchain_t.start()def annouce_blockchain(request)::param request::return:while True:time.sleep(30)with My_Lock:output = json.dumps(block_chain)try:request.send(output.encode())request.send(b'\n')except OSError:pass

最后起一个线程，同步区块链到所有节点。

看完了，节点跟Server交互的部分，接下来是最重要的部分，

POS算法实现

def pick_winner(announcements):"""选择记账人:param announcements::return:"""time.sleep(10)while True:with My_Lock:temp = temp_blockslottery_pool = []  #if temp:for block in temp:if block["Validator"] not in lottery_pool:set_validators = validatorsk = set_validators.get(block["Validator"])if k:for i in range(k):lottery_pool.append(block["Validator"])lottery_winner = choice(lottery_pool)print(lottery_winner)# add block of winner to blockchain and let all the other nodes knownfor block in temp:if block["Validator"] == lottery_winner:with My_Lock:block_chain.append(block)# write message in queue.msg = "\n{0} 赢得了记账权利\n".format(lottery_winner)announcements.put(msg)breakwith My_Lock:temp_blocks.clear()

这里我们用pick_winner 来选择记账权利，我们根据token数量构造了一个列表。 一个人获得记账权利的概率为：

p = mount['NodeA']/mount['All']

文字描述就是其token数目在总数中的占比。 比如总数有100个，他有10个，那么其获得记账权的概率就是0.1， 到这里核心的部分就写的差不多了，接下来，我们来添加节点，开始测试吧。

后面则是POS的记账方式我会在下面继续讲解，因为代码过于长了，嘿嘿。