-
Notifications
You must be signed in to change notification settings - Fork 1
/
get_ticket.go
56 lines (44 loc) · 1.37 KB
/
get_ticket.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
package c1api
import (
"context"
"fmt"
v2 "github.com/conductorone/baton-sdk/pb/c1/connector/v2"
v1 "github.com/conductorone/baton-sdk/pb/c1/connectorapi/baton/v1"
"github.com/conductorone/baton-sdk/pkg/annotations"
"github.com/conductorone/baton-sdk/pkg/types"
"github.com/grpc-ecosystem/go-grpc-middleware/logging/zap/ctxzap"
"go.uber.org/zap"
"google.golang.org/protobuf/proto"
)
type getTicketTaskHelpers interface {
ConnectorClient() types.ConnectorClient
FinishTask(ctx context.Context, resp proto.Message, annos annotations.Annotations, err error) error
}
type getTicketTaskHandler struct {
task *v1.Task
helpers getTicketTaskHelpers
}
func (c *getTicketTaskHandler) HandleTask(ctx context.Context) error {
l := ctxzap.Extract(ctx)
cc := c.helpers.ConnectorClient()
ticket, err := cc.GetTicket(ctx, &v2.TicketsServiceGetTicketRequest{
Id: c.task.GetId(),
})
if err != nil {
return err
}
if ticket.GetTicket() == nil {
return fmt.Errorf("connector returned empt ticket schema")
}
resp := &v2.TicketsServiceGetTicketResponse{
Ticket: ticket.GetTicket(),
}
l.Debug("GetTicket response", zap.Any("resp", resp))
return c.helpers.FinishTask(ctx, resp, resp.GetAnnotations(), nil)
}
func newGetTicketTaskHandler(task *v1.Task, helpers getTicketTaskHelpers) *getTicketTaskHandler {
return &getTicketTaskHandler{
task: task,
helpers: helpers,
}
}