def delete(file_name):
delete_local(file_name)
ip, port, timestamp, host_id = (None, None, None, None)
## 查询锁控制服务器，查看该文件是否被加互斥锁
with grpc.insecure_channel(str(LOCK_IP) + ':' + str(LOCK_POT)) as channel:
lock_client = LockClient(channel)
# 0 无锁, 1为共享锁,2 为互斥锁
result = lock_client.lock_exist_test(file_name)
if (result != 0):
return "Can't delete it! "
## 查询路径服务器获得文件所在的ip,port
with grpc.insecure_channel(str(DIRECTORY_IP) + ':' +
str(DIRECTORY_PORT)) as channel:
directory_client = DirectoryClient(channel)
tfile_name, ip, port, timestamp, host_id = directory_client.get_file_address(
file_name)
if tfile_name == '':
return "No such file!"
temp = directory_client.get_file_all_address(file_name)
timestamp = time.time()
if (not directory_client.del_file_address(file_name, timestamp)):
return "Unknown Error"
## 访问文件服务器，删除文件
for ip, port in temp:
with grpc.insecure_channel(str(ip) + ':' + str(port)) as channel:
file_client = FileClient(channel)
result = file_client.delete(file_name)
return True
def show_all():
## 查询路径服务器获得所有文件:
with grpc.insecure_channel(str(DIRECTORY_IP) + ':' +
str(DIRECTORY_PORT)) as channel:
directory_client = DirectoryClient(channel)
temp = directory_client.get_all_file()
print("-------------distribute system files-----------")
for file_name, timestamp, host_id in temp:
print('name:', file_name, 'timestamp:', timestamp, 'created:',
host_id)
for file_name in os.listdir(ROOT_PATH):
print('name:', file_name, 'created: local')
def read(file_name):
ip, port, timestamp, host_id = (None, None, None, None)
## 查询锁控制服务器，查看该文件是否被加互斥锁
with grpc.insecure_channel(str(LOCK_IP) + ':' + str(LOCK_POT)) as channel:
lock_client = LockClient(channel)
# 0 无锁, 1为共享锁,2 为互斥锁
result = lock_client.lock_exist_test(file_name)
if (result == 2):
return "Can't access this file, another client is writing!"
else:
## 给文件加共享锁,0表示共享锁，1表示互斥锁
if (lock_client.lock_file(file_name, 0)):
pass
else:
return "Unknown Error"
## 查询路径服务器获得文件所在的ip,port
with grpc.insecure_channel(str(DIRECTORY_IP) + ':' +
str(DIRECTORY_PORT)) as channel:
directory_client = DirectoryClient(channel)
file_name, ip, port, timestamp, host_id = directory_client.get_file_address(
file_name)
if file_name == '':
return "No such file"
#print(ip,":",port)
## 访问文件服务器，访问文件
result_str = ''
with grpc.insecure_channel(str(ip) + ':' + str(port)) as channel:
file_client = FileClient(channel)
#print(ip,":",port)
result_str = file_client.read(file_name)
## 释放共享锁
with grpc.insecure_channel(str(LOCK_IP) + ':' + str(LOCK_POT)) as channel:
lock_client = LockClient(channel)
## 给文件解锁,0表示共享锁，1表示互斥锁
if (lock_client.unlock_file(file_name, 0)):
pass
else:
return "Unknown Error"
return result_str
def upload(file_name):
'''
采用瞬时一致性通信模型
'''
## 判断文件是否存在，如果存在，不允许上传
with grpc.insecure_channel(str(DIRECTORY_IP) + ':' +
str(DIRECTORY_PORT)) as channel:
directory_client = DirectoryClient(channel)
tfile_name, ip, port, timestamp, host_id = directory_client.get_file_address(
file_name)
if tfile_name:
return False
t_file_name = os.path.join(ROOT_PATH, file_name)
with open(t_file_name, 'r', encoding='utf-8') as f:
result_str = f.read()
## 获取所有的文件服务器的地址
with grpc.insecure_channel(
str(DIRECTORY_IP) + ':' + str(DIRECTORY_PORT)) as channel:
directory_client = DirectoryClient(channel)
server_list = directory_client.get_all_server()
## 向所有的文件服务器写入该文件
for ip, port in server_list:
with grpc.insecure_channel(str(ip) + ':' + str(port)) as channel:
file_client = FileClient(channel)
if not file_client.upload(file_name, result_str):
print("Unknown Error from", ip, port)
else:
## 把写入服务器文件的更新消息反馈给路径服务器
with grpc.insecure_channel(
str(DIRECTORY_IP) + ':' +
str(DIRECTORY_PORT)) as channel:
directory_client = DirectoryClient(channel)
timestamp = time.time()
directory_client.add_file_address(
file_name, ip, port, timestamp)
return True