写了个用 socket 传送对象的 py 类

Python/PHP/Perl 开发与设计
回复
头像
BigSnake.NET
帖子: 12522
注册时间: 2006-07-02 11:16
来自: 廣州
送出感谢: 0
接收感谢: 7 次
联系:

写了个用 socket 传送对象的 py 类

#1

帖子 BigSnake.NET » 2008-07-05 11:13

代码: 全选

# -*- encoding: UTF-8 -*-
'''ObjectSocket 提供线程安全的利用 socket 传送 python 对象的方法.'''

import socket
import pickle
import threading

class ObjectSocketSendException(Exception):
    '''此异常在无法发送所有字节时抛出.'''
    pass


class InvalidObjectSocket(Exception):
    '''此异常在对无效的 ObjectSocket 对象操作时抛出.'''
    pass


class ObjectSocket(object):
    '''ObjectSocket 使用 socket 来传送对象.'''

    def __init__(self, socket):
        '''构造函数, socket 参数必须是有效的"新的" socket 对象.
        socket 用于构造 ObjectSocket 后不应该再对其做任何操作.
        '''
        self._socket = socket
        self._socket_file = self._socket.makefile('rb')
        # 锁都是私有的, 应该不会死锁.
        self._sendlock = threading.Lock()
        self._recvlock = threading.Lock()
        self._valid = True

    def IsValid(self):
        '''验证 ObjectSocket 是否有效(未关闭).'''
        return self._valid

    def send(self, obj):
        '''发送一个能被 pickle 处理的对象. 如果失败抛出异常.'''
        self._sendlock.acquire()
        try:
            if not self._valid:
                raise InvalidObjectSocket()
            s = pickle.dumps(obj)
            l = self._socket.send(s)
            if len(s) != l:
                raise ObjectSocketSendException()
        finally:
            self._sendlock.release()

    def recv(self):
        '''接收并返回对象.'''
        self._recvlock.acquire()
        try:
            if not self._valid:
                raise InvalidObjectSocket()
            obj = pickle.load(self._socket_file)
        finally:
            self._recvlock.release()
        return obj

    def close(self):
        '''关闭连接'''
        self._sendlock.acquire()
        self._recvlock.acquire()
        try:
            self._socket.close()
        finally:
            self._valid = False
            self._recvlock.release()
            self._sendlock.release()

    def __del__(self):
        self.close()




该类是线程安全的

帮忙看看有没有漏洞, 或者显然的bug..
^_^ ~~~
要理解递归,首先要理解递归。

地球人都知道,理论上,理论跟实际是没有差别的,但实际上,理论跟实际的差别是相当大滴。
yymailb
帖子: 240
注册时间: 2007-03-16 1:38
送出感谢: 0
接收感谢: 0

#2

帖子 yymailb » 2008-07-13 10:53

Pickle 可以改用 cPickle
我自己做这种事情都是用cPickle打包, 用twisted传送, 代码量很少的, 就是要先装个twisted库(ubuntu上装起来很方便).
yymailb
帖子: 240
注册时间: 2007-03-16 1:38
送出感谢: 0
接收感谢: 0

#3

帖子 yymailb » 2008-07-13 10:59

另外可以考虑 ObjectSocket 直接从 socket 类继承.
'''构造函数, socket 参数必须是有效的"新的" socket 对象.
socket 用于构造 ObjectSocket 后不应该再对其做任何操作.
'''
这个设计不是很好
yymailb
帖子: 240
注册时间: 2007-03-16 1:38
送出感谢: 0
接收感谢: 0

#4

帖子 yymailb » 2008-07-13 11:06

我对python的socket使用不熟, 这个程序里面从来没有调用过 socket.recv()函数, 不知道这样会不会有问题
joo.tsao
帖子: 37
注册时间: 2007-03-18 22:39
送出感谢: 0
接收感谢: 0

#5

帖子 joo.tsao » 2008-07-17 17:30

已有这样的类库http://pyro.sourceforge.net/,不用自己这么费心。
回复

回到 “Python/Php/Perl”