Forwards stream count parameter to driver

For returning the number of streams being managed.

Change-Id: Ibcd749314db4546554e830880c4c8295ef5c25bf
This commit is contained in:
Sandy Walsh 2015-01-27 10:47:56 -08:00
parent 1fb9325f85
commit b26cc7f04c
1 changed files with 7 additions and 6 deletions

View File

@ -54,8 +54,9 @@ class Impl(object):
self.winchester_config = scratchpad['quincy_config']
self.driver = scratchpad['quincy_driver']
def get_streams(self, state=None, older_than=None, younger_than=None,
trigger_name=None, distinguishing_traits=None):
def get_streams(self, count=False, state=None, older_than=None,
younger_than=None, trigger_name=None,
distinguishing_traits=None):
if state is not None:
try:
@ -63,10 +64,10 @@ class Impl(object):
except KeyError:
logger.error("invalid stream state %s" % state)
raise
return self.driver.find_streams(state=state, name=trigger_name,
younger_than=younger_than,
older_than=older_than,
distinguishing_traits=distinguishing_traits)
return self.driver.find_streams(count=count,
state=state, name=trigger_name,
younger_than=younger_than, older_than=older_than,
distinguishing_traits=distinguishing_traits)
def get_stream(self, stream_id, details):
stream = int(stream_id)