Skip to content

Commit b3032e9

Browse files
committed
fixed service init
1 parent 3669cf4 commit b3032e9

3 files changed

Lines changed: 15 additions & 6 deletions

File tree

src/api/api.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import (
77
"github.com/makeopensource/leviathan/common"
88
dkclient "github.com/makeopensource/leviathan/generated/docker_rpc/v1/v1connect"
99
jobClient "github.com/makeopensource/leviathan/generated/jobs/v1/v1connect"
10+
labClient "github.com/makeopensource/leviathan/generated/labs/v1/v1connect"
1011
"github.com/makeopensource/leviathan/service"
1112
"github.com/rs/zerolog/log"
1213
"golang.org/x/net/http2"
@@ -33,7 +34,7 @@ func StartGrpcServer() {
3334
}
3435

3536
func setupEndpoints() *http.ServeMux {
36-
docker, job := service.InitServices()
37+
docker, job, lab := service.InitServices()
3738

3839
interceptor := connect.WithInterceptors()
3940
if common.ApiKey.GetStr() != "" {
@@ -52,6 +53,10 @@ func setupEndpoints() *http.ServeMux {
5253
dkSrv := &v1.DockerServer{Service: docker}
5354
return dkclient.NewDockerServiceHandler(dkSrv, interceptor)
5455
},
56+
func() (string, http.Handler) {
57+
labSrv := v1.LabServer{Srv: lab}
58+
return labClient.NewLabServiceHandler(labSrv, interceptor)
59+
},
5560
}
5661

5762
mux := http.NewServeMux()

src/api/v1/lab_impl.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ import (
1111
)
1212

1313
type LabServer struct {
14-
srv *labs.LabService
14+
Srv *labs.LabService
1515
}
1616

1717
func (l LabServer) NewLab(ctx context.Context, req *connect.Request[v1.LabRequest]) (*connect.Response[v1.NewLabResponse], error) {
@@ -34,7 +34,7 @@ func (l LabServer) NewLab(ctx context.Context, req *connect.Request[v1.LabReques
3434
JobEntryCmd: req.Msg.EntryCommand,
3535
}
3636

37-
labID, err := l.srv.CreateLab(lab, req.Msg.DockerFile, req.Msg.JobFiles)
37+
labID, err := l.Srv.CreateLab(lab, req.Msg.DockerFile, req.Msg.JobFiles)
3838
if err != nil {
3939
return nil, err
4040
}

src/service/service.go

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,16 @@
11
package service
22

33
import (
4+
"github.com/makeopensource/leviathan/common"
45
"github.com/makeopensource/leviathan/service/docker"
56
"github.com/makeopensource/leviathan/service/jobs"
7+
"github.com/makeopensource/leviathan/service/labs"
68
)
79

8-
func InitServices() (*docker.DkService, *jobs.JobService) {
10+
func InitServices() (*docker.DkService, *jobs.JobService, *labs.LabService) {
11+
db, bc := common.InitDB()
912
dkService := docker.NewDockerServiceWithClients()
10-
jobService := jobs.NewJobServiceWithDeps(dkService)
11-
return dkService, jobService
13+
labService := labs.NewLabService(db, dkService)
14+
jobService := jobs.NewJobService(db, bc, dkService, labService)
15+
return dkService, jobService, labService
1216
}

0 commit comments

Comments
 (0)