This is an automated email from the ASF dual-hosted git repository.

klesh pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/incubator-devlake.git


The following commit(s) were added to refs/heads/main by this push:
     new 13b0b43da adding pull_request_id index to  
pull_request_commits/comments tables (#7559)
13b0b43da is described below

commit 13b0b43dafddc9dd8d043401ad0f27a404e50565
Author: sstojak1 <[email protected]>
AuthorDate: Fri Jun 14 11:26:40 2024 +0200

    adding pull_request_id index to  pull_request_commits/comments tables 
(#7559)
    
    * adding pull_request_id index to  pull_request_commits/comments tables
    
    * change the pull_request_commits primary key columns order
    
    * adding Apache license header
    
    * only run for mysql
    
    * adding support for postgres
    
    ---------
    
    Co-authored-by: Josip Stojak <[email protected]>
---
 ..._pull_request_id_index_for_pr_comments_table.go | 46 ++++++++++++++
 ...d_pull_request_id_index_for_pr_commits_table.go | 72 ++++++++++++++++++++++
 backend/core/models/migrationscripts/register.go   |  2 +
 3 files changed, 120 insertions(+)

diff --git 
a/backend/core/models/migrationscripts/20240603_add_pull_request_id_index_for_pr_comments_table.go
 
b/backend/core/models/migrationscripts/20240603_add_pull_request_id_index_for_pr_comments_table.go
new file mode 100644
index 000000000..9e6d7b6ed
--- /dev/null
+++ 
b/backend/core/models/migrationscripts/20240603_add_pull_request_id_index_for_pr_comments_table.go
@@ -0,0 +1,46 @@
+/*
+Licensed to the Apache Software Foundation (ASF) under one or more
+contributor license agreements.  See the NOTICE file distributed with
+this work for additional information regarding copyright ownership.
+The ASF licenses this file to You under the Apache License, Version 2.0
+(the "License"); you may not use this file except in compliance with
+the License.  You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package migrationscripts
+
+import (
+       "github.com/apache/incubator-devlake/core/context"
+       "github.com/apache/incubator-devlake/core/errors"
+       "github.com/apache/incubator-devlake/helpers/migrationhelper"
+)
+
+type addPullRequestIdIndexToPullRequestComments struct{}
+
+type pullRequestComments20240602 struct {
+       PullRequestId string `gorm:"index"`
+}
+
+func (pullRequestComments20240602) TableName() string {
+       return "pull_request_comments"
+}
+
+func (u *addPullRequestIdIndexToPullRequestComments) Up(basicRes 
context.BasicRes) errors.Error {
+       return migrationhelper.AutoMigrateTables(basicRes, 
&pullRequestComments20240602{})
+}
+
+func (*addPullRequestIdIndexToPullRequestComments) Version() uint64 {
+       return 20240602103401
+}
+
+func (*addPullRequestIdIndexToPullRequestComments) Name() string {
+       return "add pull_request_id index for pull_request_comments"
+}
diff --git 
a/backend/core/models/migrationscripts/20240603_add_pull_request_id_index_for_pr_commits_table.go
 
b/backend/core/models/migrationscripts/20240603_add_pull_request_id_index_for_pr_commits_table.go
new file mode 100644
index 000000000..ddf381edc
--- /dev/null
+++ 
b/backend/core/models/migrationscripts/20240603_add_pull_request_id_index_for_pr_commits_table.go
@@ -0,0 +1,72 @@
+/*
+Licensed to the Apache Software Foundation (ASF) under one or more
+contributor license agreements.  See the NOTICE file distributed with
+this work for additional information regarding copyright ownership.
+The ASF licenses this file to You under the Apache License, Version 2.0
+(the "License"); you may not use this file except in compliance with
+the License.  You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package migrationscripts
+
+import (
+       "github.com/apache/incubator-devlake/core/context"
+       "github.com/apache/incubator-devlake/core/errors"
+       "net/url"
+       "strings"
+)
+
+type addPullRequestIdIndexToPullRequestCommits struct{}
+
+func (*addPullRequestIdIndexToPullRequestCommits) Up(basicRes 
context.BasicRes) errors.Error {
+       dbUrl := basicRes.GetConfig("DB_URL")
+       if dbUrl == "" {
+               return errors.BadInput.New("DB_URL is required")
+       }
+       u, err1 := url.Parse(dbUrl)
+       if err1 != nil {
+               return errors.Convert(err1)
+       }
+       switch strings.ToLower(u.Scheme) {
+       case "mysql":
+               db := basicRes.GetDal()
+               err := db.Exec("ALTER TABLE pull_request_commits DROP PRIMARY 
KEY;")
+               if err != nil {
+                       return err
+               }
+               err = db.Exec("ALTER TABLE pull_request_commits ADD PRIMARY KEY 
(pull_request_id, commit_sha);")
+               if err != nil {
+                       return err
+               }
+               return nil
+       case "postgresql", "postgres", "pg":
+               db := basicRes.GetDal()
+               err := db.Exec("ALTER TABLE pull_request_commits DROP 
CONSTRAINT pull_request_commits_pkey;")
+               if err != nil {
+                       return err
+               }
+               err = db.Exec("ALTER TABLE pull_request_commits ADD PRIMARY KEY 
(pull_request_id, commit_sha);")
+               if err != nil {
+                       return err
+               }
+               return nil
+       default:
+               return nil
+       }
+}
+
+func (*addPullRequestIdIndexToPullRequestCommits) Version() uint64 {
+       return 20240602103400
+}
+
+func (*addPullRequestIdIndexToPullRequestCommits) Name() string {
+       return "changing pull_request_commits primary key columns order"
+}
diff --git a/backend/core/models/migrationscripts/register.go 
b/backend/core/models/migrationscripts/register.go
index 4f3a21824..057a2a327 100644
--- a/backend/core/models/migrationscripts/register.go
+++ b/backend/core/models/migrationscripts/register.go
@@ -121,5 +121,7 @@ func All() []plugin.MigrationScript {
                new(modifyCicdPipelineCommitsRepoUrlLength),
                new(addPrAssigneeAndReviewer),
                new(modifyPrAssigneeAndReviewerId),
+               new(addPullRequestIdIndexToPullRequestCommits),
+               new(addPullRequestIdIndexToPullRequestComments),
        }
 }

Reply via email to