-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
remind.go
83 lines (74 loc) · 2.54 KB
/
remind.go
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
// Auto-generated to Go types and interfaces using avdl-compiler v1.4.10 (https://github.com/keybase/node-avdl-compiler)
// Input file: avdl/gregor1/remind.avdl
package gregor1
import (
"github.com/keybase/go-framed-msgpack-rpc/rpc"
context "golang.org/x/net/context"
"time"
)
type GetRemindersArg struct {
MaxReminders int `codec:"maxReminders" json:"maxReminders"`
}
type DeleteRemindersArg struct {
ReminderIDs []ReminderID `codec:"reminderIDs" json:"reminderIDs"`
}
type RemindInterface interface {
// getReminders gets the reminders waiting to be sent out as a batch. Get at most
// maxReminders back.
GetReminders(context.Context, int) (ReminderSet, error)
// deleteReminders deletes all of the reminders by ReminderID
DeleteReminders(context.Context, []ReminderID) error
}
func RemindProtocol(i RemindInterface) rpc.Protocol {
return rpc.Protocol{
Name: "gregor.1.remind",
Methods: map[string]rpc.ServeHandlerDescription{
"getReminders": {
MakeArg: func() interface{} {
var ret [1]GetRemindersArg
return &ret
},
Handler: func(ctx context.Context, args interface{}) (ret interface{}, err error) {
typedArgs, ok := args.(*[1]GetRemindersArg)
if !ok {
err = rpc.NewTypeError((*[1]GetRemindersArg)(nil), args)
return
}
ret, err = i.GetReminders(ctx, typedArgs[0].MaxReminders)
return
},
},
"deleteReminders": {
MakeArg: func() interface{} {
var ret [1]DeleteRemindersArg
return &ret
},
Handler: func(ctx context.Context, args interface{}) (ret interface{}, err error) {
typedArgs, ok := args.(*[1]DeleteRemindersArg)
if !ok {
err = rpc.NewTypeError((*[1]DeleteRemindersArg)(nil), args)
return
}
err = i.DeleteReminders(ctx, typedArgs[0].ReminderIDs)
return
},
},
},
}
}
type RemindClient struct {
Cli rpc.GenericClient
}
// getReminders gets the reminders waiting to be sent out as a batch. Get at most
// maxReminders back.
func (c RemindClient) GetReminders(ctx context.Context, maxReminders int) (res ReminderSet, err error) {
__arg := GetRemindersArg{MaxReminders: maxReminders}
err = c.Cli.Call(ctx, "gregor.1.remind.getReminders", []interface{}{__arg}, &res, 0*time.Millisecond)
return
}
// deleteReminders deletes all of the reminders by ReminderID
func (c RemindClient) DeleteReminders(ctx context.Context, reminderIDs []ReminderID) (err error) {
__arg := DeleteRemindersArg{ReminderIDs: reminderIDs}
err = c.Cli.Call(ctx, "gregor.1.remind.deleteReminders", []interface{}{__arg}, nil, 0*time.Millisecond)
return
}