2015-02-12 08:28:37 +05:30
// Copyright 2015 The Gogs Authors. All rights reserved.
2019-02-18 21:30:27 +05:30
// Copyright 2017 The Gitea Authors. All rights reserved.
2015-02-12 08:28:37 +05:30
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
2015-01-22 18:19:52 +05:30
package migrations
import (
2015-08-11 20:54:40 +05:30
"bytes"
2015-07-26 19:36:28 +05:30
"encoding/json"
2015-02-12 08:28:37 +05:30
"fmt"
2015-08-11 20:54:40 +05:30
"io/ioutil"
"os"
"path"
2015-11-01 08:48:58 +05:30
"path/filepath"
2019-07-02 00:56:59 +05:30
"regexp"
2015-01-23 13:24:16 +05:30
"strings"
2016-03-10 06:23:30 +05:30
"time"
2015-01-22 18:26:50 +05:30
2018-02-18 23:44:37 +05:30
"code.gitea.io/gitea/modules/generate"
2016-11-10 21:54:48 +05:30
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/setting"
2019-08-23 22:10:30 +05:30
gouuid "github.com/satori/go.uuid"
"github.com/unknwon/com"
ini "gopkg.in/ini.v1"
2019-10-17 14:56:49 +05:30
"xorm.io/xorm"
2015-01-22 18:19:52 +05:30
)
2016-11-28 21:14:17 +05:30
const minDBVersion = 4
2015-02-12 08:28:37 +05:30
2016-11-28 21:14:17 +05:30
// Migration describes on migration from lower version to high version
2015-02-12 09:40:30 +05:30
type Migration interface {
Description ( ) string
Migrate ( * xorm . Engine ) error
}
type migration struct {
description string
migrate func ( * xorm . Engine ) error
}
2016-11-28 21:14:17 +05:30
// NewMigration creates a new migration
2015-02-12 09:40:30 +05:30
func NewMigration ( desc string , fn func ( * xorm . Engine ) error ) Migration {
return & migration { desc , fn }
}
2016-11-28 21:14:17 +05:30
// Description returns the migration's description
2015-02-12 09:40:30 +05:30
func ( m * migration ) Description ( ) string {
return m . description
}
2016-11-28 21:14:17 +05:30
// Migrate executes the migration
2015-02-12 09:40:30 +05:30
func ( m * migration ) Migrate ( x * xorm . Engine ) error {
return m . migrate ( x )
}
2015-01-22 18:19:52 +05:30
2016-11-28 21:14:17 +05:30
// Version describes the version table. Should have only one row with id==1
2015-01-22 18:19:52 +05:30
type Version struct {
2016-07-16 07:38:04 +05:30
ID int64 ` xorm:"pk autoincr" `
2015-01-22 18:19:52 +05:30
Version int64
}
2017-12-13 20:22:18 +05:30
func emptyMigration ( x * xorm . Engine ) error {
return nil
}
2015-01-22 18:19:52 +05:30
// This is a sequence of migrations. Add new migrations to the bottom of the list.
2015-02-12 23:16:21 +05:30
// If you want to "retire" a migration, remove it from the top of the list and
2016-11-28 21:14:17 +05:30
// update minDBVersion accordingly
2015-02-12 09:40:30 +05:30
var migrations = [ ] Migration {
2016-08-31 17:01:53 +05:30
// v0 -> v4: before 0.6.0 -> 0.7.33
2016-07-08 04:55:09 +05:30
NewMigration ( "fix locale file load panic" , fixLocaleFileLoadPanic ) , // V4 -> V5:v0.6.0
2016-11-28 21:14:17 +05:30
NewMigration ( "trim action compare URL prefix" , trimCommitActionAppURLPrefix ) , // V5 -> V6:v0.6.3
2016-07-08 04:55:09 +05:30
NewMigration ( "generate issue-label from issue" , issueToIssueLabel ) , // V6 -> V7:v0.6.4
NewMigration ( "refactor attachment table" , attachmentRefactor ) , // V7 -> V8:v0.6.4
NewMigration ( "rename pull request fields" , renamePullRequestFields ) , // V8 -> V9:v0.6.16
NewMigration ( "clean up migrate repo info" , cleanUpMigrateRepoInfo ) , // V9 -> V10:v0.6.20
NewMigration ( "generate rands and salt for organizations" , generateOrgRandsAndSalt ) , // V10 -> V11:v0.8.5
NewMigration ( "convert date to unix timestamp" , convertDateToUnix ) , // V11 -> V12:v0.9.2
NewMigration ( "convert LDAP UseSSL option to SecurityProtocol" , ldapUseSSLToSecurityProtocol ) , // V12 -> V13:v0.9.37
2016-08-27 02:37:21 +05:30
// v13 -> v14:v0.9.87
NewMigration ( "set comment updated with created" , setCommentUpdatedWithCreated ) ,
2017-02-04 21:23:46 +05:30
// v14 -> v15
2016-11-13 08:24:04 +05:30
NewMigration ( "create user column diff view style" , createUserColumnDiffViewStyle ) ,
2017-02-04 21:23:46 +05:30
// v15 -> v16
2016-12-31 08:03:30 +05:30
NewMigration ( "create user column allow create organization" , createAllowCreateOrganizationColumn ) ,
2017-02-04 21:23:46 +05:30
// V16 -> v17
NewMigration ( "create repo unit table and add units for all repos" , addUnitsToTables ) ,
2017-02-21 20:32:10 +05:30
// v17 -> v18
NewMigration ( "set protect branches updated with created" , setProtectedBranchUpdatedWithCreated ) ,
2017-02-22 12:44:37 +05:30
// v18 -> v19
NewMigration ( "add external login user" , addExternalLoginUser ) ,
2017-02-23 09:10:44 +05:30
// v19 -> v20
NewMigration ( "generate and migrate Git hooks" , generateAndMigrateGitHooks ) ,
2017-02-25 20:28:57 +05:30
// v20 -> v21
2017-02-26 09:28:02 +05:30
NewMigration ( "use new avatar path name for security reason" , useNewNameAvatars ) ,
2017-03-02 22:06:47 +05:30
// v21 -> v22
NewMigration ( "rewrite authorized_keys file via new format" , useNewPublickeyFormat ) ,
2017-03-14 12:09:02 +05:30
// v22 -> v23
2017-03-13 11:57:29 +05:30
NewMigration ( "generate and migrate wiki Git hooks" , generateAndMigrateWikiGitHooks ) ,
2017-03-17 19:46:08 +05:30
// v23 -> v24
NewMigration ( "add user openid table" , addUserOpenID ) ,
2017-03-17 19:54:51 +05:30
// v24 -> v25
NewMigration ( "change the key_id and primary_key_id type" , changeGPGKeysColumns ) ,
2017-03-20 14:01:08 +05:30
// v25 -> v26
NewMigration ( "add show field in user openid table" , addUserOpenIDShow ) ,
2017-03-23 06:42:51 +05:30
// v26 -> v27
NewMigration ( "generate and migrate repo and wiki Git hooks" , generateAndMigrateGitHookChains ) ,
2017-04-08 20:57:26 +05:30
// v27 -> v28
NewMigration ( "change mirror interval from hours to time.Duration" , convertIntervalToDuration ) ,
2017-04-11 19:00:15 +05:30
// v28 -> v29
NewMigration ( "add field for repo size" , addRepoSize ) ,
2017-04-21 17:02:31 +05:30
// v29 -> v30
NewMigration ( "add commit status table" , addCommitStatus ) ,
2017-05-02 14:11:44 +05:30
// v30 -> 31
NewMigration ( "add primary key to external login user" , addExternalLoginUserPK ) ,
2017-11-29 05:05:23 +05:30
// v31 -> 32
2017-05-10 18:40:18 +05:30
NewMigration ( "add field for login source synchronization" , addLoginSourceSyncEnabledColumn ) ,
2017-05-18 20:24:24 +05:30
// v32 -> v33
NewMigration ( "add units for team" , addUnitsToRepoTeam ) ,
2017-05-26 07:08:18 +05:30
// v33 -> v34
NewMigration ( "remove columns from action" , removeActionColumns ) ,
2017-06-05 16:10:25 +05:30
// v34 -> v35
NewMigration ( "give all units to owner teams" , giveAllUnitsToOwnerTeams ) ,
2017-06-25 23:50:29 +05:30
// v35 -> v36
NewMigration ( "adds comment to an action" , addCommentIDToAction ) ,
2017-07-02 19:20:57 +05:30
// v36 -> v37
NewMigration ( "regenerate git hooks" , regenerateGitHooks36 ) ,
2017-07-12 20:28:52 +05:30
// v37 -> v38
NewMigration ( "unescape user full names" , unescapeUserFullNames ) ,
2017-07-17 07:34:43 +05:30
// v38 -> v39
NewMigration ( "remove commits and settings unit types" , removeCommitsUnitType ) ,
2017-09-12 12:18:13 +05:30
// v39 -> v40
2017-12-13 20:22:18 +05:30
NewMigration ( "add tags to releases and sync existing repositories" , releaseAddColumnIsTagAndSyncTags ) ,
2017-09-14 13:46:22 +05:30
// v40 -> v41
2017-12-13 20:22:18 +05:30
NewMigration ( "fix protected branch can push value to false" , fixProtectedBranchCanPushValue ) ,
2017-09-16 05:48:25 +05:30
// v41 -> v42
2017-12-13 20:22:18 +05:30
NewMigration ( "remove duplicate unit types" , removeDuplicateUnitTypes ) ,
2017-09-20 10:56:49 +05:30
// v42 -> v43
2017-12-13 20:22:18 +05:30
NewMigration ( "empty step" , emptyMigration ) ,
2017-09-20 20:22:23 +05:30
// v43 -> v44
2017-12-13 20:22:18 +05:30
NewMigration ( "empty step" , emptyMigration ) ,
2017-09-28 18:44:51 +05:30
// v44 -> v45
2017-12-13 20:22:18 +05:30
NewMigration ( "empty step" , emptyMigration ) ,
2017-10-03 01:52:25 +05:30
// v45 -> v46
NewMigration ( "remove index column from repo_unit table" , removeIndexColumnFromRepoUnitTable ) ,
2017-10-14 20:07:43 +05:30
// v46 -> v47
NewMigration ( "remove organization watch repositories" , removeOrganizationWatchRepo ) ,
2017-10-26 06:19:16 +05:30
// v47 -> v48
NewMigration ( "add deleted branches" , addDeletedBranch ) ,
2017-10-27 11:40:54 +05:30
// v48 -> v49
NewMigration ( "add repo indexer status" , addRepoIndexerStatus ) ,
2017-11-29 05:05:23 +05:30
// v49 -> v50
2017-12-13 20:22:18 +05:30
NewMigration ( "adds time tracking and stopwatches" , addTimetracking ) ,
2017-12-04 04:44:26 +05:30
// v50 -> v51
2017-12-13 20:22:18 +05:30
NewMigration ( "migrate protected branch struct" , migrateProtectedBranchStruct ) ,
// v51 -> v52
NewMigration ( "add default value to user prohibit_login" , addDefaultValueToUserProhibitLogin ) ,
// v52 -> v53
NewMigration ( "add lfs lock table" , addLFSLock ) ,
// v53 -> v54
2017-12-04 04:44:26 +05:30
NewMigration ( "add reactions" , addReactions ) ,
2018-01-06 00:26:50 +05:30
// v54 -> v55
NewMigration ( "add pull request options" , addPullRequestOptions ) ,
2018-01-07 04:25:53 +05:30
// v55 -> v56
NewMigration ( "add writable deploy keys" , addModeToDeploKeys ) ,
2018-01-08 13:18:37 +05:30
// v56 -> v57
NewMigration ( "remove is_owner, num_teams columns from org_user" , removeIsOwnerColumnFromOrgUser ) ,
2018-02-19 08:09:26 +05:30
// v57 -> v58
NewMigration ( "add closed_unix column for issues" , addIssueClosedTime ) ,
2018-03-13 07:33:55 +05:30
// v58 -> v59
NewMigration ( "add label descriptions" , addLabelsDescriptions ) ,
2018-03-25 15:31:32 +05:30
// v59 -> v60
NewMigration ( "add merge whitelist for protected branches" , addProtectedBranchMergeWhitelist ) ,
2018-03-27 19:43:20 +05:30
// v60 -> v61
NewMigration ( "add is_fsck_enabled column for repos" , addFsckEnabledToRepo ) ,
2018-03-31 06:40:44 +05:30
// v61 -> v62
NewMigration ( "add size column for attachments" , addSizeToAttachment ) ,
2018-05-02 20:32:02 +05:30
// v62 -> v63
NewMigration ( "add last used passcode column for TOTP" , addLastUsedPasscodeTOTP ) ,
2018-05-05 05:58:30 +05:30
// v63 -> v64
NewMigration ( "add language column for user setting" , addLanguageSetting ) ,
2018-05-09 21:59:04 +05:30
// v64 -> v65
NewMigration ( "add multiple assignees" , addMultipleAssignees ) ,
2018-05-19 19:42:37 +05:30
// v65 -> v66
NewMigration ( "add u2f" , addU2FReg ) ,
2018-05-24 10:29:02 +05:30
// v66 -> v67
NewMigration ( "add login source id column for public_key table" , addLoginSourceIDToPublicKeyTable ) ,
2018-06-20 01:14:33 +05:30
// v67 -> v68
NewMigration ( "remove stale watches" , removeStaleWatches ) ,
2018-06-21 14:39:46 +05:30
// v68 -> V69
NewMigration ( "Reformat and remove incorrect topics" , reformatAndRemoveIncorrectTopics ) ,
2018-06-21 21:30:13 +05:30
// v69 -> v70
NewMigration ( "move team units to team_unit table" , moveTeamUnitsToTeamUnitTable ) ,
2018-07-18 02:53:58 +05:30
// v70 -> v71
NewMigration ( "add issue_dependencies" , addIssueDependencies ) ,
2018-08-06 10:13:22 +05:30
// v71 -> v72
2018-07-27 18:24:50 +05:30
NewMigration ( "protect each scratch token" , addScratchHash ) ,
2018-08-06 10:13:22 +05:30
// v72 -> v73
NewMigration ( "add review" , addReview ) ,
2018-09-13 17:34:25 +05:30
// v73 -> v74
NewMigration ( "add must_change_password column for users table" , addMustChangePassword ) ,
2018-12-11 16:58:37 +05:30
// v74 -> v75
NewMigration ( "add approval whitelists to protected branches" , addApprovalWhitelistsToProtectedBranches ) ,
2018-12-18 21:56:26 +05:30
// v75 -> v76
NewMigration ( "clear nonused data which not deleted when user was deleted" , clearNonusedData ) ,
2018-12-27 15:57:08 +05:30
// v76 -> v77
NewMigration ( "add pull request rebase with merge commit" , addPullRequestRebaseWithMerge ) ,
2019-01-09 22:52:57 +05:30
// v77 -> v78
NewMigration ( "add theme to users" , addUserDefaultTheme ) ,
2019-01-18 05:31:04 +05:30
// v78 -> v79
NewMigration ( "rename repo is_bare to repo is_empty" , renameRepoIsBareToIsEmpty ) ,
2019-02-11 00:57:19 +05:30
// v79 -> v80
NewMigration ( "add can close issues via commit in any branch" , addCanCloseIssuesViaCommitInAnyBranch ) ,
2019-02-19 02:25:04 +05:30
// v80 -> v81
NewMigration ( "add is locked to issues" , addIsLockedToIssues ) ,
2019-03-05 08:04:52 +05:30
// v81 -> v82
NewMigration ( "update U2F counter type" , changeU2FCounterType ) ,
2019-03-11 09:14:58 +05:30
// v82 -> v83
NewMigration ( "hot fix for wrong release sha1 on release table" , fixReleaseSha1OnReleaseTable ) ,
2019-04-03 00:55:05 +05:30
// v83 -> v84
NewMigration ( "add uploader id for table attachment" , addUploaderIDForAttachment ) ,
2019-04-14 22:13:56 +05:30
// v84 -> v85
NewMigration ( "add table to store original imported gpg keys" , addGPGKeyImport ) ,
2019-05-04 21:15:34 +05:30
// v85 -> v86
NewMigration ( "hash application token" , hashAppToken ) ,
2019-05-05 23:39:02 +05:30
// v86 -> v87
NewMigration ( "add http method to webhook" , addHTTPMethodToWebhook ) ,
2019-05-30 07:52:26 +05:30
// v87 -> v88
NewMigration ( "add avatar field to repository" , addAvatarFieldToRepository ) ,
2019-06-30 13:27:59 +05:30
// v88 -> v89
NewMigration ( "add commit status context field to commit_status" , addCommitStatusContext ) ,
2019-07-08 07:44:12 +05:30
// v89 -> v90
NewMigration ( "add original author/url migration info to issues, comments, and repo " , addOriginalMigrationInfo ) ,
2019-07-29 08:59:42 +05:30
// v90 -> v91
NewMigration ( "change length of some repository columns" , changeSomeColumnsLengthOfRepo ) ,
2019-08-05 19:59:40 +05:30
// v91 -> v92
NewMigration ( "add index on owner_id of repository and type, review_id of comment" , addIndexOnRepositoryAndComment ) ,
2019-08-06 14:27:55 +05:30
// v92 -> v93
NewMigration ( "remove orphaned repository index statuses" , removeLingeringIndexStatus ) ,
2019-08-29 19:35:42 +05:30
// v93 -> v94
NewMigration ( "add email notification enabled preference to user" , addEmailNotificationEnabledToUser ) ,
2019-09-18 11:09:45 +05:30
// v94 -> v95
NewMigration ( "add enable_status_check, status_check_contexts to protected_branch" , addStatusCheckColumnsForProtectedBranches ) ,
2019-09-20 11:15:38 +05:30
// v95 -> v96
NewMigration ( "add table columns for cross referencing issues" , addCrossReferenceColumns ) ,
2019-09-22 14:35:48 +05:30
// v96 -> v97
NewMigration ( "delete orphaned attachments" , deleteOrphanedAttachments ) ,
2019-09-24 01:38:03 +05:30
// v97 -> v98
NewMigration ( "add repo_admin_change_team_access to user" , addRepoAdminChangeTeamAccessColumnForUser ) ,
2019-10-05 16:39:27 +05:30
// v98 -> v99
NewMigration ( "add original author name and id on migrated release" , addOriginalAuthorOnMigratedReleases ) ,
2019-10-13 18:53:14 +05:30
// v99 -> v100
NewMigration ( "add task table and status column for repository table" , addTaskTable ) ,
2019-10-14 11:40:42 +05:30
// v100 -> v101
NewMigration ( "update migration repositories' service type" , updateMigrationServiceTypes ) ,
2019-10-18 12:28:36 +05:30
// v101 -> v102
NewMigration ( "change length of some external login users columns" , changeSomeColumnsLengthOfExternalLoginUser ) ,
2019-10-18 16:43:31 +05:30
// v102 -> v103
NewMigration ( "update migration repositories' service type" , dropColumnHeadUserNameOnPullRequest ) ,
2019-10-21 13:51:45 +05:30
// v103 -> v104
NewMigration ( "Add WhitelistDeployKeys to protected branch" , addWhitelistDeployKeysToBranches ) ,
2019-10-23 17:18:32 +05:30
// v104 -> v105
NewMigration ( "remove unnecessary columns from label" , removeLabelUneededCols ) ,
2019-11-06 15:07:14 +05:30
// v105 -> v106
NewMigration ( "add includes_all_repositories to teams" , addTeamIncludesAllRepositories ) ,
2019-11-10 14:52:19 +05:30
// v106 -> v107
NewMigration ( "add column `mode` to table watch" , addModeColumnToWatch ) ,
2019-11-11 20:45:29 +05:30
// v107 -> v108
NewMigration ( "Add template options to repository" , addTemplateToRepo ) ,
2019-11-12 14:03:34 +05:30
// v108 -> v109
NewMigration ( "Add comment_id on table notification" , addCommentIDOnNotification ) ,
2019-11-20 16:57:49 +05:30
// v109 -> v110
NewMigration ( "add can_create_org_repo to team" , addCanCreateOrgRepoColumnForTeam ) ,
2019-12-03 00:02:40 +05:30
// v110 -> v111
NewMigration ( "change review content type to text" , changeReviewContentToText ) ,
2019-12-04 06:38:56 +05:30
// v111 -> v112
NewMigration ( "update branch protection for can push and whitelist enable" , addBranchProtectionCanPushAndEnableWhitelist ) ,
2019-12-14 09:00:39 +05:30
// v112 -> v113
NewMigration ( "remove release attachments which repository deleted" , removeAttachmentMissedRepo ) ,
2019-12-16 11:50:25 +05:30
// v113 -> v114
NewMigration ( "new feature: change target branch of pull requests" , featureChangeTargetBranch ) ,
2015-01-23 13:24:16 +05:30
}
2015-01-22 18:19:52 +05:30
// Migrate database to current version
func Migrate ( x * xorm . Engine ) error {
2015-01-22 18:31:45 +05:30
if err := x . Sync ( new ( Version ) ) ; err != nil {
2015-02-12 08:28:37 +05:30
return fmt . Errorf ( "sync: %v" , err )
2015-01-22 18:31:45 +05:30
}
2015-01-22 18:19:52 +05:30
2016-07-16 07:38:04 +05:30
currentVersion := & Version { ID : 1 }
2015-01-22 18:19:52 +05:30
has , err := x . Get ( currentVersion )
if err != nil {
2015-02-12 08:28:37 +05:30
return fmt . Errorf ( "get: %v" , err )
2015-01-22 18:26:50 +05:30
} else if ! has {
2015-12-11 06:22:06 +05:30
// If the version record does not exist we think
// it is a fresh installation and we can skip all migrations.
2016-12-24 07:07:35 +05:30
currentVersion . ID = 0
2016-11-28 21:14:17 +05:30
currentVersion . Version = int64 ( minDBVersion + len ( migrations ) )
2015-01-23 13:24:16 +05:30
2015-01-22 18:26:50 +05:30
if _ , err = x . InsertOne ( currentVersion ) ; err != nil {
2015-02-12 08:28:37 +05:30
return fmt . Errorf ( "insert: %v" , err )
2015-01-22 18:26:50 +05:30
}
2015-01-22 18:19:52 +05:30
}
v := currentVersion . Version
2016-11-28 21:14:17 +05:30
if minDBVersion > v {
2019-04-02 13:18:31 +05:30
log . Fatal ( ` Gitea no longer supports auto - migration from your previously installed version .
2015-12-11 06:22:06 +05:30
Please try to upgrade to a lower version ( >= v0 .6 .0 ) first , then upgrade to current version . ` )
2015-11-25 19:57:27 +05:30
return nil
}
2016-11-28 21:14:17 +05:30
if int ( v - minDBVersion ) > len ( migrations ) {
2016-12-28 14:03:21 +05:30
// User downgraded Gitea.
2016-11-28 21:14:17 +05:30
currentVersion . Version = int64 ( len ( migrations ) + minDBVersion )
2017-10-05 10:13:04 +05:30
_ , err = x . ID ( 1 ) . Update ( currentVersion )
2015-08-11 20:54:40 +05:30
return err
2015-08-10 20:29:12 +05:30
}
2016-11-28 21:14:17 +05:30
for i , m := range migrations [ v - minDBVersion : ] {
2019-05-06 05:12:29 +05:30
log . Info ( "Migration[%d]: %s" , v + int64 ( i ) , m . Description ( ) )
2015-02-12 09:40:30 +05:30
if err = m . Migrate ( x ) ; err != nil {
return fmt . Errorf ( "do migrate: %v" , err )
2015-01-22 18:19:52 +05:30
}
currentVersion . Version = v + int64 ( i ) + 1
2017-10-05 10:13:04 +05:30
if _ , err = x . ID ( 1 ) . Update ( currentVersion ) ; err != nil {
2015-01-22 18:31:45 +05:30
return err
}
2015-01-22 18:19:52 +05:30
}
return nil
}
2018-05-09 21:59:04 +05:30
func dropTableColumns ( sess * xorm . Session , tableName string , columnNames ... string ) ( err error ) {
2018-03-07 12:14:12 +05:30
if tableName == "" || len ( columnNames ) == 0 {
return nil
}
2019-07-02 00:56:59 +05:30
// TODO: This will not work if there are foreign keys
2018-03-07 12:14:12 +05:30
switch {
2019-08-24 14:54:45 +05:30
case setting . Database . UseSQLite3 :
2019-07-02 00:56:59 +05:30
// First drop the indexes on the columns
res , errIndex := sess . Query ( fmt . Sprintf ( "PRAGMA index_list(`%s`)" , tableName ) )
if errIndex != nil {
return errIndex
}
for _ , row := range res {
indexName := row [ "name" ]
indexRes , err := sess . Query ( fmt . Sprintf ( "PRAGMA index_info(`%s`)" , indexName ) )
if err != nil {
return err
}
if len ( indexRes ) != 1 {
continue
}
indexColumn := string ( indexRes [ 0 ] [ "name" ] )
for _ , name := range columnNames {
if name == indexColumn {
_ , err := sess . Exec ( fmt . Sprintf ( "DROP INDEX `%s`" , indexName ) )
if err != nil {
return err
}
}
}
}
// Here we need to get the columns from the original table
sql := fmt . Sprintf ( "SELECT sql FROM sqlite_master WHERE tbl_name='%s' and type='table'" , tableName )
res , err := sess . Query ( sql )
if err != nil {
return err
}
tableSQL := string ( res [ 0 ] [ "sql" ] )
2019-08-06 03:19:49 +05:30
// Separate out the column definitions
2019-07-02 00:56:59 +05:30
tableSQL = tableSQL [ strings . Index ( tableSQL , "(" ) : ]
2019-08-06 03:19:49 +05:30
// Remove the required columnNames
2019-07-02 00:56:59 +05:30
for _ , name := range columnNames {
2019-08-06 03:19:49 +05:30
tableSQL = regexp . MustCompile ( regexp . QuoteMeta ( "`" + name + "`" ) + "[^`,)]*?[,)]" ) . ReplaceAllString ( tableSQL , "" )
}
// Ensure the query is ended properly
tableSQL = strings . TrimSpace ( tableSQL )
if tableSQL [ len ( tableSQL ) - 1 ] != ')' {
if tableSQL [ len ( tableSQL ) - 1 ] == ',' {
tableSQL = tableSQL [ : len ( tableSQL ) - 1 ]
}
tableSQL += ")"
2019-07-02 00:56:59 +05:30
}
2019-08-06 03:19:49 +05:30
// Find all the columns in the table
2019-07-02 00:56:59 +05:30
columns := regexp . MustCompile ( "`([^`]*)`" ) . FindAllString ( tableSQL , - 1 )
tableSQL = fmt . Sprintf ( "CREATE TABLE `new_%s_new` " , tableName ) + tableSQL
if _ , err := sess . Exec ( tableSQL ) ; err != nil {
return err
}
// Now restore the data
columnsSeparated := strings . Join ( columns , "," )
insertSQL := fmt . Sprintf ( "INSERT INTO `new_%s_new` (%s) SELECT %s FROM %s" , tableName , columnsSeparated , columnsSeparated , tableName )
if _ , err := sess . Exec ( insertSQL ) ; err != nil {
return err
}
// Now drop the old table
if _ , err := sess . Exec ( fmt . Sprintf ( "DROP TABLE `%s`" , tableName ) ) ; err != nil {
return err
}
// Rename the table
if _ , err := sess . Exec ( fmt . Sprintf ( "ALTER TABLE `new_%s_new` RENAME TO `%s`" , tableName , tableName ) ) ; err != nil {
return err
}
2019-08-24 14:54:45 +05:30
case setting . Database . UsePostgreSQL :
2019-07-02 00:56:59 +05:30
cols := ""
for _ , col := range columnNames {
if cols != "" {
cols += ", "
}
cols += "DROP COLUMN `" + col + "` CASCADE"
}
if _ , err := sess . Exec ( fmt . Sprintf ( "ALTER TABLE `%s` %s" , tableName , cols ) ) ; err != nil {
return fmt . Errorf ( "Drop table `%s` columns %v: %v" , tableName , columnNames , err )
}
2019-08-24 14:54:45 +05:30
case setting . Database . UseMySQL :
2019-07-02 00:56:59 +05:30
// Drop indexes on columns first
sql := fmt . Sprintf ( "SHOW INDEX FROM %s WHERE column_name IN ('%s')" , tableName , strings . Join ( columnNames , "','" ) )
res , err := sess . Query ( sql )
if err != nil {
return err
}
for _ , index := range res {
indexName := index [ "column_name" ]
2019-10-12 10:25:07 +05:30
if len ( indexName ) > 0 {
_ , err := sess . Exec ( fmt . Sprintf ( "DROP INDEX `%s` ON `%s`" , indexName , tableName ) )
if err != nil {
return err
}
2019-07-02 00:56:59 +05:30
}
}
// Now drop the columns
2018-03-07 12:14:12 +05:30
cols := ""
for _ , col := range columnNames {
if cols != "" {
cols += ", "
}
cols += "DROP COLUMN `" + col + "`"
}
2018-05-09 21:59:04 +05:30
if _ , err := sess . Exec ( fmt . Sprintf ( "ALTER TABLE `%s` %s" , tableName , cols ) ) ; err != nil {
2018-03-07 12:14:12 +05:30
return fmt . Errorf ( "Drop table `%s` columns %v: %v" , tableName , columnNames , err )
}
2019-08-24 14:54:45 +05:30
case setting . Database . UseMSSQL :
2018-03-07 12:14:12 +05:30
cols := ""
for _ , col := range columnNames {
if cols != "" {
cols += ", "
}
cols += "`" + strings . ToLower ( col ) + "`"
}
sql := fmt . Sprintf ( "SELECT Name FROM SYS.DEFAULT_CONSTRAINTS WHERE PARENT_OBJECT_ID = OBJECT_ID('%[1]s') AND PARENT_COLUMN_ID IN (SELECT column_id FROM sys.columns WHERE lower(NAME) IN (%[2]s) AND object_id = OBJECT_ID('%[1]s'))" ,
tableName , strings . Replace ( cols , "`" , "'" , - 1 ) )
constraints := make ( [ ] string , 0 )
if err := sess . SQL ( sql ) . Find ( & constraints ) ; err != nil {
sess . Rollback ( )
return fmt . Errorf ( "Find constraints: %v" , err )
}
for _ , constraint := range constraints {
if _ , err := sess . Exec ( fmt . Sprintf ( "ALTER TABLE `%s` DROP CONSTRAINT `%s`" , tableName , constraint ) ) ; err != nil {
sess . Rollback ( )
return fmt . Errorf ( "Drop table `%s` constraint `%s`: %v" , tableName , constraint , err )
}
}
if _ , err := sess . Exec ( fmt . Sprintf ( "ALTER TABLE `%s` DROP COLUMN %s" , tableName , cols ) ) ; err != nil {
sess . Rollback ( )
return fmt . Errorf ( "Drop table `%s` columns %v: %v" , tableName , columnNames , err )
}
return sess . Commit ( )
default :
2019-04-02 13:18:31 +05:30
log . Fatal ( "Unrecognized DB" )
2018-03-07 12:14:12 +05:30
}
return nil
}
2015-03-26 05:21:22 +05:30
func fixLocaleFileLoadPanic ( _ * xorm . Engine ) error {
cfg , err := ini . Load ( setting . CustomConf )
if err != nil {
return fmt . Errorf ( "load custom config: %v" , err )
}
cfg . DeleteSection ( "i18n" )
if err = cfg . SaveTo ( setting . CustomConf ) ; err != nil {
return fmt . Errorf ( "save custom config: %v" , err )
}
setting . Langs = strings . Split ( strings . Replace ( strings . Join ( setting . Langs , "," ) , "fr-CA" , "fr-FR" , 1 ) , "," )
return nil
}
2015-07-26 19:36:28 +05:30
2016-11-28 21:14:17 +05:30
func trimCommitActionAppURLPrefix ( x * xorm . Engine ) error {
2015-07-26 19:36:28 +05:30
type PushCommit struct {
Sha1 string
Message string
AuthorEmail string
AuthorName string
}
type PushCommits struct {
Len int
Commits [ ] * PushCommit
2016-11-28 21:14:17 +05:30
CompareURL string ` json:"CompareUrl" `
2015-07-26 19:36:28 +05:30
}
type Action struct {
ID int64 ` xorm:"pk autoincr" `
Content string ` xorm:"TEXT" `
}
results , err := x . Query ( "SELECT `id`,`content` FROM `action` WHERE `op_type`=?" , 5 )
if err != nil {
return fmt . Errorf ( "select commit actions: %v" , err )
}
sess := x . NewSession ( )
2017-06-21 06:27:05 +05:30
defer sess . Close ( )
2015-07-26 19:36:28 +05:30
if err = sess . Begin ( ) ; err != nil {
return err
}
var pushCommits * PushCommits
for _ , action := range results {
actID := com . StrTo ( string ( action [ "id" ] ) ) . MustInt64 ( )
if actID == 0 {
continue
}
pushCommits = new ( PushCommits )
if err = json . Unmarshal ( action [ "content" ] , pushCommits ) ; err != nil {
2015-11-09 01:01:49 +05:30
return fmt . Errorf ( "unmarshal action content[%d]: %v" , actID , err )
2015-07-26 19:36:28 +05:30
}
2016-11-28 21:14:17 +05:30
infos := strings . Split ( pushCommits . CompareURL , "/" )
2015-07-26 19:36:28 +05:30
if len ( infos ) <= 4 {
continue
}
2016-11-28 21:14:17 +05:30
pushCommits . CompareURL = strings . Join ( infos [ len ( infos ) - 4 : ] , "/" )
2015-07-26 19:36:28 +05:30
p , err := json . Marshal ( pushCommits )
if err != nil {
2015-11-09 01:01:49 +05:30
return fmt . Errorf ( "marshal action content[%d]: %v" , actID , err )
2015-07-26 19:36:28 +05:30
}
2019-06-13 01:11:28 +05:30
if _ , err = sess . ID ( actID ) . Update ( & Action {
2015-07-26 19:36:28 +05:30
Content : string ( p ) ,
} ) ; err != nil {
2015-08-19 17:38:57 +05:30
return fmt . Errorf ( "update action[%d]: %v" , actID , err )
2015-07-26 19:36:28 +05:30
}
}
return sess . Commit ( )
}
2015-08-10 12:12:50 +05:30
func issueToIssueLabel ( x * xorm . Engine ) error {
type IssueLabel struct {
ID int64 ` xorm:"pk autoincr" `
IssueID int64 ` xorm:"UNIQUE(s)" `
LabelID int64 ` xorm:"UNIQUE(s)" `
}
issueLabels := make ( [ ] * IssueLabel , 0 , 50 )
results , err := x . Query ( "SELECT `id`,`label_ids` FROM `issue`" )
if err != nil {
2015-12-11 06:22:06 +05:30
if strings . Contains ( err . Error ( ) , "no such column" ) ||
strings . Contains ( err . Error ( ) , "Unknown column" ) {
2015-08-10 12:12:50 +05:30
return nil
}
return fmt . Errorf ( "select issues: %v" , err )
}
for _ , issue := range results {
issueID := com . StrTo ( issue [ "id" ] ) . MustInt64 ( )
// Just in case legacy code can have duplicated IDs for same label.
mark := make ( map [ int64 ] bool )
for _ , idStr := range strings . Split ( string ( issue [ "label_ids" ] ) , "|" ) {
labelID := com . StrTo ( strings . TrimPrefix ( idStr , "$" ) ) . MustInt64 ( )
if labelID == 0 || mark [ labelID ] {
continue
}
mark [ labelID ] = true
issueLabels = append ( issueLabels , & IssueLabel {
IssueID : issueID ,
LabelID : labelID ,
} )
}
}
sess := x . NewSession ( )
2017-06-21 06:27:05 +05:30
defer sess . Close ( )
2015-08-10 12:12:50 +05:30
if err = sess . Begin ( ) ; err != nil {
return err
}
if err = sess . Sync2 ( new ( IssueLabel ) ) ; err != nil {
2016-08-27 06:02:41 +05:30
return fmt . Errorf ( "Sync2: %v" , err )
2015-08-10 12:12:50 +05:30
} else if _ , err = sess . Insert ( issueLabels ) ; err != nil {
return fmt . Errorf ( "insert issue-labels: %v" , err )
}
return sess . Commit ( )
}
2015-08-11 20:54:40 +05:30
func attachmentRefactor ( x * xorm . Engine ) error {
type Attachment struct {
ID int64 ` xorm:"pk autoincr" `
UUID string ` xorm:"uuid INDEX" `
// For rename purpose.
Path string ` xorm:"-" `
NewPath string ` xorm:"-" `
}
results , err := x . Query ( "SELECT * FROM `attachment`" )
if err != nil {
return fmt . Errorf ( "select attachments: %v" , err )
}
attachments := make ( [ ] * Attachment , 0 , len ( results ) )
for _ , attach := range results {
if ! com . IsExist ( string ( attach [ "path" ] ) ) {
// If the attachment is already missing, there is no point to update it.
continue
}
attachments = append ( attachments , & Attachment {
ID : com . StrTo ( attach [ "id" ] ) . MustInt64 ( ) ,
UUID : gouuid . NewV4 ( ) . String ( ) ,
Path : string ( attach [ "path" ] ) ,
} )
}
sess := x . NewSession ( )
2017-06-21 06:27:05 +05:30
defer sess . Close ( )
2015-08-11 20:54:40 +05:30
if err = sess . Begin ( ) ; err != nil {
return err
}
if err = sess . Sync2 ( new ( Attachment ) ) ; err != nil {
return fmt . Errorf ( "Sync2: %v" , err )
}
// Note: Roll back for rename can be a dead loop,
// so produces a backup file.
var buf bytes . Buffer
buf . WriteString ( "# old path -> new path\n" )
// Update database first because this is where error happens the most often.
for _ , attach := range attachments {
2019-06-13 01:11:28 +05:30
if _ , err = sess . ID ( attach . ID ) . Update ( attach ) ; err != nil {
2015-08-11 20:54:40 +05:30
return err
}
attach . NewPath = path . Join ( setting . AttachmentPath , attach . UUID [ 0 : 1 ] , attach . UUID [ 1 : 2 ] , attach . UUID )
buf . WriteString ( attach . Path )
buf . WriteString ( "\t" )
buf . WriteString ( attach . NewPath )
buf . WriteString ( "\n" )
}
// Then rename attachments.
isSucceed := true
defer func ( ) {
if isSucceed {
return
}
dumpPath := path . Join ( setting . LogRootPath , "attachment_path.dump" )
ioutil . WriteFile ( dumpPath , buf . Bytes ( ) , 0666 )
2017-01-30 01:43:57 +05:30
log . Info ( "Failed to rename some attachments, old and new paths are saved into: %s" , dumpPath )
2015-08-11 20:54:40 +05:30
} ( )
for _ , attach := range attachments {
if err = os . MkdirAll ( path . Dir ( attach . NewPath ) , os . ModePerm ) ; err != nil {
isSucceed = false
return err
}
if err = os . Rename ( attach . Path , attach . NewPath ) ; err != nil {
isSucceed = false
return err
}
}
return sess . Commit ( )
}
2015-10-19 05:00:39 +05:30
func renamePullRequestFields ( x * xorm . Engine ) ( err error ) {
type PullRequest struct {
ID int64 ` xorm:"pk autoincr" `
PullID int64 ` xorm:"INDEX" `
PullIndex int64
HeadBarcnh string
IssueID int64 ` xorm:"INDEX" `
Index int64
HeadBranch string
}
if err = x . Sync ( new ( PullRequest ) ) ; err != nil {
return fmt . Errorf ( "sync: %v" , err )
}
results , err := x . Query ( "SELECT `id`,`pull_id`,`pull_index`,`head_barcnh` FROM `pull_request`" )
if err != nil {
if strings . Contains ( err . Error ( ) , "no such column" ) {
return nil
}
return fmt . Errorf ( "select pull requests: %v" , err )
}
sess := x . NewSession ( )
2017-06-21 06:27:05 +05:30
defer sess . Close ( )
2015-10-19 05:00:39 +05:30
if err = sess . Begin ( ) ; err != nil {
return err
}
var pull * PullRequest
for _ , pr := range results {
pull = & PullRequest {
ID : com . StrTo ( pr [ "id" ] ) . MustInt64 ( ) ,
IssueID : com . StrTo ( pr [ "pull_id" ] ) . MustInt64 ( ) ,
Index : com . StrTo ( pr [ "pull_index" ] ) . MustInt64 ( ) ,
HeadBranch : string ( pr [ "head_barcnh" ] ) ,
}
2015-12-11 06:22:06 +05:30
if pull . Index == 0 {
continue
}
2019-06-13 01:11:28 +05:30
if _ , err = sess . ID ( pull . ID ) . Update ( pull ) ; err != nil {
2015-10-19 05:00:39 +05:30
return err
}
}
return sess . Commit ( )
}
2015-11-01 08:48:58 +05:30
func cleanUpMigrateRepoInfo ( x * xorm . Engine ) ( err error ) {
type (
User struct {
ID int64 ` xorm:"pk autoincr" `
LowerName string
}
Repository struct {
ID int64 ` xorm:"pk autoincr" `
OwnerID int64
LowerName string
}
)
repos := make ( [ ] * Repository , 0 , 25 )
if err = x . Where ( "is_mirror=?" , false ) . Find ( & repos ) ; err != nil {
return fmt . Errorf ( "select all non-mirror repositories: %v" , err )
}
var user * User
for _ , repo := range repos {
user = & User { ID : repo . OwnerID }
has , err := x . Get ( user )
if err != nil {
return fmt . Errorf ( "get owner of repository[%d - %d]: %v" , repo . ID , repo . OwnerID , err )
} else if ! has {
continue
}
configPath := filepath . Join ( setting . RepoRootPath , user . LowerName , repo . LowerName + ".git/config" )
2015-11-01 08:55:08 +05:30
// In case repository file is somehow missing.
if ! com . IsFile ( configPath ) {
continue
}
2015-11-01 08:48:58 +05:30
cfg , err := ini . Load ( configPath )
if err != nil {
return fmt . Errorf ( "open config file: %v" , err )
}
cfg . DeleteSection ( "remote \"origin\"" )
if err = cfg . SaveToIndent ( configPath , "\t" ) ; err != nil {
return fmt . Errorf ( "save config file: %v" , err )
}
}
return nil
}
2015-12-15 03:36:54 +05:30
func generateOrgRandsAndSalt ( x * xorm . Engine ) ( err error ) {
type User struct {
ID int64 ` xorm:"pk autoincr" `
Rands string ` xorm:"VARCHAR(10)" `
Salt string ` xorm:"VARCHAR(10)" `
}
orgs := make ( [ ] * User , 0 , 10 )
if err = x . Where ( "type=1" ) . And ( "rands=''" ) . Find ( & orgs ) ; err != nil {
return fmt . Errorf ( "select all organizations: %v" , err )
}
sess := x . NewSession ( )
2017-06-21 06:27:05 +05:30
defer sess . Close ( )
2015-12-15 03:36:54 +05:30
if err = sess . Begin ( ) ; err != nil {
return err
}
for _ , org := range orgs {
2018-02-18 23:44:37 +05:30
if org . Rands , err = generate . GetRandomString ( 10 ) ; err != nil {
2016-12-20 18:02:02 +05:30
return err
}
2018-02-18 23:44:37 +05:30
if org . Salt , err = generate . GetRandomString ( 10 ) ; err != nil {
2016-12-20 18:02:02 +05:30
return err
}
2019-06-13 01:11:28 +05:30
if _ , err = sess . ID ( org . ID ) . Update ( org ) ; err != nil {
2015-12-15 03:36:54 +05:30
return err
}
}
return sess . Commit ( )
}
2016-03-10 06:23:30 +05:30
2016-11-28 21:14:17 +05:30
// TAction defines the struct for migrating table action
2016-03-10 06:23:30 +05:30
type TAction struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TAction ) TableName ( ) string { return "action" }
2016-11-28 21:14:17 +05:30
// TNotice defines the struct for migrating table notice
2016-03-10 06:23:30 +05:30
type TNotice struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TNotice ) TableName ( ) string { return "notice" }
2016-11-28 21:14:17 +05:30
// TComment defines the struct for migrating table comment
2016-03-10 06:23:30 +05:30
type TComment struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TComment ) TableName ( ) string { return "comment" }
2016-11-28 21:14:17 +05:30
// TIssue defines the struct for migrating table issue
2016-03-10 06:23:30 +05:30
type TIssue struct {
ID int64 ` xorm:"pk autoincr" `
DeadlineUnix int64
CreatedUnix int64
UpdatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TIssue ) TableName ( ) string { return "issue" }
2016-11-28 21:14:17 +05:30
// TMilestone defines the struct for migrating table milestone
2016-03-10 06:23:30 +05:30
type TMilestone struct {
ID int64 ` xorm:"pk autoincr" `
DeadlineUnix int64
ClosedDateUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TMilestone ) TableName ( ) string { return "milestone" }
2016-11-28 21:14:17 +05:30
// TAttachment defines the struct for migrating table attachment
2016-03-10 06:23:30 +05:30
type TAttachment struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TAttachment ) TableName ( ) string { return "attachment" }
2016-11-28 21:14:17 +05:30
// TLoginSource defines the struct for migrating table login_source
2016-03-10 06:23:30 +05:30
type TLoginSource struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
UpdatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TLoginSource ) TableName ( ) string { return "login_source" }
2016-11-28 21:14:17 +05:30
// TPull defines the struct for migrating table pull_request
2016-03-10 06:23:30 +05:30
type TPull struct {
ID int64 ` xorm:"pk autoincr" `
MergedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TPull ) TableName ( ) string { return "pull_request" }
2016-11-28 21:14:17 +05:30
// TRelease defines the struct for migrating table release
2016-03-10 06:23:30 +05:30
type TRelease struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TRelease ) TableName ( ) string { return "release" }
2016-11-28 21:14:17 +05:30
// TRepo defines the struct for migrating table repository
2016-03-10 06:23:30 +05:30
type TRepo struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
UpdatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TRepo ) TableName ( ) string { return "repository" }
2016-11-28 21:14:17 +05:30
// TMirror defines the struct for migrating table mirror
2016-03-10 06:23:30 +05:30
type TMirror struct {
ID int64 ` xorm:"pk autoincr" `
UpdatedUnix int64
NextUpdateUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TMirror ) TableName ( ) string { return "mirror" }
2016-11-28 21:14:17 +05:30
// TPublicKey defines the struct for migrating table public_key
2016-03-10 06:23:30 +05:30
type TPublicKey struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
UpdatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TPublicKey ) TableName ( ) string { return "public_key" }
2016-11-28 21:14:17 +05:30
// TDeployKey defines the struct for migrating table deploy_key
2016-03-10 06:23:30 +05:30
type TDeployKey struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
UpdatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TDeployKey ) TableName ( ) string { return "deploy_key" }
2016-11-28 21:14:17 +05:30
// TAccessToken defines the struct for migrating table access_token
2016-03-10 06:23:30 +05:30
type TAccessToken struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
UpdatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TAccessToken ) TableName ( ) string { return "access_token" }
2016-11-28 21:14:17 +05:30
// TUser defines the struct for migrating table user
2016-03-10 06:23:30 +05:30
type TUser struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
UpdatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TUser ) TableName ( ) string { return "user" }
2016-11-28 21:14:17 +05:30
// TWebhook defines the struct for migrating table webhook
2016-03-10 06:23:30 +05:30
type TWebhook struct {
ID int64 ` xorm:"pk autoincr" `
CreatedUnix int64
UpdatedUnix int64
}
2016-11-28 21:14:17 +05:30
// TableName will be invoked by XORM to customrize the table name
2016-03-10 06:23:30 +05:30
func ( t * TWebhook ) TableName ( ) string { return "webhook" }
func convertDateToUnix ( x * xorm . Engine ) ( err error ) {
2016-07-08 04:55:09 +05:30
log . Info ( "This migration could take up to minutes, please be patient." )
2016-03-10 06:23:30 +05:30
type Bean struct {
ID int64 ` xorm:"pk autoincr" `
Created time . Time
Updated time . Time
Merged time . Time
Deadline time . Time
ClosedDate time . Time
NextUpdate time . Time
}
var tables = [ ] struct {
name string
cols [ ] string
bean interface { }
} {
{ "action" , [ ] string { "created" } , new ( TAction ) } ,
{ "notice" , [ ] string { "created" } , new ( TNotice ) } ,
{ "comment" , [ ] string { "created" } , new ( TComment ) } ,
{ "issue" , [ ] string { "deadline" , "created" , "updated" } , new ( TIssue ) } ,
{ "milestone" , [ ] string { "deadline" , "closed_date" } , new ( TMilestone ) } ,
{ "attachment" , [ ] string { "created" } , new ( TAttachment ) } ,
{ "login_source" , [ ] string { "created" , "updated" } , new ( TLoginSource ) } ,
{ "pull_request" , [ ] string { "merged" } , new ( TPull ) } ,
{ "release" , [ ] string { "created" } , new ( TRelease ) } ,
{ "repository" , [ ] string { "created" , "updated" } , new ( TRepo ) } ,
{ "mirror" , [ ] string { "updated" , "next_update" } , new ( TMirror ) } ,
{ "public_key" , [ ] string { "created" , "updated" } , new ( TPublicKey ) } ,
{ "deploy_key" , [ ] string { "created" , "updated" } , new ( TDeployKey ) } ,
{ "access_token" , [ ] string { "created" , "updated" } , new ( TAccessToken ) } ,
{ "user" , [ ] string { "created" , "updated" } , new ( TUser ) } ,
{ "webhook" , [ ] string { "created" , "updated" } , new ( TWebhook ) } ,
}
for _ , table := range tables {
log . Info ( "Converting table: %s" , table . name )
if err = x . Sync2 ( table . bean ) ; err != nil {
return fmt . Errorf ( "Sync [table: %s]: %v" , table . name , err )
}
offset := 0
for {
beans := make ( [ ] * Bean , 0 , 100 )
2017-08-04 10:12:51 +05:30
if err = x . Table ( table . name ) . Asc ( "id" ) . Limit ( 100 , offset ) . Find ( & beans ) ; err != nil {
2016-03-10 06:23:30 +05:30
return fmt . Errorf ( "select beans [table: %s, offset: %d]: %v" , table . name , offset , err )
}
log . Trace ( "Table [%s]: offset: %d, beans: %d" , table . name , offset , len ( beans ) )
if len ( beans ) == 0 {
break
}
offset += 100
baseSQL := "UPDATE `" + table . name + "` SET "
for _ , bean := range beans {
valSQLs := make ( [ ] string , 0 , len ( table . cols ) )
for _ , col := range table . cols {
fieldSQL := ""
fieldSQL += col + "_unix = "
switch col {
case "deadline" :
if bean . Deadline . IsZero ( ) {
continue
}
2016-07-23 17:54:44 +05:30
fieldSQL += com . ToStr ( bean . Deadline . Unix ( ) )
2016-03-10 06:23:30 +05:30
case "created" :
2016-07-23 17:54:44 +05:30
fieldSQL += com . ToStr ( bean . Created . Unix ( ) )
2016-03-10 06:23:30 +05:30
case "updated" :
2016-07-23 17:54:44 +05:30
fieldSQL += com . ToStr ( bean . Updated . Unix ( ) )
2016-03-10 06:23:30 +05:30
case "closed_date" :
2016-07-23 17:54:44 +05:30
fieldSQL += com . ToStr ( bean . ClosedDate . Unix ( ) )
2016-03-10 06:23:30 +05:30
case "merged" :
2016-07-23 17:54:44 +05:30
fieldSQL += com . ToStr ( bean . Merged . Unix ( ) )
2016-03-10 06:23:30 +05:30
case "next_update" :
2016-07-23 17:54:44 +05:30
fieldSQL += com . ToStr ( bean . NextUpdate . Unix ( ) )
2016-03-10 06:23:30 +05:30
}
valSQLs = append ( valSQLs , fieldSQL )
}
if len ( valSQLs ) == 0 {
continue
}
if _ , err = x . Exec ( baseSQL + strings . Join ( valSQLs , "," ) + " WHERE id = " + com . ToStr ( bean . ID ) ) ; err != nil {
return fmt . Errorf ( "update bean [table: %s, id: %d]: %v" , table . name , bean . ID , err )
}
}
}
}
return nil
}