blob: e4be526b0d972e655f054653e940089b16a78aca [file] [log] [blame]
#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
''' module for defining resources'''
class Resources(object):
"""Resources needed by the topology are encapsulated in this class.
Currently we deal with CPU and RAM. Others can be added later.
def __init__(self):
self._cpu = 0.0
self._ram = 0
def set_cpu(self, cpu):
self._cpu = float(cpu)
return self
def set_ram(self, ram):
self._ram = int(ram)
return self
def get_cpu(self):
return self._cpu
def get_ram(self):
return self._ram
def set_ram_in_mb(self, ram):
return self.set_ram(ram * 1024 * 1024)
def set_ram_in_gb(self, ram):
return self.set_ram_in_mb(ram * 1024)
def __repr__(self):
return 'Resource {cpu: %f, ram: %d}' % (self._cpu, self._ram)