chore(runner): get token from header

Signed-off-by: Bo-Yi.Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi.Wu 2022-10-10 17:17:23 +08:00 committed by Jason Song
parent f79bc4f605
commit 4233090738

View File

@ -35,7 +35,9 @@ func (s *Service) Register(
ctx context.Context,
req *connect.Request[runnerv1.RegisterRequest],
) (*connect.Response[runnerv1.RegisterResponse], error) {
if req.Msg.Token == "" || req.Msg.Name == "" || req.Msg.Url == "" {
token := req.Header().Get("X-Runner-Token")
if token == "" || req.Msg.Name == "" || req.Msg.Url == "" {
return nil, errors.New("missing runner token, name or URL")
}