mirror of https://github.com/harness/drone.git
45 lines
842 B
Go
45 lines
842 B
Go
package gitrpc
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
"github.com/harness/gitness/internal/gitrpc/rpc"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type Server struct {
|
|
*grpc.Server
|
|
Port int
|
|
}
|
|
|
|
func NewServer(port int) (*Server, error) {
|
|
setting.Git.HomePath = "home"
|
|
adapter, err := newGitea()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
s := grpc.NewServer()
|
|
store := newLocalStore()
|
|
rpc.RegisterRepositoryServiceServer(s, &repositoryService{adapter: adapter, store: store})
|
|
rpc.RegisterUploadServiceServer(s, newUploadService(adapter, store))
|
|
return &Server{
|
|
Server: s,
|
|
Port: port,
|
|
}, nil
|
|
}
|
|
|
|
func (s *Server) Start() error {
|
|
lis, err := net.Listen("tcp", fmt.Sprintf(":%d", s.Port))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.Server.Serve(lis)
|
|
}
|
|
|
|
func (s *Server) Stop() error {
|
|
s.Server.GracefulStop()
|
|
return nil
|
|
}
|