Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Merge pull request #658 from matrix-org/markjh/cleanup
Browse files Browse the repository at this point in the history
Remove some unused parameters from persist_event
  • Loading branch information
NegativeMjark committed Mar 22, 2016
2 parents e4054ab + 5244c0b commit 63137bb
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 44 deletions.
1 change: 0 additions & 1 deletion synapse/federation/federation_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -531,7 +531,6 @@ def _handle_new_pdu(self, origin, pdu, get_missing=True):
yield self.handler.on_receive_pdu(
origin,
pdu,
backfilled=False,
state=state,
auth_chain=auth_chain,
)
Expand Down
40 changes: 15 additions & 25 deletions synapse/handlers/federation.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,7 @@ def handle_new_event(self, event, destinations):

@log_function
@defer.inlineCallbacks
def on_receive_pdu(self, origin, pdu, backfilled, state=None,
def on_receive_pdu(self, origin, pdu, state=None,
auth_chain=None):
""" Called by the ReplicationLayer when we have a new pdu. We need to
do auth checks and put it through the StateHandler.
Expand All @@ -123,7 +123,6 @@ def on_receive_pdu(self, origin, pdu, backfilled, state=None,

# FIXME (erikj): Awful hack to make the case where we are not currently
# in the room work
current_state = None
is_in_room = yield self.auth.check_host_in_room(
event.room_id,
self.server_name
Expand Down Expand Up @@ -186,8 +185,6 @@ def on_receive_pdu(self, origin, pdu, backfilled, state=None,
origin,
event,
state=state,
backfilled=backfilled,
current_state=current_state,
)
except AuthError as e:
raise FederationError(
Expand Down Expand Up @@ -216,18 +213,17 @@ def on_receive_pdu(self, origin, pdu, backfilled, state=None,
except StoreError:
logger.exception("Failed to store room.")

if not backfilled:
extra_users = []
if event.type == EventTypes.Member:
target_user_id = event.state_key
target_user = UserID.from_string(target_user_id)
extra_users.append(target_user)
extra_users = []
if event.type == EventTypes.Member:
target_user_id = event.state_key
target_user = UserID.from_string(target_user_id)
extra_users.append(target_user)

with PreserveLoggingContext():
self.notifier.on_new_room_event(
event, event_stream_id, max_stream_id,
extra_users=extra_users
)
with PreserveLoggingContext():
self.notifier.on_new_room_event(
event, event_stream_id, max_stream_id,
extra_users=extra_users
)

if event.type == EventTypes.Member:
if event.membership == Membership.JOIN:
Expand Down Expand Up @@ -647,7 +643,7 @@ def do_invite_join(self, target_hosts, room_id, joinee, content):
continue

try:
self.on_receive_pdu(origin, p, backfilled=False)
self.on_receive_pdu(origin, p)
except:
logger.exception("Couldn't handle pdu")

Expand Down Expand Up @@ -779,7 +775,6 @@ def on_invite_request(self, origin, pdu):
event_stream_id, max_stream_id = yield self.store.persist_event(
event,
context=context,
backfilled=False,
)

target_user = UserID.from_string(event.state_key)
Expand Down Expand Up @@ -819,7 +814,6 @@ def do_remotely_reject_invite(self, target_hosts, room_id, user_id):
event_stream_id, max_stream_id = yield self.store.persist_event(
event,
context=context,
backfilled=False,
)

target_user = UserID.from_string(event.state_key)
Expand Down Expand Up @@ -1074,8 +1068,7 @@ def user_joined_room(self, user, room_id):

@defer.inlineCallbacks
@log_function
def _handle_new_event(self, origin, event, state=None, backfilled=False,
current_state=None, auth_events=None):
def _handle_new_event(self, origin, event, state=None, auth_events=None):

outlier = event.internal_metadata.is_outlier()

Expand All @@ -1085,7 +1078,7 @@ def _handle_new_event(self, origin, event, state=None, backfilled=False,
auth_events=auth_events,
)

if not backfilled and not event.internal_metadata.is_outlier():
if not event.internal_metadata.is_outlier():
action_generator = ActionGenerator(self.hs)
yield action_generator.handle_push_actions_for_event(
event, context, self
Expand All @@ -1094,9 +1087,7 @@ def _handle_new_event(self, origin, event, state=None, backfilled=False,
event_stream_id, max_stream_id = yield self.store.persist_event(
event,
context=context,
backfilled=backfilled,
is_new_state=(not outlier and not backfilled),
current_state=current_state,
is_new_state=not outlier,
)

defer.returnValue((context, event_stream_id, max_stream_id))
Expand Down Expand Up @@ -1194,7 +1185,6 @@ def _persist_auth_tree(self, auth_events, state, event):

event_stream_id, max_stream_id = yield self.store.persist_event(
event, new_event_context,
backfilled=False,
is_new_state=True,
current_state=state,
)
Expand Down
22 changes: 4 additions & 18 deletions synapse/storage/events.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,30 +101,16 @@ def stream_ordering_manager():

@defer.inlineCallbacks
@log_function
def persist_event(self, event, context, backfilled=False,
def persist_event(self, event, context,
is_new_state=True, current_state=None):
stream_ordering = None
if backfilled:
self.min_stream_token -= 1
stream_ordering = self.min_stream_token

if stream_ordering is None:
stream_ordering_manager = self._stream_id_gen.get_next()
else:
@contextmanager
def stream_ordering_manager():
yield stream_ordering
stream_ordering_manager = stream_ordering_manager()

try:
with stream_ordering_manager as stream_ordering:
with self._stream_id_gen.get_next() as stream_ordering:
event.internal_metadata.stream_ordering = stream_ordering
yield self.runInteraction(
"persist_event",
self._persist_event_txn,
event=event,
context=context,
backfilled=backfilled,
is_new_state=is_new_state,
current_state=current_state,
)
Expand Down Expand Up @@ -166,7 +152,7 @@ def get_event(self, event_id, check_redacted=True,
defer.returnValue(events[0] if events else None)

@log_function
def _persist_event_txn(self, txn, event, context, backfilled,
def _persist_event_txn(self, txn, event, context,
is_new_state=True, current_state=None):
# We purposefully do this first since if we include a `current_state`
# key, we *want* to update the `current_state_events` table
Expand Down Expand Up @@ -198,7 +184,7 @@ def _persist_event_txn(self, txn, event, context, backfilled,
return self._persist_events_txn(
txn,
[(event, context)],
backfilled=backfilled,
backfilled=False,
is_new_state=is_new_state,
)

Expand Down

0 comments on commit 63137bb

Please sign in to comment.