get_jobs.go 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. package release
  2. import (
  3. "fmt"
  4. "net/http"
  5. "github.com/porter-dev/porter/api/server/authz"
  6. "github.com/porter-dev/porter/api/server/handlers"
  7. "github.com/porter-dev/porter/api/server/shared"
  8. "github.com/porter-dev/porter/api/server/shared/apierrors"
  9. "github.com/porter-dev/porter/api/server/shared/config"
  10. "github.com/porter-dev/porter/api/types"
  11. "github.com/porter-dev/porter/internal/kubernetes"
  12. "github.com/porter-dev/porter/internal/models"
  13. "helm.sh/helm/v3/pkg/release"
  14. )
  15. type GetJobsHandler struct {
  16. handlers.PorterHandlerReadWriter
  17. authz.KubernetesAgentGetter
  18. }
  19. func NewGetJobsHandler(
  20. config *config.Config,
  21. writer shared.ResultWriter,
  22. ) *GetJobsHandler {
  23. return &GetJobsHandler{
  24. PorterHandlerReadWriter: handlers.NewDefaultPorterHandler(config, nil, writer),
  25. KubernetesAgentGetter: authz.NewOutOfClusterAgentGetter(config),
  26. }
  27. }
  28. func (c *GetJobsHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
  29. helmRelease, _ := r.Context().Value(types.ReleaseScope).(*release.Release)
  30. cluster, _ := r.Context().Value(types.ClusterScope).(*models.Cluster)
  31. agent, err := c.GetAgent(r, cluster, "")
  32. if err != nil {
  33. c.HandleAPIError(w, r, apierrors.NewErrInternal(err))
  34. return
  35. }
  36. jobs, err := agent.ListJobsByLabel(helmRelease.Namespace, getJobLabels(helmRelease)...)
  37. if err != nil {
  38. c.HandleAPIError(w, r, apierrors.NewErrInternal(err))
  39. return
  40. }
  41. c.WriteResult(w, r, jobs)
  42. }
  43. func getJobLabels(helmRelease *release.Release) []kubernetes.Label {
  44. return []kubernetes.Label{
  45. {
  46. Key: "helm.sh/chart",
  47. Val: fmt.Sprintf("%s-%s", helmRelease.Chart.Metadata.Name, helmRelease.Chart.Metadata.Version),
  48. },
  49. {
  50. Key: "meta.helm.sh/release-name",
  51. Val: helmRelease.Name,
  52. },
  53. }
  54. }