pass version when registering and declaring

This commit is contained in:
sillyguodong 2023-05-20 20:22:22 +08:00
parent 8427d62126
commit 9a0d46fa66
No known key found for this signature in database
GPG key ID: 4A0646FA86256DF4
3 changed files with 12 additions and 8 deletions
internal
app
pkg/client

View file

@ -337,9 +337,10 @@ func doRegister(cfg *config.Config, inputs *registerInputs) error {
}
// register new runner.
resp, err := cli.Register(ctx, connect.NewRequest(&runnerv1.RegisterRequest{
Name: reg.Name,
Token: reg.Token,
Labels: ls,
Name: reg.Name,
Token: reg.Token,
Version: ver.Version(),
Labels: ls,
}))
if err != nil {
log.WithError(err).Error("poller: cannot register new runner")

View file

@ -217,11 +217,14 @@ func (r *Runner) run(ctx context.Context, task *runnerv1.Task, reporter *report.
}
func (r *Runner) Declare(ctx context.Context, labels []string) error {
_, err := r.client.Declare(ctx, connect.NewRequest(&runnerv1.DeclareRequest{
Labels: labels,
resp, err := r.client.Declare(ctx, connect.NewRequest(&runnerv1.DeclareRequest{
Version: ver.Version(),
Labels: labels,
}))
if err != nil {
return err
}
log.Info("runner [%s], with version: [%s], with labels: [%v], declare successfully",
resp.Msg.Runner.Name, resp.Msg.Runner.Version, resp.Msg.Runner.Labels)
return nil
}

View file

@ -39,9 +39,9 @@ func New(endpoint string, insecure bool, uuid, token, version string, opts ...co
if token != "" {
req.Header().Set(TokenHeader, token)
}
if version != "" {
req.Header().Set(VersionHeader, version)
}
// if version != "" {
// req.Header().Set(VersionHeader, version)
// }
return next(ctx, req)
}
})))