1 # -*- coding: utf-8 -*-
   2 #
   3 #
   4 # $Id: ussd.py,v 1.8 2004/09/14 12:45:00 hoxide Exp $
   5 """$,1$U现了(BUSS德腄aemon程序
   6 """
   7 
   8 from uss.message import usspmsg
   9 from uss.protocols import ussp
  10 from twisted.internet import reactor, protocol
  11 from time import clock
  12 
  13 import Queue
  14 import thread
  15 
  16 class USSServerProtocol(ussp.USSProtocol):
  17     """建立$,1){嘎?(BUSS服务器的协议处理"""
  18     """
  19     def __init__(self):
  20         ussp.USSProtocol.__init__(self)
  21     """
  22     funs={
  23             'connect': lambda self, m: self.p_ussp_connect(m),
  24             'mail_counter': lambda self, m: self.p_ussp_mail_counter(m),
  25             'terminate': lambda self, m: self.p_ussp_terminate(m)
  26             }
  27     def connectionMade(self):
  28         """建立连接?$,1%#?R舳疾蛔?(B"""
  29         print 'connectionMade'
  30         self.funs=USSServerProtocol.funs
  31     def on_ussp_connect(self, message):
  32         getqueue.put((self, 'connect', message))
  33     def on_ussp_mail_counter(self, message):
  34         getqueue.put((self, 'mail_counter', message))
  35     def on_ussp_terminate(self, message):
  36         getqueue.put((self, 'terminate', message))
  37         
  38     def p_ussp_connect(self, message):
  39         """处理?$,1&[Щ?(Bconnect报文"""
  40         smessage = usspmsg.USSPMessage()
  41         smessage.head.sequence = message.head.sequence
  42         smessage.setMsgName('connect_resp')
  43         smessage.body.setField('status',1)
  44         smessage.body.setField('version',1)
  45         self.sendData(smessage.packed())
  46 
  47     def p_ussp_mail_counter(self, message):
  48         """处理?$,1&[Щ⒗吹?(Bmail_counter报文"""
  49         smessage = usspmsg.USSPMessage()
  50         smessage.head.sequence = message.head.sequence
  51         smessage.setMsgName('mail_counter_resp')
  52         smessage.body.setField('uid',message.body.fields['uid'])
  53         smessage.body.setField('number',2028)
  54         self.sendData(smessage.packed())
  55 
  56     def p_ussp_terminate(self, message):
  57         """处理?$,1&[Щ?(Bterminate报文"""
  58         smessage = usspmsg.USSPMessage()
  59         smessage.head.sequence = message.head.sequence
  60         smessage.setMsgName('terminate_resp')
  61         self.sendData(smessage.packed())
  62         # 完事了！！！?$,1'?>的连接
  63 (B        self.transport.loseConnection()
  64     
  65     
  66 
  67 class USSServerFactory(protocol.ServerFactory):
  68     """USS服务器工厂"""
  69     protocol = USSServerProtocol
  70 
  71 import time
  72 def processData(queue):
  73     while 1:
  74         """
  75         if queue.empty():
  76             print 'sleep 1 sec'
  77             time.sleep(1)
  78             continue
  79         """
  80         #print 'to get'
  81         m = queue.get(block=True)
  82         #print 'get'
  83         m[0].funs[m[1]](m[0], m[2])
  84 
  85 getqueue = Queue.Queue(100)
  86 
  87 #thread.start_new(processData, (getqueue,))
  88 reactor.callInThread(processData, getqueue)
  89 
  90 def main():
  91     f = USSServerFactory()
  92     reactor.listenTCP(7890, f)
  93     reactor.run()
  94 
  95 if __name__ == "__main__":
  96     main()
  97