CE: Move some EE only codes in models to their EE modules
CE for https://gitlab.com/gitlab-org/gitlab-ee/merge_requests/6276
Not particularly related but similar quick win.
CE for https://gitlab.com/gitlab-org/gitlab-ee/merge_requests/6276
Not particularly related but similar quick win.