mirror of
https://github.com/go-gitea/gitea.git
synced 2025-08-21 08:04:07 +09:00
81 lines
2.2 KiB
Go
81 lines
2.2 KiB
Go
// Copyright 2022 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package bots
|
|
|
|
import (
|
|
"context"
|
|
|
|
"code.gitea.io/gitea/modules/log"
|
|
"code.gitea.io/gitea/modules/web"
|
|
runnerv1 "gitea.com/gitea/proto-go/runner/v1"
|
|
"gitea.com/gitea/proto-go/runner/v1/runnerv1connect"
|
|
|
|
"github.com/bufbuild/connect-go"
|
|
grpchealth "github.com/bufbuild/connect-grpchealth-go"
|
|
grpcreflect "github.com/bufbuild/connect-grpcreflect-go"
|
|
)
|
|
|
|
type RunnerService struct{}
|
|
|
|
func (s *RunnerService) Connect(
|
|
ctx context.Context,
|
|
req *connect.Request[runnerv1.ConnectRequest],
|
|
) (*connect.Response[runnerv1.ConnectResponse], error) {
|
|
log.Info("Request headers: %v", req.Header())
|
|
res := connect.NewResponse(&runnerv1.ConnectResponse{
|
|
Stage: &runnerv1.Stage{
|
|
RunnerUuid: "foobar",
|
|
BuildUuid: "foobar",
|
|
},
|
|
})
|
|
res.Header().Set("Gitea-Version", "runnerv1")
|
|
return res, nil
|
|
}
|
|
|
|
func (s *RunnerService) Accept(
|
|
ctx context.Context,
|
|
req *connect.Request[runnerv1.AcceptRequest],
|
|
) (*connect.Response[runnerv1.AcceptResponse], error) {
|
|
log.Info("Request headers: %v", req.Header())
|
|
res := connect.NewResponse(&runnerv1.AcceptResponse{
|
|
JobId: 100,
|
|
})
|
|
res.Header().Set("Gitea-Version", "runnerv1")
|
|
return res, nil
|
|
}
|
|
|
|
func runnerServiceRoute(r *web.Route) {
|
|
compress1KB := connect.WithCompressMinBytes(1024)
|
|
|
|
runnerService := &RunnerService{}
|
|
connectPath, connecthandler := runnerv1connect.NewRunnerServiceHandler(
|
|
runnerService,
|
|
compress1KB,
|
|
)
|
|
|
|
// grpcV1
|
|
grpcPath, gHandler := grpcreflect.NewHandlerV1(
|
|
grpcreflect.NewStaticReflector(runnerv1connect.RunnerServiceName),
|
|
compress1KB,
|
|
)
|
|
|
|
// grpcV1Alpha
|
|
grpcAlphaPath, gAlphaHandler := grpcreflect.NewHandlerV1Alpha(
|
|
grpcreflect.NewStaticReflector(runnerv1connect.RunnerServiceName),
|
|
compress1KB,
|
|
)
|
|
|
|
// grpcHealthCheck
|
|
grpcHealthPath, gHealthHandler := grpchealth.NewHandler(
|
|
grpchealth.NewStaticChecker(runnerv1connect.RunnerServiceName),
|
|
compress1KB,
|
|
)
|
|
|
|
r.Post(connectPath+"{name}", grpcHandler(connecthandler))
|
|
r.Post(grpcPath+"{name}", grpcHandler(gHandler))
|
|
r.Post(grpcAlphaPath+"{name}", grpcHandler(gAlphaHandler))
|
|
r.Post(grpcHealthPath+"{name}", grpcHandler(gHealthHandler))
|
|
}
|