我們曾經(jīng)提到，Oyster.com的Python web服務(wù)器怎樣利用一個巨大的Python dicts（hash table），緩存大量的靜態(tài)資源。我們最近在Image類中，用僅僅一行__slots__代碼，讓每個6G內(nèi)存占用的服務(wù)進(jìn)程（共4個），省出超過2G來。

這是其中一個服務(wù)器在部署代碼前后的截圖：

我們alloc了大約一百萬個類似如下class的實例：
 
class Image(object):
    def __init__(self, id, caption, url):
        self.id = id
        self.caption = caption
        self.url = url
        self._setup()
 
    # ... other methods ...

默認(rèn)情況下，Python用一個dict來存儲對象實例的屬性。這在一般情況下還不錯，而且非常靈活，乃至你在運(yùn)行時可以隨意設(shè)置新的屬性。

但是，對一些在”編譯”前就知道該有幾個固定屬性的小class來說，這個dict就有點(diǎn)浪費(fèi)內(nèi)存了。而當(dāng)你把這個小浪費(fèi)乘上一百萬，那可就大不同了。在Python中，你可以在class中設(shè)置__slots__，它是一個包含這些固定的屬性名的list。這樣Python就不會再使用dict，而且只分配這些屬性的空間。
 
class Image(object):
    __slots__ = ['id', 'caption', 'url']
 
    def __init__(self, id, caption, url):
        self.id = id
        self.caption = caption
        self.url = url
        self._setup()
 
    # ... other methods ...

你還可以用collections.namedtuple，它允許訪問參數(shù)，但只占用一個tuple的空間。這跟__slots__類似。不過我總覺得繼承一個namedtuple類很奇怪。另外，如果你需要自定義初始化，你應(yīng)該重載__new__而不是__init__。

警告：不要貿(mào)然進(jìn)行這個優(yōu)化，把它用在所有地方。這種做法不利于代碼維護(hù)，而且只有當(dāng)你有數(shù)以千計的實例的時候才會有明顯效果。

譯注：作者在評論中關(guān)于”不利于代碼維護(hù)“的說法：

webreac：我覺得__slots__關(guān)鍵字不只是速度優(yōu)化（注：這里應(yīng)該是內(nèi)存優(yōu)化），也是類字段名的一個可靠”文檔“。這有利于代碼維護(hù)。為什么你覺得它不好？

Ben Hoyt(作者)：有趣的說法——我不確定應(yīng)不應(yīng)該把__slots__作為文檔。不過的確是不錯的注意。我之前這么說的原因是，你需要對字段名”定義“兩次（不夠DRY）。namedtuple也類似。

最新評論