Initial YakPanel commit

class_v2/btdockerModelV2/monitorModel.py

@@ -0,0 +1,129 @@
+# coding: utf-8
+# -------------------------------------------------------------------
+# YakPanel
+# -------------------------------------------------------------------
+# Copyright (c) 2015-2099 YakPanel(www.yakpanel.com) All rights reserved.
+# -------------------------------------------------------------------
+# Author: zouhw <zhw@yakpanel.com>
+# -------------------------------------------------------------------
+
+# ------------------------------
+# Docker模型
+# ------------------------------
+import sys
+import threading
+import time
+
+sys.path.insert(0, "/www/server/panel/class_v2/")
+sys.path.insert(1, "/www/server/panel/")
+import public
+from btdockerModelV2 import dk_public as dp
+from btdockerModelV2 import containerModel as dc
+from btdockerModelV2 import statusModel as ds
+from btdockerModelV2 import imageModel as di
+from public.validate import Param
+
+class main:
+    __save_date = None
+    __day_sec = 86400
+
+    def __init__(self, save_date):
+        if not save_date:
+            self.__save_date = 30
+        else:
+            self.__save_date = save_date
+
+    def docker_client(self, url):
+        return dp.docker_client(url)
+
+    def get_all_host_stats(self, fun):
+        """
+        获取所有主机信息并获取该主机下的容器状态
+        :param fun: 需要调用的方法，用于获取并记录容器状态
+        :return:
+        """
+        hosts = dp.sql('hosts').select()
+        for i in hosts:
+            t = threading.Thread(target=fun, args=(i,))
+            t.setDaemon(True)
+            t.start()
+
+    # 获取所有docker容器的状态信息
+    def container_status_for_all_hosts(self):
+        """
+        获取所有服务器的容器数量
+        :return:
+        """
+        # while True:
+        args = public.to_dict_obj({})
+        container_list = dc.main().get_list(args)['message']
+        for c in container_list['container_list']:
+            args.id = c['id']
+            args.write = 1
+            args.save_date = self.__save_date
+            ds.main().stats(args)
+            # time.sleep(60)
+
+    # 获取所有服务器的容器数量
+    def container_count(self):
+        # while True:
+        hosts = dp.sql('hosts').select()
+        n = 0
+        for i in hosts:
+            args = public.to_dict_obj({})
+            args.url = i['url']
+            container_list = dc.main().get_list(args)['message']
+            n += len(container_list)
+        pdata = {
+            "time": int(time.time()),
+            "container_count": n
+        }
+        expired = time.time() - (self.__save_date * self.__day_sec)
+        dp.sql("container_count").where("time<?", (expired,)).delete()
+        dp.sql("container_count").insert(pdata)
+        # time.sleep(60)
+
+    def image_for_all_host(self):
+        # while True:
+        hosts = dp.sql('hosts').select()
+        num = 0
+        size = 0
+        for i in hosts:
+            args = public.to_dict_obj({})
+            args.url = i['url']
+            res = di.main().image_for_host(args)['message']
+            num += res['num']
+            size += res['size']
+        pdata = {
+            "time": int(time.time()),
+            "num": num,
+            "size": int(size)
+        }
+        expired = time.time() - (self.__save_date * self.__day_sec)
+        dp.sql("image_infos").where("time<?", (expired,)).delete()
+        dp.sql("image_infos").insert(pdata)
+        # time.sleep(60)
+
+
+def monitor():
+    # 获取所有容器信息
+    while True:
+        save_date = dp.docker_conf()['SAVE']
+        m = main(save_date)
+        m.get_all_host_stats(m.container_status_for_all_hosts)
+        # 开始获取容器总数
+        t = threading.Thread(target=m.container_count)
+        t.setDaemon(True)
+        t.start()
+        # 获取镜像详情
+        t = threading.Thread(target=m.image_for_all_host)
+        t.setDaemon(True)
+        t.start()
+        time.sleep(60)
+    # condition=threading.Condition()
+    # condition.acquire()
+    # condition.wait()
+
+
+if __name__ == "__main__":
+    monitor()