Skip to content

Commit

Permalink
feat: collect Zentao issue-repo-commit from DB (apache#8185)
Browse files Browse the repository at this point in the history
* feat: collect Zentao Story-Repo-Commit from DB

* feat: collect Zentao bug and task related repo commits from DB
  • Loading branch information
usharerose authored Nov 14, 2024
1 parent 7d3a792 commit 89cdcba
Show file tree
Hide file tree
Showing 4 changed files with 569 additions and 6 deletions.
9 changes: 3 additions & 6 deletions backend/plugins/zentao/impl/impl.go
Original file line number Diff line number Diff line change
Expand Up @@ -137,8 +137,7 @@ func (p Zentao) SubTaskMetas() []plugin.SubTaskMeta {

tasks.CollectTaskCommitsMeta,
tasks.ExtractTaskCommitsMeta,
tasks.CollectTaskRepoCommitsMeta,
tasks.ExtractTaskRepoCommitsMeta,
tasks.DBGetTaskRepoCommitsMeta,
tasks.ConvertTaskRepoCommitsMeta,

// product
Expand All @@ -153,14 +152,12 @@ func (p Zentao) SubTaskMetas() []plugin.SubTaskMeta {

tasks.CollectStoryCommitsMeta,
tasks.ExtractStoryCommitsMeta,
tasks.CollectStoryRepoCommitsMeta,
tasks.ExtractStoryRepoCommitsMeta,
tasks.DBGetStoryRepoCommitsMeta,
tasks.ConvertStoryRepoCommitsMeta,

tasks.CollectBugCommitsMeta,
tasks.ExtractBugCommitsMeta,
tasks.CollectBugRepoCommitsMeta,
tasks.ExtractBugRepoCommitsMeta,
tasks.DBGetBugRepoCommitsMeta,
tasks.ConvertBugRepoCommitsMeta,

tasks.DBGetChangelogMeta,
Expand Down
187 changes: 187 additions & 0 deletions backend/plugins/zentao/tasks/bug_repo_commits_dbget.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,187 @@
/*
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 tasks

import (
"encoding/json"
"reflect"
"strconv"

"github.com/apache/incubator-devlake/core/dal"
"github.com/apache/incubator-devlake/core/errors"
"github.com/apache/incubator-devlake/core/plugin"
"github.com/apache/incubator-devlake/helpers/pluginhelper/api"
"github.com/apache/incubator-devlake/plugins/zentao/models"
)

var _ plugin.SubTaskEntryPoint = DBGetBugRepoCommits

var DBGetBugRepoCommitsMeta = plugin.SubTaskMeta{
Name: "collectBugRepoCommits",
EntryPoint: DBGetBugRepoCommits,
EnabledByDefault: true,
Description: "Get bug commits data from Zentao database",
DomainTypes: []string{plugin.DOMAIN_TYPE_TICKET},
}

func DBGetBugRepoCommits(taskCtx plugin.SubTaskContext) errors.Error {
data := taskCtx.GetData().(*ZentaoTaskData)

// skip if no RemoteDb
if data.RemoteDb == nil {
return nil
}

divider := api.NewBatchSaveDivider(taskCtx, 500, "", "")
defer func() {
err1 := divider.Close()
if err1 != nil {
panic(err1)
}
}()
handler, err := newBugRepoCommitHandler(taskCtx, divider)
if err != nil {
return err
}
return handler.collectBugRepoCommit(
taskCtx.GetDal(),
data.RemoteDb,
data.Options.ProjectId,
data.Options.ConnectionId,
)
}

type bugRepoCommitHandler struct {
rawDataParams string
bugRepoCommitBachSave *api.BatchSave
}

type toolZentaoBug struct {
BugID int64 `gorm:"type:integer"`
}

type RemoteBugRepoCommit struct {
Project int64 `gorm:"type:integer"`
IssueID int64 `gorm:"type:integer"`
RepoUrl string
CommitSha string
}

func (h bugRepoCommitHandler) collectBugRepoCommit(db dal.Dal, rdb dal.Dal, projectId int64, connectionId uint64) errors.Error {
bugCursor, err := db.RawCursor(`
SELECT
DISTINCT id AS bug_id
FROM
_tool_zentao_bugs AS fact_bug
WHERE
fact_bug.project = ? AND
fact_bug.connection_id = ?
`, projectId, connectionId)
if err != nil {
return err
}
defer bugCursor.Close()
var bugIds []int64
for bugCursor.Next() {
var row toolZentaoBug
err := db.Fetch(bugCursor, &row)
if err != nil {
return errors.Default.Wrap(err, "error fetching bugCursor")
}
bugIds = append(bugIds, row.BugID)
}

remoteCursor, err := rdb.RawCursor(`
SELECT
DISTINCT dim_bug_commits.bug_id AS issue_id,
dim_repo.path AS repo_url,
dim_revision.revision AS commit_sha
FROM (
SELECT
fact_action.objectID AS bug_id,
fact_action.project AS project_id,
fact_action.product AS product_id,
fact_action.extra AS short_commit_hexsha
FROM
zt_action AS fact_action
WHERE
fact_action.objectType IN ('bug') AND
fact_action.objectID IN ? AND
fact_action.action IN ('gitcommited')
) AS dim_bug_commits
INNER JOIN (
SELECT
fact_repo_hist.repo AS repo_id,
fact_repo_hist.revision AS revision,
LEFT(fact_repo_hist.revision, 10) AS short_commit_hexsha
FROM
zt_repohistory AS fact_repo_hist
) AS dim_revision
ON
dim_bug_commits.short_commit_hexsha = dim_revision.short_commit_hexsha
INNER JOIN
zt_repo AS dim_repo
ON
dim_revision.repo_id = dim_repo.id
`, bugIds)
if err != nil {
return err
}
defer remoteCursor.Close()

for remoteCursor.Next() {
var remoteBugRepoCommit RemoteBugRepoCommit
err = rdb.Fetch(remoteCursor, &remoteBugRepoCommit)
if err != nil {
return err
}
bugRepoCommit := &models.ZentaoBugRepoCommit{
ConnectionId: connectionId,
Project: projectId,
RepoUrl: remoteBugRepoCommit.RepoUrl,
CommitSha: remoteBugRepoCommit.CommitSha,
IssueId: strconv.FormatInt(remoteBugRepoCommit.IssueID, 10),
}
bugRepoCommit.NoPKModel.RawDataParams = h.rawDataParams
err = h.bugRepoCommitBachSave.Add(bugRepoCommit)
if err != nil {
return err
}
}
return h.bugRepoCommitBachSave.Flush()
}

func newBugRepoCommitHandler(taskCtx plugin.SubTaskContext, divider *api.BatchSaveDivider) (*bugRepoCommitHandler, errors.Error) {
data := taskCtx.GetData().(*ZentaoTaskData)

bugRepoCommitBachSave, err := divider.ForType(reflect.TypeOf(&models.ZentaoBugRepoCommit{}))
if err != nil {
return nil, err
}
blob, _ := json.Marshal(data.Options.GetParams())
rawDataParams := string(blob)
db := taskCtx.GetDal()
err = db.Delete(&models.ZentaoBugRepoCommit{}, dal.Where("_raw_data_params = ?", rawDataParams))
if err != nil {
return nil, err
}
return &bugRepoCommitHandler{
rawDataParams: rawDataParams,
bugRepoCommitBachSave: bugRepoCommitBachSave,
}, nil
}
Loading

0 comments on commit 89cdcba

Please sign in to comment.