Asynchronous roomserver input from federation API /send

This commit is contained in:
Neil Alexander 2021-06-30 15:01:19 +01:00
parent 2647f6e9c5
commit ca42568640
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
4 changed files with 42 additions and 15 deletions

View file

@ -86,6 +86,9 @@ type TransactionID struct {
// InputRoomEventsRequest is a request to InputRoomEvents
type InputRoomEventsRequest struct {
InputRoomEvents []InputRoomEvent `json:"input_room_events"`
// Asynchronous requests will queue up work into the roomserver and
// return straight away, rather than waiting for the results.
Asynchronous bool `json:"async"`
}
// InputRoomEventsResponse is a response to InputRoomEvents

View file

@ -22,11 +22,18 @@ import (
"github.com/matrix-org/util"
)
type InputOption int
const (
InputOptionAsync InputOption = iota
)
// SendEvents to the roomserver The events are written with KindNew.
func SendEvents(
ctx context.Context, rsAPI RoomserverInternalAPI,
kind Kind, events []*gomatrixserverlib.HeaderedEvent,
sendAsServer gomatrixserverlib.ServerName, txnID *TransactionID,
options ...InputOption,
) error {
ires := make([]InputRoomEvent, len(events))
for i, event := range events {
@ -38,7 +45,7 @@ func SendEvents(
TransactionID: txnID,
}
}
return SendInputRoomEvents(ctx, rsAPI, ires)
return SendInputRoomEvents(ctx, rsAPI, ires, options...)
}
// SendEventWithState writes an event with the specified kind to the roomserver
@ -47,7 +54,7 @@ func SendEvents(
func SendEventWithState(
ctx context.Context, rsAPI RoomserverInternalAPI, kind Kind,
state *gomatrixserverlib.RespState, event *gomatrixserverlib.HeaderedEvent,
haveEventIDs map[string]bool,
haveEventIDs map[string]bool, options ...InputOption,
) error {
outliers, err := state.Events()
if err != nil {
@ -79,14 +86,25 @@ func SendEventWithState(
StateEventIDs: stateEventIDs,
})
return SendInputRoomEvents(ctx, rsAPI, ires)
return SendInputRoomEvents(ctx, rsAPI, ires, options...)
}
// SendInputRoomEvents to the roomserver.
func SendInputRoomEvents(
ctx context.Context, rsAPI RoomserverInternalAPI, ires []InputRoomEvent,
ctx context.Context, rsAPI RoomserverInternalAPI,
ires []InputRoomEvent, options ...InputOption,
) error {
request := InputRoomEventsRequest{InputRoomEvents: ires}
async := false
for _, opt := range options {
switch opt {
case InputOptionAsync:
async = true
}
}
request := InputRoomEventsRequest{
InputRoomEvents: ires,
Asynchronous: async,
}
var response InputRoomEventsResponse
rsAPI.InputRoomEvents(ctx, &request, &response)
return response.Err()