5
5
6
6
"google.golang.org/grpc"
7
7
8
- pb "github.com/cluttrdev/gitlab-exporter/grpc/exporterpb"
8
+ "github.com/cluttrdev/gitlab-exporter/protobuf/servicepb"
9
+ "github.com/cluttrdev/gitlab-exporter/protobuf/typespb"
9
10
)
10
11
11
12
type EndpointConfig struct {
@@ -15,7 +16,7 @@ type EndpointConfig struct {
15
16
16
17
type Client struct {
17
18
conn grpc.ClientConnInterface
18
- client pb .GitLabExporterClient
19
+ client servicepb .GitLabExporterClient
19
20
}
20
21
21
22
func NewCLient (cfg EndpointConfig ) (* Client , error ) {
@@ -24,7 +25,7 @@ func NewCLient(cfg EndpointConfig) (*Client, error) {
24
25
return nil , err
25
26
}
26
27
27
- client := pb .NewGitLabExporterClient (conn )
28
+ client := servicepb .NewGitLabExporterClient (conn )
28
29
29
30
return & Client {
30
31
conn : conn ,
@@ -43,90 +44,90 @@ func send[T any](stream grpc.ClientStream, data []*T) error {
43
44
return err
44
45
}
45
46
46
- m := new (pb .RecordSummary )
47
+ m := new (servicepb .RecordSummary )
47
48
if err := stream .RecvMsg (m ); err != nil {
48
49
return err
49
50
}
50
51
return nil
51
52
}
52
53
53
- func RecordPipelines (c * Client , ctx context.Context , data []* pb .Pipeline ) error {
54
+ func RecordPipelines (c * Client , ctx context.Context , data []* typespb .Pipeline ) error {
54
55
stream , err := c .client .RecordPipelines (ctx )
55
56
if err != nil {
56
57
return err
57
58
}
58
59
59
- return send [pb .Pipeline ](stream , data )
60
+ return send [typespb .Pipeline ](stream , data )
60
61
}
61
62
62
- func RecordJobs (c * Client , ctx context.Context , data []* pb .Job ) error {
63
+ func RecordJobs (c * Client , ctx context.Context , data []* typespb .Job ) error {
63
64
stream , err := c .client .RecordJobs (ctx )
64
65
if err != nil {
65
66
return err
66
67
}
67
68
68
- return send [pb .Job ](stream , data )
69
+ return send [typespb .Job ](stream , data )
69
70
}
70
71
71
- func RecordSections (c * Client , ctx context.Context , data []* pb .Section ) error {
72
+ func RecordSections (c * Client , ctx context.Context , data []* typespb .Section ) error {
72
73
stream , err := c .client .RecordSections (ctx )
73
74
if err != nil {
74
75
return err
75
76
}
76
77
77
- return send [pb .Section ](stream , data )
78
+ return send [typespb .Section ](stream , data )
78
79
}
79
80
80
- func RecordBridges (c * Client , ctx context.Context , data []* pb .Bridge ) error {
81
+ func RecordBridges (c * Client , ctx context.Context , data []* typespb .Bridge ) error {
81
82
stream , err := c .client .RecordBridges (ctx )
82
83
if err != nil {
83
84
return err
84
85
}
85
86
86
- return send [pb .Bridge ](stream , data )
87
+ return send [typespb .Bridge ](stream , data )
87
88
}
88
89
89
- func RecordTestReports (c * Client , ctx context.Context , data []* pb .TestReport ) error {
90
+ func RecordTestReports (c * Client , ctx context.Context , data []* typespb .TestReport ) error {
90
91
stream , err := c .client .RecordTestReports (ctx )
91
92
if err != nil {
92
93
return err
93
94
}
94
95
95
- return send [pb .TestReport ](stream , data )
96
+ return send [typespb .TestReport ](stream , data )
96
97
}
97
98
98
- func RecordTestSuites (c * Client , ctx context.Context , data []* pb .TestSuite ) error {
99
+ func RecordTestSuites (c * Client , ctx context.Context , data []* typespb .TestSuite ) error {
99
100
stream , err := c .client .RecordTestSuites (ctx )
100
101
if err != nil {
101
102
return err
102
103
}
103
104
104
- return send [pb .TestSuite ](stream , data )
105
+ return send [typespb .TestSuite ](stream , data )
105
106
}
106
107
107
- func RecordTestCases (c * Client , ctx context.Context , data []* pb .TestCase ) error {
108
+ func RecordTestCases (c * Client , ctx context.Context , data []* typespb .TestCase ) error {
108
109
stream , err := c .client .RecordTestCases (ctx )
109
110
if err != nil {
110
111
return err
111
112
}
112
113
113
- return send [pb .TestCase ](stream , data )
114
+ return send [typespb .TestCase ](stream , data )
114
115
}
115
116
116
- func RecordMetrics (c * Client , ctx context.Context , data []* pb .Metric ) error {
117
+ func RecordMetrics (c * Client , ctx context.Context , data []* typespb .Metric ) error {
117
118
stream , err := c .client .RecordMetrics (ctx )
118
119
if err != nil {
119
120
return err
120
121
}
121
122
122
- return send [pb .Metric ](stream , data )
123
+ return send [typespb .Metric ](stream , data )
123
124
}
124
125
125
- func RecordTraces (c * Client , ctx context.Context , data []* pb .Trace ) error {
126
+ func RecordTraces (c * Client , ctx context.Context , data []* typespb .Trace ) error {
126
127
stream , err := c .client .RecordTraces (ctx )
127
128
if err != nil {
128
129
return err
129
130
}
130
131
131
- return send [pb .Trace ](stream , data )
132
+ return send [typespb .Trace ](stream , data )
132
133
}
0 commit comments