14e59956创建于 2020年7月20日历史提交
#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Copyright (c) 2019 Huawei Technologies Co., Ltd.
# A-Tune is licensed under the Mulan PSL v2.
# You can use this software according to the terms and conditions of the Mulan PSL v2.
# You may obtain a copy of Mulan PSL v2 at:
#     http://license.coscl.org.cn/MulanPSL2
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR
# PURPOSE.
# See the Mulan PSL v2 for more details.
# Create: 2019-10-29

"""
Used to create tasks for optimizer.
"""


class TasksCache:
    """used to create tasks for optimizer"""
    tasks = {}

    __instance = None

    def __new__(cls, *args, **kwargs):
        if cls.__instance is None:
            cls.__instance = super(TasksCache, cls).__new__(cls, *args, **kwargs)
        return cls.__instance

    @classmethod
    def get_instance(cls):
        """get instance"""
        if cls.__instance is None:
            cls.__instance = TasksCache()
        return cls.__instance

    def get(self, key):
        """get task according to key"""
        return self.tasks.get(key) if key in self.tasks.keys() else None

    def set(self, key, value):
        """set task according to key"""
        self.tasks[key] = value

    def delete(self, key):
        """delete task according to key"""
        del self.tasks[key]

    def get_all(self):
        """get all task"""
        return self.tasks