def UpdateFile(self, request, context):
container_id = request.container_id
path = request.path
file_name = request.file_name
old_version = request.old_version
exit_code, old_file_stat = ContainerUtils.ExecCommand(\
container_id=container_id,
exec_cmd = ["stat", "-c", "%Y", path+"/"+file_name]
)
if exit_code == 0:
if old_version == old_file_stat or request.force:
# generate_data = GenerateData(request)
if not ContainerUtils.PutArchive(container_id, path,
request.data):
return Manager.UpdateFile_Response(
exit_code=Manager.UpdateFile_Response.ExitCode.
UNKNOWN_ERROR)
else:
exit_code, new_version = ContainerUtils.ExecCommand(\
container_id=container_id,
exec_cmd = ["stat", "-c", "%Y", path]
)
return Manager.UpdateFile_Response(\
exit_code = Manager.UpdateFile_Response.ExitCode.SUCCESS,
new_version = new_version
)
else:
return Manager.UpdateFile_Response(\
exit_code=Manager.UpdateFile_Response.ExitCode.MTIME_SYNC_ERROR)
elif exit_code == 1:
return Manager.UpdateFile_Response(\
exit_code=Manager.UpdateFile_Response.ExitCode.FILE_IS_NOT_EXIST)
def ListFile(self, request, context):
cmd = ["ls", "-Al", request.path]
exit_code, output = ContainerUtils.ExecCommand(request.container_id,
cmd)
response = Manager.ListFile_Response()
if exit_code == 0:
response.exit_code = Manager.ListFile_Response.ExitCode.SUCCESS
List = output.strip().split('\n')[1:]
for file in List:
temp = file.split(' ')
stat = temp[0]
filename = temp[-1]
temp = response.files.add()
temp.file_name = filename
if stat[0] == '-':
temp.file_type = Manager.FileStat.FileType.FILE
elif stat[0] == 'd':
temp.file_type = Manager.FileStat.FileType.FOLDER
elif stat[0] == 'i':
temp.file_type = Manager.FileStat.FileType.LINK
elif exit_code == 1:
response.exit_code = Manager.ListFile_Response.ExitCode.MINOR_PROBLEMS
elif exit_code == 2:
response.exit_code = Manager.ListFile_Response.ExitCode.SERIOUS_TROUBLE
return response
def CreateContainer(self, request, context):
container_id = ContainerUtils.CreateContainer(
image_id=request.image_id,
mount_path="/workplace/{username}/mount".format(
username=request.username),
hostname=request.username,
container_name=request.container_name)
return Manager.CreateContainer_Response(container_id=container_id)
def test_BuildImage():
with grpc.insecure_channel('0.0.0.0:8666') as channel:
# 客户端通过stub来实现rpc通信
stub = Manager_grpc.ManagerStub(channel)
file = open("dockerfile", "rb")
dockerfile = file.read()
request = Manager.BuildImage_Request(dockerfile=dockerfile,
tag="test:v2.0")
# 客户端必须使用定义好的类型，这里是HelloRequest类型
response = stub.BuildImage(request=request)
print("hello client received: ", response.exit_code)
def ListContainers(self, request, context):
containers = ContainerUtils.ListContainers()
response = Manager.ListContainers_Response()
for container in containers:
attr = response.containers.add()
attr.id = container['Id']
attr.created = container['Created']
attr.status = GetContainerStatus(container['Status'])
attr.image = container['Image']
attr.name = container['Name']
return response
def PullImage(self, request, context):
if request.tag == '':
request.tag = 'latest'
auth_config = None
if request.auth_config is not None:
auth_config = {}
auth_config['username'] = request.auth_config.username
auth_config['password'] = request.auth_config.password
ImageUtils.PullImage(repository=request.repository,
tag=request.tag,
auth_config=auth_config)
return Manager.PullImage_Response()
def ListImages(self, request, context):
List = ImageUtils.ListImages()
response = Manager.ListImages_Response()
for image in List:
attr = response.images.add()
attr.id = image['Id']
for tag in image['RepoTags']:
attr.repoTags.append(tag)
attr.created = image['Created']
attr.size = int(image['Size'])
attr.author = image['Author']
return response
def GetFile():
with grpc.insecure_channel('localhost:8666') as channel:
# 客户端通过stub来实现rpc通信
stub = Manager_grpc.ContainerManagerStub(channel)
request = Manager.GetFile_Request(\
container_id = 'container1',
path = '/workplace/test.py'
)
# 客户端必须使用定义好的类型，这里是HelloRequest类型
response = stub.GetFile(request=request)
file = open("test.tar", "wb+")
for Data in response:
file.write(Data.data)
file.close()
def GetContainer(self, request, context):
response = Manager.GetContainer_Response()
try:
container = ContainerUtils.GetContainer(request.container_id)
except docker.errors.APIError:
response.exit_code = Manager.GetContainer_Response.ExitCode.NOTFOUND
return response
response.exit_code = Manager.GetContainer_Response.ExitCode.SUCCESS
response.container_attr.id = container['Id']
response.container_attr.created = container['Created']
response.container_attr.status = GetContainerStatus(container['State'])
response.container_attr.image = container['Image']
response.container_attr.name = container['Name']
return response
def GetImage(self, request, context):
response = Manager.GetImage_Response()
try:
image = ImageUtils.GetImage(image_id=request.image_id)
except docker.errors.ImageNotFound:
response.exit_code = Manager.GetImage_Response.ExitCode.NOTFOUND
return response
response.exit_code = Manager.GetImage_Response.ExitCode.SUCCESS
response.image_attr.id = image['Id']
for tag in image['RepoTags']:
temp = response.image_attr.repoTags.add()
temp = tag
response.image_attr.created = image['Created']
response.image_attr.size = int(image['Size'])
response.image_attr.author = image['Author']
return response
def LoadImage(self, request, context):
g = GenerateData(request)
response = Manager.LoadImage_Response()
try:
List = ImageUtils.LoadImage(g)
except docker.errors.ImageLoadError:
response.exit_code = Manager.LoadImage_Response.ExitCode.ERROR
return response
response.exit_code = Manager.LoadImage_Response.ExitCode.SUCCESS
for image_obj in List:
image = ImageUtils.GetImage(image_obj.id)
image_attr = response.image_attr.add()
for tag in image['RepoTags']:
temp = image_attr.repoTags.add()
temp = tag
image_attr.created = image['Created']
image_attr.size = int(image['Size'])
image_attr.author = image['Author']
return response
def BuildImage(self, request, context):
response = Manager.BuildImage_Response()
dockerfile = TemporaryFile()
dockerfile.write(request.dockerfile)
# print("-------")
dockerfile.seek(0)
image_obj = ImageUtils.BuildImageByFile(dockerfile, request.tag)
# try:
# except docker.errors.APIError:
# response.exit_code = Manager.BuildImage_Response.ExitCode.ERROR
# return response
response.exit_code = Manager.BuildImage_Response.ExitCode.SUCCESS
image = ImageUtils.GetImage(image_obj.id)
response.image_attr.id = image['Id']
for tag in image['RepoTags']:
response.image_attr.repoTags.append(tag)
response.image_attr.created = image['Created']
response.image_attr.size = int(image['Size'])
response.image_attr.author = image['Author']
return response
def GetFile(self, request, context):
bits, stat = ContainerUtils.GetArchive(request.container_id,
request.path)
for data in bits:
yield Manager.GetFile_Response(data=data)
def RestartContainer(self, request, context):
ContainerUtils.RestartContainer(request.container_id)
return Manager.RestartContainer_Response()
def RemoveContainer(self, request, context):
ContainerUtils.RemoveContainer(request.container_id, request.force)
return Manager.RemoveContainer_Response()
def StopContainer(self, request, context):
ContainerUtils.StopContainer(request.container_id)
return Manager.StopContainer_Response()
def PruneImages(self, request, context):
ImageUtils.PruneImages()
return Manager.PruneImages_Response()
def PruneContainers(self, request, context):
ContainerUtils.PruneContainers()
return Manager.PruneContainers_Response()
def RemoveImage(self, request, context):
ImageUtils.RemoveImage(image_id=request.image_id, force=request.force)
return Manager.RemoveImage_Response()