chore(runner): add new token in header

Signed-off-by: Bo-Yi.Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi.Wu 2022-10-28 23:23:19 +08:00 committed by Jason Song
parent 2f879c41c4
commit d178051832
4 changed files with 29 additions and 1 deletions

View File

@ -77,3 +77,21 @@ func WithUUIDHeader(uuid string) Option {
)
})
}
// WithTokenHeader add runner token in header
func WithTokenHeader(token string) Option {
return OptionFunc(func(cfg *config) {
if token == "" {
return
}
cfg.opts = append(
cfg.opts,
connect.WithInterceptors(connect.UnaryInterceptorFunc(func(next connect.UnaryFunc) connect.UnaryFunc {
return func(ctx context.Context, req connect.AnyRequest) (connect.AnyResponse, error) {
req.Header().Set(core.TokenHeader, token)
return next(ctx, req)
}
})),
)
})
}

View File

@ -82,6 +82,9 @@ func runDaemon(ctx context.Context, envFile string) func(cmd *cobra.Command, arg
if data.UUID != "" {
cfg.Runner.UUID = data.UUID
}
if data.Token != "" {
cfg.Runner.Token = data.Token
}
}
// try to connect to docker daemon
@ -98,6 +101,7 @@ func runDaemon(ctx context.Context, envFile string) func(cmd *cobra.Command, arg
client.WithGRPC(cfg.Client.GRPC),
client.WithGRPCWeb(cfg.Client.GRPCWeb),
client.WithUUIDHeader(cfg.Runner.UUID),
client.WithTokenHeader(cfg.Runner.Token),
)
runner := &runtime.Runner{

View File

@ -69,6 +69,9 @@ func FromEnviron() (Config, error) {
if runner.UUID != "" {
cfg.Runner.UUID = runner.UUID
}
if runner.Token != "" {
cfg.Runner.Token = runner.Token
}
cfg.ForgeInstance = runner.ForgeInstance
}

View File

@ -1,6 +1,9 @@
package core
const UUIDHeader = "x-runner-uuid"
const (
UUIDHeader = "x-runner-uuid"
TokenHeader = "x-runner-token"
)
// Runner struct
type Runner struct {