implemented getStatus rpc

Author: RA341

src/api/v1/job_impl.go

@@ -63,15 +63,24 @@ func (job *JobServer) NewJob(ctx context.Context, req *connect.Request[v1.NewJob
 	return res, nil
 }
 
+func (job *JobServer) GetStatus(_ context.Context, req *connect.Request[v1.JobLogRequest]) (*connect.Response[v1.JobLogsResponse], error) {
+	status, logs, err := job.srv.GetJobStatusAndLogs(req.Msg.GetJobId())
+	if err != nil {
+		return nil, err
+	}
+
+	res := connect.NewResponse(&v1.JobLogsResponse{
+		JobInfo: status.ToProto(),
+		Logs:    logs,
+	})
+	return res, nil
+}
+
 func (job *JobServer) StreamStatus(ctx context.Context, req *connect.Request[v1.JobLogRequest], stream *connect.ServerStream[v1.JobLogsResponse]) error {
 	streamFunc := func(jobInfo *models.Job, logs string) error {
 		return stream.Send(&v1.JobLogsResponse{
-			JobInfo: &v1.JobStatus{
-				JobId:         jobInfo.JobId,
-				Status:        string(jobInfo.Status),
-				StatusMessage: jobInfo.StatusMessage,
-			},
-			Logs: logs,
+			JobInfo: jobInfo.ToProto(),
+			Logs:    logs,
 		})
 	}
 

src/models/job.go

@@ -3,6 +3,7 @@ package models
 import (
 	"context"
 	"fmt"
+	v1 "github.com/makeopensource/leviathan/generated/jobs/v1"
 	"github.com/rs/zerolog/log"
 	"gorm.io/gorm"
 )
@@ -63,6 +64,14 @@ type Job struct {
 	JobCtx           context.Context `gorm:"-"`
 }
 
+func (j *Job) ToProto() *v1.JobStatus {
+	return &v1.JobStatus{
+		JobId:         j.JobId,
+		Status:        string(j.Status),
+		StatusMessage: j.StatusMessage,
+	}
+}
+
 // ValidateForQueue checks for fields required before sending job to queue
 func (j *Job) ValidateForQueue() error {
 	if j == nil {

src/service/jobs/job_service.go

@@ -249,6 +249,15 @@ func (job *JobService) StreamJobAndLogs(
 	}
 }
 
+// GetJobStatusAndLogs gets the status once whatever it may be and current logs
+func (job *JobService) GetJobStatusAndLogs(jobUuid string) (*models.Job, string, error) {
+	jobInfo, _, logs, err := job.checkJob(jobUuid)
+	if err != nil {
+		return nil, "", err
+	}
+	return jobInfo, logs, nil
+}
+
 func (job *JobService) ListenToJobLogs(ctx context.Context, jobInfo *models.Job) chan string {
 	logChannel := make(chan string, 2)
 	go func(ctx context.Context) {