Python WebAcademy

学習モード

メモリリークとその防止

Section 5 / 8

weakrefモジュールの活用

Pythonでは、オブジェクトが参照されている限りメモリに保持されます。しかし、長期間にわたって使用されるオブジェクトや大量のオブジェクトが生成される場合、メモリの消費が問題になることがあります。weakrefモジュールは、オブジェクトへの「弱い参照」を提供することで、これらの問題を緩和する手段を提供します。

weakrefモジュールとは？

weakrefモジュールは、オブジェクトが他のオブジェクトによって参照されている場合でも、そのオブジェクトがガーベジコレクションの対象になることを可能にします。これにより、メモリリークのリスクを減らし、リソースを効率的に使用できます。

使用目的

  • 循環参照を避ける
  • キャッシュやオブジェクトプールの実装
  • プログラムのパフォーマンス向上

基本的な使い方

まず、weakrefモジュールをインポートして、簡単な例を見てみましょう。

import weakref

class MyObject:
    def __init__(self, name):
        self.name = name

    def __repr__(self):
        return f"MyObject({self.name})"

# 通常の参照
obj = MyObject("A")
print("通常の参照:", obj)

# 弱い参照を作成
weak_obj = weakref.ref(obj)
print("弱い参照:", weak_obj())

# オブジェクトを明示的に削除
del obj
print("オブジェクト削除後の弱い参照:", weak_obj())

コード解説

  1. クラス定義: MyObjectというクラスを定義し、コンストラクタで名前を受け取ります。
  2. 通常の参照: objとしてMyObjectのインスタンスを作成し、通常の参照として保持します。
  3. 弱い参照: weakref.ref(obj)を使用して、objの弱い参照を作成します。この弱い参照は、weak_obj()を呼び出すことで元のオブジェクトにアクセスできます。
  4. オブジェクト削除: del objでobjを削除すると、弱い参照はNoneを返します。

コールバック関数の使用

weakrefモジュールでは、オブジェクトが削除されるときに自動的に呼び出されるコールバック関数を設定することもできます。以下の例では、オブジェクトが削除される際にメッセージを表示します。

import weakref

class MyObject:
    def __init__(self, name):
        self.name = name

    def __repr__(self):
        return f"MyObject({self.name})"

def on_object_delete(weak_ref):
    print(f"{weak_ref} が削除されました。")

# オブジェクトを作成
obj = MyObject("B")
weak_obj = weakref.ref(obj, on_object_delete)

print("弱い参照:", weak_obj())
del obj  # これによりコールバックが呼び出される

コード解説

  1. コールバック関数: on_object_delete関数を定義し、弱い参照が指しているオブジェクトが削除されるときに呼び出されるメッセージを表示します。
  2. 弱い参照の作成: weakref.ref(obj, on_object_delete)を使って、objの弱い参照を作成し、オブジェクトが削除されたときにコールバックを呼び出します。
  3. オブジェクト削除: del objを実行すると、コールバックが呼び出され、メッセージが表示されます。

まとめ

weakrefモジュールを使用すると、オブジェクトの管理がより柔軟になり、メモリの効率的な使用が促進されます。特に、キャッシュやオブジェクトプールの実装時に、不要なメモリリークを防ぐ手段として非常に有用です。これを活用することで、Pythonプログラムのパフォーマンスを向上させることができます。

前に戻る