Method GetWorkload

pkg/async-gateway/service.go:95–136

GetWorkload retrieves the status and result, if available, of a given workload

(id string, apiName string)

// GetWorkload retrieves the status and result, if available, of a given workload
func (s *service) GetWorkload(id string, apiName string) (GetWorkloadResponse, error) {
	log := s.logger.With(zap.String("id", id), zap.String("apiName", apiName))

	st, err := s.getStatus(id, apiName)
	if err != nil {
		return GetWorkloadResponse{}, err
	}

	if st != async.StatusCompleted {
		return GetWorkloadResponse{
			ID:     id,
			Status: st,
		}, nil
	}

	// attempt to download user result
	prefix := async.StoragePath(s.clusterUID, apiName)
	resultPath := async.ResultPath(prefix, id)
	log.Debug("downloading user result", zap.String("path", resultPath))
	resultBuf, err := s.storage.Download(resultPath)
	if err != nil {
		return GetWorkloadResponse{}, err
	}

	var userResponse UserResponse
	if err = json.Unmarshal(resultBuf, &userResponse); err != nil {
		return GetWorkloadResponse{}, err
	}

	log.Debug("getting workload timestamp")
	timestamp, err := s.storage.GetLastModified(resultPath)
	if err != nil {
		return GetWorkloadResponse{}, err
	}

	return GetWorkloadResponse{
		ID:        id,
		Status:    st,
		Result:    &userResponse,
		Timestamp: &timestamp,
	}, nil
}

func (s *service) getStatus(id string, apiName string) (async.Status, error) {
	prefix := async.StoragePath(s.clusterUID, apiName)