Skip to content

Commit ed97fcd

Browse files
authored
Publish context (#47)
1 parent 4005976 commit ed97fcd

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

‎pkg/sql/publisher.go‎

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,13 @@ func (p *Publisher) Publish(topic string, messages ...*message.Message) (err err
117117
"query_args": sqlArgsToLog(insertQuery.Args),
118118
})
119119

120-
_, err = p.db.ExecContext(context.Background(), insertQuery.Query, insertQuery.Args...)
120+
var ctx context.Context
121+
if len(messages) > 0 {
122+
ctx = messages[0].Context()
123+
} else {
124+
ctx = context.Background()
125+
}
126+
_, err = p.db.ExecContext(ctx, insertQuery.Query, insertQuery.Args...)
121127
if err != nil {
122128
return fmt.Errorf("could not insert message as row: %w", err)
123129
}

0 commit comments

Comments
 (0)