debian-mirror-gitlab/doc/development/github_importer.md

210 lines
9.1 KiB
Markdown
Raw Normal View History

2018-03-17 18:26:18 +05:30
# Working with the GitHub importer
In GitLab 10.2 a new version of the GitHub importer was introduced. This new
importer performs its work in parallel using Sidekiq, greatly reducing the time
necessary to import GitHub projects into a GitLab instance.
The GitHub importer offers two different types of importers: a sequential
importer and a parallel importer. The Rake task `import:github` uses the
sequential importer, while everything else uses the parallel importer. The
difference between these two importers is quite simple: the sequential importer
does all work in a single thread, making it more useful for debugging purposes
2020-04-08 14:13:33 +05:30
or Rake tasks. The parallel importer, on the other hand, uses Sidekiq.
2018-03-17 18:26:18 +05:30
## Requirements
2019-03-02 22:35:43 +05:30
- GitLab CE 10.2.0 or newer.
- Sidekiq workers that process the `github_importer` and
2018-03-17 18:26:18 +05:30
`github_importer_advance_stage` queues (this is enabled by default).
2019-03-02 22:35:43 +05:30
- Octokit (used for interacting with the GitHub API).
2018-03-17 18:26:18 +05:30
## Code structure
The importer's codebase is broken up into the following directories:
2019-03-02 22:35:43 +05:30
- `lib/gitlab/github_import`: this directory contains most of the code such as
2018-03-17 18:26:18 +05:30
the classes used for importing resources.
2019-03-02 22:35:43 +05:30
- `app/workers/gitlab/github_import`: this directory contains the Sidekiq
2018-03-17 18:26:18 +05:30
workers.
2019-03-02 22:35:43 +05:30
- `app/workers/concerns/gitlab/github_import`: this directory contains a few
2018-03-17 18:26:18 +05:30
modules reused by the various Sidekiq workers.
## Architecture overview
2020-04-08 14:13:33 +05:30
When a GitHub project is imported, we schedule and execute a job for the
`RepositoryImportWorker` worker as all other importers. However, unlike other
importers, we don't immediately perform the work necessary. Instead work is
2018-03-17 18:26:18 +05:30
divided into separate stages, with each stage consisting out of a set of Sidekiq
jobs that are executed. Between every stage a job is scheduled that periodically
checks if all work of the current stage is completed, advancing the import
process to the next stage when this is the case. The worker handling this is
called `Gitlab::GithubImport::AdvanceStageWorker`.
## Stages
### 1. RepositoryImportWorker
This worker will kick off the import process by simply scheduling a job for the
next worker.
### 2. Stage::ImportRepositoryWorker
This worker will import the repository and wiki, scheduling the next stage when
done.
### 3. Stage::ImportBaseDataWorker
This worker will import base data such as labels, milestones, and releases. This
work is done in a single thread since it can be performed fast enough that we
don't need to perform this work in parallel.
### 4. Stage::ImportPullRequestsWorker
This worker will import all pull requests. For every pull request a job for the
`Gitlab::GithubImport::ImportPullRequestWorker` worker is scheduled.
### 5. Stage::ImportIssuesAndDiffNotesWorker
2020-04-08 14:13:33 +05:30
This worker will import all issues and pull request comments. For every issue, we
2018-03-17 18:26:18 +05:30
schedule a job for the `Gitlab::GithubImport::ImportIssueWorker` worker. For
2020-04-08 14:13:33 +05:30
pull request comments, we instead schedule jobs for the
2018-03-17 18:26:18 +05:30
`Gitlab::GithubImport::DiffNoteImporter` worker.
This worker processes both issues and diff notes in parallel so we don't need to
schedule a separate stage and wait for the previous one to complete.
Issues are imported separately from pull requests because only the "issues" API
includes labels for both issue and pull requests. Importing issues and setting
label links in the same worker removes the need for performing a separate crawl
through the API data, reducing the number of API calls necessary to import a
project.
### 6. Stage::ImportNotesWorker
This worker imports regular comments for both issues and pull requests. For
2020-04-08 14:13:33 +05:30
every comment, we schedule a job for the
2018-03-17 18:26:18 +05:30
`Gitlab::GithubImport::ImportNoteWorker` worker.
Regular comments have to be imported at the end since the GitHub API used
returns comments for both issues and pull requests. This means we have to wait
for all issues and pull requests to be imported before we can import regular
comments.
### 7. Stage::FinishImportWorker
This worker will wrap up the import process by performing some housekeeping
(such as flushing any caches) and by marking the import as completed.
## Advancing stages
Advancing stages is done in one of two ways:
2019-02-15 15:39:39 +05:30
- Scheduling the worker for the next stage directly.
- Scheduling a job for `Gitlab::GithubImport::AdvanceStageWorker` which will
2018-03-17 18:26:18 +05:30
advance the stage when all work of the current stage has been completed.
The first approach should only be used by workers that perform all their work in
a single thread, while `AdvanceStageWorker` should be used for everything else.
The way `AdvanceStageWorker` works is fairly simple. When scheduling a job it
will be given a project ID, a list of Redis keys, and the name of the next
stage. The Redis keys (produced by `Gitlab::JobWaiter`) are used to check if the
currently running stage has been completed or not. If the stage has not yet been
completed `AdvanceStageWorker` will reschedule itself. Once a stage finishes
`AdvanceStageworker` will refresh the import JID (more on this below) and
schedule the worker of the next stage.
To reduce the number of `AdvanceStageWorker` jobs scheduled this worker will
briefly wait for jobs to complete before deciding what the next action should
2020-04-08 14:13:33 +05:30
be. For small projects, this may slow down the import process a bit, but it will
2018-03-17 18:26:18 +05:30
also reduce pressure on the system as a whole.
## Refreshing import JIDs
2020-06-23 00:09:42 +05:30
GitLab includes a worker called `Gitlab::Import::StuckProjectImportJobsWorker`
that will periodically run and mark project imports as failed if they have been
running for more than 15 hours. For GitHub projects, this poses a bit of a
problem: importing large projects could take several hours depending on how
often we hit the GitHub rate limit (more on this below), but we don't want
`Gitlab::Import::StuckProjectImportJobsWorker` to mark our import as failed because of this.
2018-03-17 18:26:18 +05:30
To prevent this from happening we periodically refresh the expiration time of
the import process. This works by storing the JID of the import job in the
database, then refreshing this JID's TTL at various stages throughout the import
2019-02-15 15:39:39 +05:30
process. This is done by calling `ProjectImportState#refresh_jid_expiration`. By
2018-03-17 18:26:18 +05:30
refreshing this TTL we can ensure our import does not get marked as failed so
long we're still performing work.
## GitHub rate limit
2020-04-08 14:13:33 +05:30
GitHub has a rate limit of 5,000 API calls per hour. The number of requests
2018-03-17 18:26:18 +05:30
necessary to import a project is largely dominated by the number of unique users
involved in a project (e.g. issue authors). Other data such as issue pages
2020-03-13 15:44:24 +05:30
and comments typically only requires a few dozen requests to import. This is
2018-03-17 18:26:18 +05:30
because we need the Email address of users in order to map them to GitLab users.
We handle this by doing the following:
1. Once we hit the rate limit all jobs will automatically reschedule themselves
in such a way that they are not executed until the rate limit has been reset.
2019-02-15 15:39:39 +05:30
1. We cache the mapping of GitHub users to GitLab users in Redis.
2018-03-17 18:26:18 +05:30
More information on user caching can be found below.
## Caching user lookups
When mapping GitHub users to GitLab users we need to (in the worst case)
perform:
1. One API call to get the user's Email address.
2019-02-15 15:39:39 +05:30
1. Two database queries to see if a corresponding GitLab user exists. One query
2018-03-17 18:26:18 +05:30
will try to find the user based on the GitHub user ID, while the second query
is used to find the user using their GitHub Email address.
Because this process is quite expensive we cache the result of these lookups in
Redis. For every user looked up we store three keys:
2019-02-15 15:39:39 +05:30
- A Redis key mapping GitHub usernames to their Email addresses.
- A Redis key mapping a GitHub Email addresses to a GitLab user ID.
- A Redis key mapping a GitHub user ID to GitLab user ID.
2018-03-17 18:26:18 +05:30
There are two types of lookups we cache:
2019-02-15 15:39:39 +05:30
- A positive lookup, meaning we found a GitLab user ID.
- A negative lookup, meaning we didn't find a GitLab user ID. Caching this
2018-03-17 18:26:18 +05:30
prevents us from performing the same work for users that we know don't exist
in our GitLab database.
The expiration time of these keys is 24 hours. When retrieving the cache of a
2020-04-08 14:13:33 +05:30
positive lookup, we refresh the TTL automatically. The TTL of false lookups is
2018-03-17 18:26:18 +05:30
never refreshed.
2020-04-08 14:13:33 +05:30
Because of this caching layer, it's possible newly registered GitLab accounts
won't be linked to their corresponding GitHub accounts. This, however, will sort
2018-03-17 18:26:18 +05:30
itself out once the cached keys expire.
The user cache lookup is shared across projects. This means that the more
projects get imported the fewer GitHub API calls will be needed.
The code for this resides in:
2019-03-02 22:35:43 +05:30
- `lib/gitlab/github_import/user_finder.rb`
- `lib/gitlab/github_import/caching.rb`
2018-03-17 18:26:18 +05:30
## Mapping labels and milestones
To reduce pressure on the database we do not query it when setting labels and
2020-04-08 14:13:33 +05:30
milestones on issues and merge requests. Instead, we cache this data when we
2018-03-17 18:26:18 +05:30
import labels and milestones, then we reuse this cache when assigning them to
issues/merge requests. Similar to the user lookups these cache keys are expired
automatically after 24 hours of not being used.
2020-04-08 14:13:33 +05:30
Unlike the user lookup caches, these label and milestone caches are scoped to the
2018-03-17 18:26:18 +05:30
project that is being imported.
The code for this resides in:
2019-03-02 22:35:43 +05:30
- `lib/gitlab/github_import/label_finder.rb`
- `lib/gitlab/github_import/milestone_finder.rb`
- `lib/gitlab/github_import/caching.rb`