def __init__(self): """get nodes""" self.malist=[] self.macnum=0 conf=Configuration() self.malist=conf.get_list() self.macnum=conf.get_count() print "nodelist:",self.malist self.hring=HashRing(self.malist)
class Client:
def __init__(self):
"""get nodes"""
self.malist=[]
self.macnum=0
conf=Configuration()
self.malist=conf.get_list()
self.macnum=conf.get_count()
print "nodelist:",self.malist
self.hring=HashRing(self.malist)
def write(self,src):
"""write src file into node"""
print "File is",src
toIP,toPort=self.src_to_node(src)
print "Send to",toIP,toPort
self.send(src,toIP,toPort)
def src_to_node(self,src):
des=self.hring.get_node_pos(src)
iPort=des.split(':')
return iPort[0],iPort[1]
def send(self,src,ip,port):
clsoc=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
clsoc.connect((ip,int(port)))
fp=open(src,'rb')
formStr="WRITFILE,%04d"%len(src)
print "formStr",formStr,len(formStr)
clsoc.send(formStr)
resdata=clsoc.recv(1024)
if resdata.startswith('OK'):
print "OK"
print "sending....",src
clsoc.send(src)
print "sending data...."
while True:
data=fp.read(4096)
if not data:
break
while len(data)>0:
sent=clsoc.send(data)
data=data[sent:]
print "Fished to send ",src
fp.close()
def fetch(self,src):
toIP,toPort=self.src_to_node(src)
print "fetch from",toIP,toPort
self.get(src,toIP,toPort)
def get(self,src,ip,port):
clsoc=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
clsoc.connect((ip,int(port)))
formStr="FETCFILE,%04d"%len(src)
print "formStr",formStr
clsoc.send(formStr)
resdata=clsoc.recv(1024)
if resdata.startswith('OK'):
print "OK"
print "sending....",src
clsoc.send(src)
print "fetching data...."
ffile=src[src.rindex('\\')+1:]
fp=open(ffile,'wb')
while True:
data=clsoc.recv(1024)
if not data:break
fp.write(data)
print "fetching",data[-8:]
fp.flush()
fp.close()
print "finished!",src