Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
G
gitlab-ce
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
Analytics
Analytics
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Commits
Issue Boards
Open sidebar
Kirill Smelkov
gitlab-ce
Commits
38d23c0e
Commit
38d23c0e
authored
11 years ago
by
Dmitriy Zaporozhets
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Replace db:backup/restore with native mysq/pg solution
parent
b65903e0
No related merge requests found
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
62 additions
and
41 deletions
+62
-41
lib/backup.rb
lib/backup.rb
+56
-0
lib/tasks/gitlab/backup.rake
lib/tasks/gitlab/backup.rake
+6
-41
No files found.
lib/backup.rb
0 → 100644
View file @
38d23c0e
require
'yaml'
class
Backup
attr_reader
:config
,
:db_dir
def
initialize
@config
=
YAML
.
load_file
(
File
.
join
(
Rails
.
root
,
'config'
,
'database.yml'
))[
Rails
.
env
]
@db_dir
=
File
.
join
(
Gitlab
.
config
.
backup
.
path
,
'db'
)
FileUtils
.
mkdir_p
(
@db_dir
)
unless
Dir
.
exists?
(
@db_dir
)
end
def
backup_db
case
config
[
"adapter"
]
when
/^mysql/
then
system
(
"mysqldump
#{
mysql_args
}
#{
config
[
'database'
]
}
>
#{
db_file_name
}
"
)
when
"postgresql"
then
pg_env
system
(
"pg_dump
#{
config
[
'database'
]
}
>
#{
db_file_name
}
"
)
end
end
def
restore_db
case
config
[
"adapter"
]
when
/^mysql/
then
system
(
"mysql
#{
mysql_args
}
#{
config
[
'database'
]
}
<
#{
db_file_name
}
"
)
when
"postgresql"
then
pg_env
system
(
"pg_restore
#{
config
[
'database'
]
}
#{
db_file_name
}
"
)
end
end
protected
def
db_file_name
File
.
join
(
db_dir
,
'database.sql'
)
end
def
mysql_args
args
=
{
'host'
=>
'--host'
,
'port'
=>
'--port'
,
'socket'
=>
'--socket'
,
'username'
=>
'--user'
,
'encoding'
=>
'--default-character-set'
,
'password'
=>
'--password'
}
args
.
map
{
|
opt
,
arg
|
"
#{
arg
}
=
#{
config
[
opt
]
}
"
if
config
[
opt
]
}.
compact
.
join
(
' '
)
end
def
pg_env
ENV
[
'PGUSER'
]
=
config
[
"username"
]
if
config
[
"username"
]
ENV
[
'PGHOST'
]
=
config
[
"host"
]
if
config
[
"host"
]
ENV
[
'PGPORT'
]
=
config
[
"port"
].
to_s
if
config
[
"port"
]
ENV
[
'PGPASSWORD'
]
=
config
[
"password"
].
to_s
if
config
[
"password"
]
end
end
This diff is collapsed.
Click to expand it.
lib/tasks/gitlab/backup.rake
View file @
38d23c0e
...
...
@@ -109,11 +109,6 @@ namespace :gitlab do
end
end
################################################################################
################################# invoked tasks ################################
################################# REPOSITORIES #################################
namespace
:repo
do
task
:create
=>
:environment
do
backup_path_repo
=
File
.
join
(
Gitlab
.
config
.
backup
.
path
,
"repositories"
)
...
...
@@ -167,48 +162,18 @@ namespace :gitlab do
end
end
###################################### DB ######################################
namespace
:db
do
task
:create
=>
:environment
do
backup_path_db
=
File
.
join
(
Gitlab
.
config
.
backup
.
path
,
"db"
)
FileUtils
.
mkdir_p
(
backup_path_db
)
unless
Dir
.
exists?
(
backup_path_db
)
puts
"Dumping database tables ... "
.
blue
ActiveRecord
::
Base
.
connection
.
tables
.
each
do
|
tbl
|
print
" *
#{
tbl
.
yellow
}
... "
count
=
1
safe_tablename
=
ActiveRecord
::
Base
.
connection
.
quote_table_name
(
tbl
)
File
.
open
(
File
.
join
(
backup_path_db
,
tbl
+
".yml"
),
"w+"
)
do
|
file
|
ActiveRecord
::
Base
.
connection
.
select_all
(
"SELECT * FROM
#{
safe_tablename
}
"
).
each
do
|
line
|
line
.
delete_if
{
|
k
,
v
|
v
.
blank?
}
output
=
{
tbl
+
'_'
+
count
.
to_s
=>
line
}
file
<<
output
.
to_yaml
.
gsub
(
/^---\n/
,
''
)
+
"
\n
"
count
+=
1
end
puts
"done"
.
green
end
end
puts
"Dumping database ... "
.
blue
Backup
.
new
.
backup_db
puts
"done"
.
green
end
task
:restore
=>
:environment
do
backup_path_db
=
File
.
join
(
Gitlab
.
config
.
backup
.
path
,
"db"
)
puts
"Restoring database tables (loading fixtures) ... "
Rake
::
Task
[
"db:reset"
].
invoke
Dir
.
glob
(
File
.
join
(
backup_path_db
,
"*.yml"
)
).
each
do
|
dir
|
fixture_file
=
File
.
basename
(
dir
,
".*"
)
print
"
#{
fixture_file
.
yellow
}
... "
if
File
.
size
(
dir
)
>
0
ActiveRecord
::
Fixtures
.
create_fixtures
(
backup_path_db
,
fixture_file
)
puts
"done"
.
green
else
puts
"skipping"
.
yellow
end
end
puts
"Restoring database ... "
.
blue
Backup
.
new
.
restore_db
puts
"done"
.
green
end
end
end
# namespace end: backup
end
# namespace end: gitlab
This diff is collapsed.
Click to expand it.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment