-
Notifications
You must be signed in to change notification settings - Fork 167
Expand file tree
/
Copy pathmiddleware_defaults.go
More file actions
65 lines (48 loc) · 3.01 KB
/
Copy pathmiddleware_defaults.go
File metadata and controls
65 lines (48 loc) · 3.01 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package river
import (
"context"
"github.com/riverqueue/river/rivertype"
)
// MiddlewareDefaults should be embedded on any middleware implementation. It
// helps identify a struct as middleware and a plugin, and guarantees forward
// compatibility in case additions are necessary to the rivertype.Middleware
// interface.
type MiddlewareDefaults struct{}
func (d *MiddlewareDefaults) IsMiddleware() bool { return true }
func (d *MiddlewareDefaults) IsPlugin() bool { return true }
// JobInsertMiddlewareDefaults is an embeddable struct that provides default
// implementations for the rivertype.JobInsertMiddleware. Use of this struct is
// recommended in case rivertype.JobInsertMiddleware is expanded in the future
// so that existing code isn't unexpectedly broken during an upgrade.
//
// Deprecated: Prefer embedding the more general MiddlewareDefaults instead.
type JobInsertMiddlewareDefaults struct{ MiddlewareDefaults }
func (d *JobInsertMiddlewareDefaults) InsertMany(ctx context.Context, manyParams []*rivertype.JobInsertParams, doInner func(ctx context.Context) ([]*rivertype.JobInsertResult, error)) ([]*rivertype.JobInsertResult, error) {
return doInner(ctx)
}
// JobInsertMiddlewareFunc is a convenience helper for implementing
// rivertype.JobInsertMiddleware using a simple function instead of a struct.
type JobInsertMiddlewareFunc func(ctx context.Context, manyParams []*rivertype.JobInsertParams, doInner func(ctx context.Context) ([]*rivertype.JobInsertResult, error)) ([]*rivertype.JobInsertResult, error)
func (f JobInsertMiddlewareFunc) InsertMany(ctx context.Context, manyParams []*rivertype.JobInsertParams, doInner func(ctx context.Context) ([]*rivertype.JobInsertResult, error)) ([]*rivertype.JobInsertResult, error) {
return f(ctx, manyParams, doInner)
}
func (f JobInsertMiddlewareFunc) IsMiddleware() bool { return true }
func (f JobInsertMiddlewareFunc) IsPlugin() bool { return true }
// WorkerInsertMiddlewareDefaults is an embeddable struct that provides default
// implementations for the rivertype.WorkerMiddleware. Use of this struct is
// recommended in case rivertype.WorkerMiddleware is expanded in the future so
// that existing code isn't unexpectedly broken during an upgrade.
//
// Deprecated: Prefer embedding the more general MiddlewareDefaults instead.
type WorkerMiddlewareDefaults struct{ MiddlewareDefaults }
func (d *WorkerMiddlewareDefaults) Work(ctx context.Context, job *rivertype.JobRow, doInner func(ctx context.Context) error) error {
return doInner(ctx)
}
// WorkerMiddlewareFunc is a convenience helper for implementing
// rivertype.WorkerMiddleware using a simple function instead of a struct.
type WorkerMiddlewareFunc func(ctx context.Context, job *rivertype.JobRow, doInner func(ctx context.Context) error) error
func (f WorkerMiddlewareFunc) IsMiddleware() bool { return true }
func (f WorkerMiddlewareFunc) IsPlugin() bool { return true }
func (f WorkerMiddlewareFunc) Work(ctx context.Context, job *rivertype.JobRow, doInner func(ctx context.Context) error) error {
return f(ctx, job, doInner)
}