Совет: в REST-сервисах ты часто получаешь только ID сущностей. И не всегда нужно тащить их из БД.

Чтобы проставить связь (по сути, foreign key в базе), можно “создать” ссылку на сущность через getReferenceById(id) без SELECT и использовать её, чтобы задать relation.

@Service
public SomeService {

    @Autowired
    private ProductRepository productRepository;

    @Autowired
    private CategoryRepository featureRepository;

    public void setCategory(long productId, long categoryId) {
        Product product = productRepository.findById(productId).orElseThrow();  // executes SELECT
        Category category = categoryRepository.getReferenceById(categoryId);    // does not execute a SELECT
        product.setCategory(category);
        productRepository.save(product);
    }

}

👉 Java Portal