#!/usr/bin/python
|
# -*- coding: GBK -*-
|
#-------------------------------------------------------------------------------
|
#
|
##@package ObjPool
|
#
|
# @todo:¶ÔÏó³Ø¹ÜÀí
|
# @author hxp
|
# @date 2025-08-06
|
# @version 1.0
|
#
|
# ÏêϸÃèÊö: ¶ÔÏó³Ø¹ÜÀí£¬½¨Òé¶ÔƵ·±´´½¨µÄÀàʹÓöÔÏó³Ø£¬Èç·â°üÏà¹Ø¡¢Õ½¶·Ïà¹Ø
|
#
|
#-------------------------------------------------------------------------------
|
#"""Version = 2025-08-06 18:30"""
|
#-------------------------------------------------------------------------------
|
|
import GameWorld
|
import PyGameData
|
import collections
|
import types
|
#import sys
|
#import ctypes
|
|
class RecursiveObjectPoolManager(object):
|
|
def __init__(self):
|
self._pools = {} # ´æ´¢¸÷ÀàµÄ¶ÔÏó³Ø {class: pool}
|
# ʹÓöÔÏóID×÷Ϊ¼üµÄÓ³É䣺{id(obj): obj_class}
|
self._obj_id_to_pool = {}
|
self._releasing = set() # ÕýÔÚÊͷŵĶÔÏóID¼¯ºÏ
|
|
def create_pool(self, obj_class, max_size=10):
|
"""Ϊָ¶¨Àà´´½¨¶ÔÏó³Ø
|
:param obj_class: Òª¹ÜÀíµÄ¶ÔÏóÀà
|
:param max_size: ¶ÔÏó³Ø×î´óÈÝÁ¿
|
"""
|
if obj_class in self._pools:
|
GameWorld.Log("Pool for %s already exists" % obj_class.__name__)
|
return
|
GameWorld.Log("create_pool %s, max_size=%s" % ( obj_class.__name__, max_size))
|
|
self._pools[obj_class] = {
|
'max_size': max_size,
|
'free_list': [], # ¿ÕÏжÔÏóÁбí
|
'active_list': [], # ʹÓÃÖжÔÏóÁбí
|
'created_count': 0 # ÒÑ´´½¨¶ÔÏó¼ÆÊý
|
}
|
return self
|
|
def acquire(self, obj_class, *args, **kwargs):
|
"""»ñÈ¡¶ÔÏ󲢼ǼÆäËùÊô³Ø"""
|
if obj_class not in self._pools:
|
# Èç¹û³Ø²»´æÔÚ£¬×Ô¶¯´´½¨´óСΪ0µÄÎÞÏÞÖƳØ
|
self.create_pool(obj_class, 0)
|
|
pool = self._pools[obj_class]
|
|
if pool['free_list']:
|
# ¸´ÓÿÕÏжÔÏó
|
obj = pool['free_list'].pop()
|
# ÖØгõʼ»¯¶ÔÏó
|
obj.__init__(*args, **kwargs)
|
else:
|
# ´´½¨Ð¶ÔÏó
|
if pool['max_size'] <= 0 or pool['created_count'] < pool['max_size']:
|
obj = obj_class(*args, **kwargs)
|
pool['created_count'] += 1
|
# ¼Ç¼¶ÔÏ󵽳صÄÓ³Éä
|
obj_id = id(obj)
|
self._obj_id_to_pool[obj_id] = obj_class
|
else:
|
GameWorld.ErrLog("%s pool exhausted" % obj_class.__name__)
|
return None # ·µ»ØNone±íʾ»ñȡʧ°Ü
|
|
pool['active_list'].append(obj)
|
return obj
|
|
def release(self, obj):
|
"""ÊͷŶÔÏ󲢵ݹéÊÍ·ÅÆäǶÌ׶ÔÏó³Ø¶ÔÏó"""
|
obj_id = id(obj)
|
|
# ¼ì²éÊÇ·ñÕýÔڵݹéÊÍ·ÅÖÐ
|
if obj_id in self._releasing:
|
return
|
|
# Ö»´¦Àí¶ÔÏó³Ø¹ÜÀíµÄ¶ÔÏó
|
if obj_id not in self._obj_id_to_pool:
|
return
|
|
# ±ê¼ÇΪÕýÔÚÊÍ·Å
|
self._releasing.add(obj_id)
|
visited = set([obj_id]) # ³õʼ»¯ÒÑ·ÃÎʼ¯ºÏ
|
|
try:
|
# 1. µÝ¹éÊÍ·ÅǶÌ׶ÔÏó³Ø¶ÔÏó
|
self._recursive_release(obj, visited)
|
# 2. Êͷŵ±Ç°¶ÔÏó
|
self._return_to_pool(obj)
|
finally:
|
# ÒƳýÊͷűê¼Ç
|
self._releasing.discard(obj_id)
|
|
def _recursive_release(self, obj, visited):
|
"""µÝ¹éÊÍ·ÅǶÌ׶ÔÏó³Ø¶ÔÏó£¨Éî¶ÈÓÅÏÈ£©"""
|
# »ñÈ¡¶ÔÏóµÄËùÓÐÒýÓ㨲»°üº¬»ù´¡ÀàÐÍ£©
|
for ref in self._get_references(obj):
|
try:
|
ref_id = id(ref)
|
except TypeError:
|
# Èç¹û¶ÔÏ󲻿ɹþÏ££¬Ìø¹ý
|
continue
|
|
if ref_id in visited:
|
continue
|
visited.add(ref_id)
|
|
# Ö»´¦Àí¶ÔÏó³Ø¹ÜÀíµÄ¶ÔÏó
|
if ref_id in self._obj_id_to_pool:
|
# ÏȵݹéÊÍ·ÅǶÌ׶ÔÏó
|
self._recursive_release(ref, visited)
|
# ÔÙÊͷŵ±Ç°ÒýÓöÔÏó
|
self._return_to_pool(ref)
|
else:
|
# ¶ÔÓڷǶÔÏó³Ø¹ÜÀíµÄ¶ÔÏ󣬼ÌÐøµÝ¹é²éÕÒÆäÖеĶÔÏó³Ø¶ÔÏó
|
self._recursive_release(ref, visited)
|
|
def _get_references(self, obj):
|
"""°²È«»ñÈ¡¶ÔÏóµÄËùÓÐÖ±½ÓÒýÓã¬Ö§³Ö¸´ÔÓÊý¾Ý½á¹¹"""
|
refs = []
|
obj_type = type(obj)
|
|
# ´¦Àí×Ô¶¨Òå¶ÔÏóµÄÊôÐÔ
|
try:
|
# Ê×Ïȳ¢ÊÔ__dict__ÊôÐÔ
|
if hasattr(obj, '__dict__'):
|
for attr in dir(obj):
|
if attr.startswith('__'):
|
continue
|
|
try:
|
attr_value = getattr(obj, attr)
|
except Exception:
|
continue
|
|
# ¹ýÂË»ù´¡ÀàÐÍ
|
if not self._is_basic_type(attr_value):
|
refs.append(attr_value)
|
except Exception:
|
pass
|
|
# ´¦Àí³£¼ûÈÝÆ÷ÀàÐÍ
|
if obj_type in (list, tuple, set, collections.deque):
|
for item in obj:
|
if not self._is_basic_type(item):
|
refs.append(item)
|
elif obj_type is dict:
|
for key, value in obj.iteritems():
|
if not self._is_basic_type(key):
|
refs.append(key)
|
if not self._is_basic_type(value):
|
refs.append(value)
|
|
# ´¦ÀíÌØÊâÊôÐÔ£¨Èç__slots__¶¨ÒåµÄ¶ÔÏó£©
|
try:
|
if hasattr(obj, '__slots__'):
|
for slot_name in obj.__slots__:
|
if hasattr(obj, slot_name):
|
slot_value = getattr(obj, slot_name)
|
if not self._is_basic_type(slot_value):
|
refs.append(slot_value)
|
except Exception:
|
pass
|
|
return refs
|
|
def _is_basic_type(self, value):
|
"""ÅжÏÖµÊÇ·ñΪ»ù±¾ÀàÐÍ£¬²»ÐèÒªµÝ¹é´¦Àí"""
|
if value is None:
|
return True
|
if isinstance(value, (int, long, float, bool, str, unicode)):
|
return True
|
if isinstance(value, (types.FunctionType, types.MethodType, types.ModuleType)):
|
return True
|
if isinstance(value, (types.BuiltinFunctionType, type)):
|
return True
|
return False
|
|
def _return_to_pool(self, obj):
|
"""½«¶ÔÏó¹é»¹µ½ÆäËùÊô³Ø"""
|
obj_id = id(obj)
|
|
# »ñÈ¡¶ÔÏóËùÊôÀà
|
if obj_id not in self._obj_id_to_pool:
|
return # Èç¹û²»ÔÚ³ØÖУ¬ºöÂÔ
|
|
obj_class = self._obj_id_to_pool[obj_id]
|
|
if obj_class not in self._pools:
|
return # Èç¹û³Ø²»´æÔÚ£¬ºöÂÔ
|
|
pool = self._pools[obj_class]
|
|
# È·±£¶ÔÏóÔÚ»îÔ¾ÁбíÖÐ
|
if obj in pool['active_list']:
|
# ´Ó»îÔ¾ÁбíÒƳý
|
try:
|
pool['active_list'].remove(obj)
|
except ValueError:
|
# Èç¹û¶ÔÏó²»ÔÚÁбíÖУ¬ºöÂÔ
|
return
|
|
# Ö´ÐжÔÏóÇåÀí·½·¨
|
#if hasattr(obj, 'cleanup'):
|
# try:
|
# obj.cleanup()
|
# except Exception:
|
# pass
|
|
# Ìí¼Óµ½¿ÕÏÐÁбí
|
pool['free_list'].append(obj)
|
else:
|
# Èç¹ûÔÚ¿ÕÏÐÁбíÖУ¬Ìø¹ý
|
pass
|
|
def destroy_pool(self, obj_class):
|
"""Ïú»ÙÖ¸¶¨ÀàµÄ¶ÔÏó³Ø"""
|
if obj_class not in self._pools:
|
return
|
|
pool = self._pools[obj_class]
|
|
# ÇåÀíËùÓжÔÏó
|
for obj in list(pool['free_list'] + pool['active_list']):
|
# ÇåÀí¶ÔÏóÊôÐÔ
|
#if hasattr(obj, 'cleanup'):
|
# try:
|
# obj.cleanup()
|
# except Exception:
|
# pass
|
|
# ´ÓÓ³ÉäÖÐÒƳý
|
obj_id = id(obj)
|
if obj_id in self._obj_id_to_pool:
|
del self._obj_id_to_pool[obj_id]
|
|
del self._pools[obj_class]
|
|
def pool_status(self):
|
"""Êä³öËùÓжÔÏó³ØµÄ״̬ÐÅÏ¢"""
|
statusList = []
|
max_name_len = 0
|
max_created_len = 0
|
max_free_len = 0
|
max_active_len = 0
|
|
# ÊÕ¼¯×´Ì¬²¢¼ÆËã×î´ó³¤¶È
|
for obj_class, pool in self._pools.items():
|
# »ñÈ¡ÀàÃû
|
class_name = getattr(obj_class, '__name__', str(obj_class))
|
created = pool['created_count']
|
free = len(pool['free_list'])
|
active = len(pool['active_list'])
|
|
statusList.append({
|
'obj_class': class_name,
|
'created': created,
|
'free': free,
|
'active': active
|
})
|
|
# ¼ÆËã×î´ó³¤¶È
|
max_name_len = max(max_name_len, len(class_name))
|
max_created_len = max(max_created_len, len(str(created)))
|
max_free_len = max(max_free_len, len(str(free)))
|
max_active_len = max(max_active_len, len(str(active)))
|
|
# Ìí¼Ó±íÍ·³¤¶È
|
max_name_len = max(max_name_len, 8) # "¶ÔÏó³ØÃû³Æ"µÄ³¤¶È
|
max_created_len = max(max_created_len, 4) # "×ÜÊý"µÄ³¤¶È
|
max_free_len = max(max_free_len, 4) # "¿ÕÏÐ"µÄ³¤¶È
|
max_active_len = max(max_active_len, 4) # "»îÔ¾"µÄ³¤¶È
|
|
# °´»îÔ¾¶ÔÏóÊýÁ¿ÅÅÐò
|
statusList.sort(key=lambda s: (s['created'], s['active']), reverse=True)
|
|
# ´´½¨¸ñʽ×Ö·û´®
|
header_format = "| {:<{name_width}} | {:>{created_width}} | {:>{free_width}} | {:>{active_width}} |"
|
row_format = "| {:<{name_width}} | {:>{created_width}} | {:>{free_width}} | {:>{active_width}} |"
|
|
# Êä³ö±íÍ·
|
GameWorld.Log("----- ¶ÔÏó³Ø״̬ -----")
|
GameWorld.Log(header_format.format(
|
"¶ÔÏó³ØÃû³Æ", "×ÜÊý", "¿ÕÏÐ", "»îÔ¾",
|
name_width=max_name_len,
|
created_width=max_created_len,
|
free_width=max_free_len,
|
active_width=max_active_len
|
))
|
|
# Êä³ö·Ö¸ôÏß
|
separator = "+-{}-+-{}-+-{}-+-{}-+".format(
|
"-" * max_name_len,
|
"-" * max_created_len,
|
"-" * max_free_len,
|
"-" * max_active_len
|
)
|
GameWorld.Log(separator)
|
|
# Êä³öÊý¾ÝÐÐ
|
for s in statusList:
|
GameWorld.Log(row_format.format(
|
s['obj_class'], s['created'], s['free'], s['active'],
|
name_width=max_name_len,
|
created_width=max_created_len,
|
free_width=max_free_len,
|
active_width=max_active_len
|
))
|
|
return
|
|
def GetPoolMgr():
|
"""»ñÈ¡Ö¸¶¨Àà´´½¨¶ÔÏó³Ø"""
|
poolMgr = PyGameData.g_objPoolMgr
|
if not poolMgr:
|
poolMgr = RecursiveObjectPoolManager()
|
PyGameData.g_objPoolMgr = poolMgr
|
return poolMgr
|
|
def OnMinute():
|
"""ÿ·ÖÖÓÖ´ÐУ¬Êä³ö¶ÔÏó³Ø״̬"""
|
GetPoolMgr().pool_status()
|
return
|
|
## ʹÓÃʾÀý
|
#if __name__ == "__main__":
|
#
|
# # ¶¨ÒåÒ»¸öǶÌ×µÄStructureÀà
|
# class SubStructure(ctypes.Structure):
|
# _fields_ = [("z", ctypes.c_int)]
|
#
|
# def __init__(self, name):
|
# self.name = name
|
# print "³õʼ»¯SubStructure: name=%s" % (self.name)
|
# return
|
#
|
# def cleanup(self):
|
# """ÇåÀí·½·¨"""
|
# print "ÇåÀíSubStructure: name=%s" % (self.name)
|
# self.name = None
|
#
|
# # ¶¨ÒåÒ»¸ö¼Ì³Ð×ÔStructureµÄÀà
|
# class MyStructure(ctypes.Structure):
|
# _fields_ = [("x", ctypes.c_int), ("y", ctypes.c_int)]
|
#
|
# def __init__(self, name):
|
# self.name = name
|
# # ǶÌ×µÄStructure¶ÔÏó
|
# self.sub = GetPoolMgr().acquire(SubStructure, "%s_sub" % name)
|
# print "³õʼ»¯MyStructure: name=%s" % (self.name)
|
# return
|
#
|
# def cleanup(self):
|
# """ÇåÀí·½·¨"""
|
# print "ÇåÀíMyStructure: name=%s" % (self.name)
|
# self.name = None
|
# # ²»ÐèÒªÊÖ¶¯ÊÍ·Åsub£¬µÝ¹éÊͷŻᴦÀí
|
# self.sub = None
|
#
|
# class ClassE():
|
#
|
# def __init__(self, name):
|
# self.name = name
|
# print "³õʼ»¯¶ÔÏó ClassE: name=%s" % (self.name)
|
# self.ed1 = GetPoolMgr().acquire(ClassD, "%s_ed_1" % name)
|
# return
|
#
|
# class ClassD():
|
#
|
# def __init__(self, name):
|
# self.name = name
|
# print "³õʼ»¯¶ÔÏó ClassD: name=%s" % (self.name)
|
# return
|
#
|
# def cleanup(self):
|
# """ÇåÀí·½·¨"""
|
# print "ÇåÀíClassD: name=%s" % (self.name)
|
# self.name = None
|
#
|
# class ClassC():
|
#
|
# def __init__(self, name, bb):
|
# self.name = name
|
# print "³õʼ»¯¶ÔÏó ClassC: name=%s" % (self.name)
|
# self.bb = bb
|
# self.cd1 = GetPoolMgr().acquire(ClassD, "%s_cd_1" % name)
|
# self.cd2 = GetPoolMgr().acquire(ClassD, "%s_cd_2" % name)
|
# return
|
#
|
# class ClassB():
|
#
|
# def __init__(self, name):
|
# self.name = name
|
# print "³õʼ»¯¶ÔÏó ClassB: name=%s" % (self.name)
|
# bc1 = GetPoolMgr().acquire(ClassC, "%s_c_1" % name, self)
|
# bc2 = GetPoolMgr().acquire(ClassC, "%s_c_2" % name, self)
|
# self.cList = [bc1, bc2]
|
# self.dDict = {1:GetPoolMgr().acquire(ClassD, "%s_d_1" % name), GetPoolMgr().acquire(ClassD, "%s_d_2" % name):2, "bc1":bc1, "bc2":bc2}
|
# return
|
#
|
# def do(self, doStr):
|
# print "ClassB.do name=%s, %s" % (self.name, doStr)
|
# return
|
#
|
# class ClassA():
|
#
|
# def __init__(self, objID, name):
|
# self.objID = objID
|
# self.name = name
|
# self.b = GetPoolMgr().acquire(ClassB, "%s_b" % name)
|
# self.e = ClassE("%s_e" % name)
|
# self.struct = GetPoolMgr().acquire(MyStructure, "%s_struct" % name) # Structure¶ÔÏó
|
# self.dDict = {1:GetPoolMgr().acquire(ClassD, "%s_d_1" % name), GetPoolMgr().acquire(ClassD, "%s_d_2" % name):2}
|
# print "³õʼ»¯¶ÔÏó ClassA: objID=%s,name=%s" % (self.objID, self.name)
|
#
|
# def do(self, doStr):
|
# print "ClassA.do objID=%s,name=%s, %s" % (self.objID, self.name, doStr)
|
# return
|
#
|
# def cleanup(self):
|
# """ÇåÀí·½·¨"""
|
# print "ÇåÀíClassA: objID=%s,name=%s" % (self.objID, self.name)
|
# self.b = None
|
# self.e = None
|
# self.struct = None
|
# self.dDict = None
|
# self.name = None
|
# self.objID = None
|
#
|
#
|
# # ´´½¨¶ÔÏó³Ø¹ÜÀíÆ÷
|
# poolMgr = GetPoolMgr()
|
#
|
# # ΪËùÓÐÀà´´½¨¶ÔÏó³Ø£¬max_size=0 ±íʾÎÞÏÞÖÆ
|
# poolMgr.create_pool(ClassA, max_size=0)
|
# poolMgr.create_pool(ClassB, max_size=0)
|
# poolMgr.create_pool(ClassC, max_size=0)
|
# poolMgr.create_pool(ClassD, max_size=0)
|
# poolMgr.create_pool(MyStructure, max_size=0)
|
# poolMgr.create_pool(SubStructure, max_size=0) # ǶÌ×µÄStructureÀà
|
#
|
# # ´´½¨²âÊÔ¶ÔÏó
|
# a1 = poolMgr.acquire(ClassA, 1000, "hanmeimei")
|
# if a1 is None:
|
# GameWorld.ErrLog("Failed to acquire ClassA instance for hanmeimei")
|
# else:
|
# a1.do("hello 1ilei")
|
#
|
# a2 = poolMgr.acquire(ClassA, 1001, "lilei")
|
# if a2 is None:
|
# GameWorld.ErrLog("Failed to acquire ClassA instance for lilei")
|
# else:
|
# a2.do("hello hanmeimei")
|
#
|
# # Êä³ö״̬
|
# print("\nÊÍ·Åǰ״̬")
|
# poolMgr.pool_status()
|
#
|
# # ÊͷŶÔÏó
|
# if a1:
|
# poolMgr.release(a1)
|
# if a2:
|
# poolMgr.release(a2)
|
#
|
# # Êä³ö״̬
|
# print("\nÊͷźó״̬")
|
# poolMgr.pool_status()
|
#
|
# # ÔٴλñÈ¡Ó¦¸´ÓöÔÏó
|
# a3 = poolMgr.acquire(ClassA, 1002, "lihua")
|
# if a3 is None:
|
# GameWorld.ErrLog("Failed to acquire ClassA instance for lihua")
|
# else:
|
# a3.do("fuck you")
|
#
|
# # Êä³ö״̬
|
# print("\nÔٴλñÈ¡ºó״̬")
|
# poolMgr.pool_status()
|
#
|
# # ÇåÀí
|
# if a3:
|
# poolMgr.release(a3)
|
# poolMgr.destroy_pool(ClassA)
|
# poolMgr.destroy_pool(ClassB)
|
# poolMgr.destroy_pool(ClassC)
|
# poolMgr.destroy_pool(ClassD)
|
# poolMgr.destroy_pool(MyStructure)
|
# poolMgr.destroy_pool(SubStructure)
|
