Improve stage_id foreign key migration

What does this MR do?

This MR improves a migration that adds a foreign key on ci_builds.stage_id.

What are the relevant issue numbers?

Closes gitlab-org/gitlab-ce#34495