Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Better sequence start and end numbers.

Browse Source
This commit is contained in:
Steve Pulec 2014-11-26 21:55:01 -05:00
parent c63b00b8ba
commit 45de4a46ec
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
moto/kinesis/models.py
View File

@ -55,8 +55,15 @@ class Shard(object):
self.records[sequence_number] = Record(partition_key, data, sequence_number) self.records[sequence_number] = Record(partition_key, data, sequence_number)
return sequence_number return sequence_number
def get_min_sequence_number(self):
if self.records:
return list(self.records.keys())[0]
return 0
def get_max_sequence_number(self): def get_max_sequence_number(self):
if self.records:
return list(self.records.keys())[-1] return list(self.records.keys())[-1]
return 0
def to_json(self): def to_json(self):
return { return {
@ -65,8 +72,8 @@ class Shard(object):
"StartingHashKey": "0" "StartingHashKey": "0"
}, },
"SequenceNumberRange": { "SequenceNumberRange": {
"EndingSequenceNumber": "21269319989741826081360214168359141376", "EndingSequenceNumber": self.get_max_sequence_number(),
"StartingSequenceNumber": "21267647932558653966460912964485513216" "StartingSequenceNumber": self.get_min_sequence_number(),
}, },
"ShardId": self.shard_id "ShardId": self.shard_id
} }