Method PatchCheckpoints

master/internal/api_checkpoint.go:253–350

(
	ctx context.Context,
	req *apiv1.PatchCheckpointsRequest,
)

}

func (a *apiServer) PatchCheckpoints(
	ctx context.Context,
	req *apiv1.PatchCheckpointsRequest,
) (*apiv1.PatchCheckpointsResponse, error) {
	var uuidStrings []string
	for _, c := range req.Checkpoints {
		uuidStrings = append(uuidStrings, c.Uuid)
	}

	conv := &protoconverter.ProtoConverter{}
	uuids := conv.ToUUIDList(uuidStrings)
	if cErr := conv.Error(); cErr != nil {
		return nil, status.Errorf(codes.InvalidArgument, "converting checkpoint: %s", cErr)
	}

	if _, _, err := a.checkpointsRBACEditCheck(ctx, uuids); err != nil {
		return nil, err
	}

	registeredCheckpointUUIDs, err := checkpoints.GetRegisteredCheckpoints(ctx, uuids)
	if err != nil {
		return nil, err
	}
	if len(registeredCheckpointUUIDs) > 0 {
		errMsg, err := makeRegisteredCheckpointErrorMessage(
			ctx,
			"this subset of checkpoints provided are in the model registry and cannot be patched: %v.",
			registeredCheckpointUUIDs,
		)
		if err != nil {
			return nil, err
		}
		return nil, status.Errorf(codes.InvalidArgument, *errMsg)
	}

	err = internaldb.Bun().RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error {
		var updatedCheckpointSizes []uuid.UUID
		for i, c := range req.Checkpoints {
			if c.Resources != nil {
				size := int64(0)
				for _, v := range c.Resources.Resources {
					size += v
				}

				v2Update := tx.NewUpdate().Model(&model.CheckpointV2{}).
					Where("uuid = ?", c.Uuid)

				if len(c.Resources.Resources) == 0 { // Full delete case.
					v2Update = v2Update.Set("state = ?", model.DeletedState)
				} else { // Partial delete case.
					v2Update = v2Update.
						Set("resources = ?", c.Resources.Resources).
						Set("size = ?", size)

					oldResources := struct {
						bun.BaseModel `bun:"table:checkpoints_view"`
						Resources     map[string]int64
					}{}