message_formats.py

from Message import MessageFormat
from Node import *
from NodeInfo import *

def SET_PREV_handler(node, message):
    node.set_prev_info(NodeInfo(message.data['id'], message.data['ip']))
    log.info("prev_node set on " + node.myInfo.to_str())

def DHT_INSERT__KEY_SEARCH(node, message):
    node.find_proper_node_for_key(message.data['key'], message.data['ip'])

def DHT_INSERT__GET_VALUE(node, message):
    log.debug("DHT_INSERT__GET_VALUE call")
    return node.fetch_pending_value(message.data['key'])

DHT_LOOKUP__KEY_SEARCH = "DHT_LOOKUP__KEY_SEARCH"
def DHT_LOOKUP__KEY_SEARCH_handler(node, message):
    node.lookup_recursive(message.data['key'], message.data['ip'])

DHT_LOOKUP__FOUND_VAL = "DHT_LOOKUP__FOUND_VAL"
def DHT_LOOKUP__FOUND_VAL_handler(node, message):
    node.add_founded_value(message.data['key'], message.data['value'])

DHT_REMOVE__KEY_SEARCH = "DHT_REMOVE__KEY_SEARCH"
def DHT_REMOVE__KEY_SEARCH_handler(node, message):
    node.remove_recursive(message.data['key'])

# messages are json objects:
# { "title" : TITLE, "data" : DATA }
defaults = \
    {
        "SET_PREV": MessageFormat("SET_PREV").set_handler(SET_PREV_handler),
        "DHT_INSERT__KEY_SEARCH": MessageFormat("DHT_INSERT__KEY_SEARCH").set_handler(DHT_INSERT__KEY_SEARCH),
        "DHT_INSERT__GET_VALUE": MessageFormat("DHT_INSERT__GET_VALUE").set_handler(DHT_INSERT__GET_VALUE),
        DHT_LOOKUP__KEY_SEARCH : MessageFormat(DHT_LOOKUP__KEY_SEARCH).set_handler(DHT_LOOKUP__KEY_SEARCH_handler),
        DHT_LOOKUP__FOUND_VAL : MessageFormat(DHT_LOOKUP__FOUND_VAL).set_handler(DHT_LOOKUP__FOUND_VAL_handler),
        DHT_REMOVE__KEY_SEARCH : MessageFormat(DHT_REMOVE__KEY_SEARCH).set_handler(DHT_REMOVE__KEY_SEARCH_handler)
    }