当前时区为 UTC + 8 小时



发表新帖 回复这个主题  [ 5 篇帖子 ] 
作者 内容
1 楼 
 文章标题 : 写了个用 socket 传送对象的 py 类
帖子发表于 : 2008-07-05 11:13 
头像

注册: 2006-07-02 11:16
帖子: 12522
地址: 廣州
送出感谢: 0 次
接收感谢: 8
代码:
# -*- 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..


_________________
^_^ ~~~
要理解递归,首先要理解递归。

地球人都知道,理论上,理论跟实际是没有差别的,但实际上,理论跟实际的差别是相当大滴。


页首
 用户资料  
 
2 楼 
 文章标题 :
帖子发表于 : 2008-07-13 10:53 

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


页首
 用户资料  
 
3 楼 
 文章标题 :
帖子发表于 : 2008-07-13 10:59 

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


页首
 用户资料  
 
4 楼 
 文章标题 :
帖子发表于 : 2008-07-13 11:06 

注册: 2007-03-16 1:38
帖子: 240
送出感谢: 0 次
接收感谢: 0 次
我对python的socket使用不熟, 这个程序里面从来没有调用过 socket.recv()函数, 不知道这样会不会有问题


页首
 用户资料  
 
5 楼 
 文章标题 :
帖子发表于 : 2008-07-17 17:30 

注册: 2007-03-18 22:39
帖子: 37
送出感谢: 0 次
接收感谢: 0 次
已有这样的类库http://pyro.sourceforge.net/,不用自己这么费心。


页首
 用户资料  
 
显示帖子 :  排序  
发表新帖 回复这个主题  [ 5 篇帖子 ] 

当前时区为 UTC + 8 小时


在线用户

正在浏览此版面的用户:没有注册用户 和 2 位游客


不能 在这个版面发表主题
不能 在这个版面回复主题
不能 在这个版面编辑帖子
不能 在这个版面删除帖子
不能 在这个版面提交附件

前往 :  
本站点为公益性站点,用于推广开源自由软件,由 DiaHosting VPSBudgetVM VPS 提供服务。
我们认为:软件应可免费取得,软件工具在各种语言环境下皆可使用,且不会有任何功能上的差异;
人们应有定制和修改软件的自由,且方式不受限制,只要他们自认为合适。

Powered by phpBB © 2000, 2002, 2005, 2007 phpBB Group
简体中文语系由 王笑宇 翻译