summaryrefslogtreecommitdiff
path: root/py/demo/IceStorm/clock/Subscriber.py
blob: 6ba56916ff6a6cfacd0a0f6fe24451728d836f7e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
#!/usr/bin/env python
# **********************************************************************
#
# Copyright (c) 2003-2010 ZeroC, Inc. All rights reserved.
#
# This copy of Ice is licensed to you under the terms described in the
# ICE_LICENSE file included in this distribution.
#
# **********************************************************************

import sys, traceback, Ice, IceStorm, getopt

Ice.loadSlice('Clock.ice')
import Demo

class ClockI(Demo.Clock):
    def tick(self, date, current=None):
        print date

class Subscriber(Ice.Application):
    def usage(self):
        print "Usage: " + self.appName() + \
	    " [--batch] [--datagram|--twoway|--ordered|--oneway] [--retryCount count] [--id id] [topic]"

    def run(self, args):
        try:
            opts, args = getopt.getopt(args[1:], '', ['datagram', 'twoway', 'oneway', 'ordered', 'batch',
	    	'retryCount=', 'id='])
        except getopt.GetoptError:
            self.usage()
            return 1

	batch = False
	option = "None"
        topicName = "time"
        id = ""
        retryCount = ""

        for o, a in opts:
	    oldoption = option
            if o == "--datagram":
                option = "Datagram"
            elif o =="--twoway":
                option = "Twoway"
            elif o =="--ordered":
                option = "Ordered"
            elif o =="--oneway":
	    	option = "Oneway"
            elif o =="--batch":
                batch = True
	    elif o == "--id":
		id = a
	    elif o == "--retryCount":
		retryCount = a
	    if oldoption != option and oldoption != "None":
		self.usage()
		return 1

        if len(args) > 1:
		self.usage()
		return 1

        if len(args) > 0:
            topicName = args[0]

	if len(retryCount) > 0:
	    if option == "None":
		option = "Twoway"
	    elif option != "Twoway" and option != "Ordered":
	    	print self.appName() + ": retryCount requires a twoway proxy"
		return 1

        if batch and (option in ("Twoway", "Ordered")):
            print self.appName() + ": batch can only be set with oneway or datagram"
            return 1

        manager = IceStorm.TopicManagerPrx.checkedCast(self.communicator().propertyToProxy('TopicManager.Proxy'))
        if not manager:
            print args[0] + ": invalid proxy"
            return 1

        #
        # Retrieve the topic.
        #
        try:
            topic = manager.retrieve(topicName)
        except IceStorm.NoSuchTopic, e:
            try:
                topic = manager.create(topicName)
            except IceStorm.TopicExists, ex:
                print self.appName() + ": temporary error. try again"
                return 1

        adapter = self.communicator().createObjectAdapter("Clock.Subscriber")

	#
	# Add a servant for the Ice object. If --id is used the identity
	# comes from the command line, otherwise a UUID is used.
	#
	# id is not directly altered since it is used below to detect
	# whether subscribeAndGetPublisher can raise AlreadySubscribed.
	#

	subId = Ice.Identity()
	subId.name = id
	if len(subId.name) == 0:
	    subId.name = Ice.generateUUID()
        subscriber = adapter.add(ClockI(), subId)

        qos = {}
	if len(retryCount) > 0:
	    qos["retryCount"] = retryCount

	#
	# Set up the proxy.
	#
        if option == "Datagram":
	    if batch:
		subscriber = subscriber.ice_batchDatagram()
	    else:
		subscriber = subscriber.ice_datagram()
        elif option == "Twoway":
            # Do nothing to the subscriber proxy. Its already twoway.
             pass
        elif option == "Ordered":
            # Do nothing to the subscriber proxy. Its already twoway.
            qos["reliability"] = "ordered"
        elif option == "Oneway" or option == "None":
	    if batch:
		subscriber = subscriber.ice_batchOneway()
	    else:
		subscriber = subscriber.ice_oneway()

	try:
	    topic.subscribeAndGetPublisher(qos, subscriber)
	except IceStorm.AlreadySubscribed, ex:
	    # If we're manually setting the subscriber id ignore.
	    if len(id) == 0:
		raise
	    print "reactivating persistent subscriber"

        adapter.activate()

        self.shutdownOnInterrupt()
        self.communicator().waitForShutdown()

        #
        # Unsubscribe all subscribed objects.
        #
        topic.unsubscribe(subscriber)
            
        return 0

app = Subscriber()
sys.exit(app.main(sys.argv, "config.sub"))