Delete project member when delete project

Signed-off-by: stonezdj <[email protected]>

src/controller/event/handler/internal/project.go

@@ -21,6 +21,7 @@ import (
 	"github.com/goharbor/harbor/src/controller/immutable"
 	"github.com/goharbor/harbor/src/controller/retention"
 	"github.com/goharbor/harbor/src/lib/log"
+	"github.com/goharbor/harbor/src/pkg/member"
 )
 
 // ProjectEventHandler process project event data
@@ -39,16 +40,15 @@ func (a *ProjectEventHandler) IsStateful() bool {
 
 func (a *ProjectEventHandler) onProjectDelete(ctx context.Context, event *event.DeleteProjectEvent) error {
 	log.Infof("delete project id: %d", event.ProjectID)
-	// delete tag immutable
-	err := immutable.Ctr.DeleteImmutableRuleByProject(ctx, event.ProjectID)
-	if err != nil {
+	if err := immutable.Ctr.DeleteImmutableRuleByProject(ctx, event.ProjectID); err != nil {
 		log.Errorf("failed to delete immutable rule, error %v", err)
 	}
-	// delete tag retention
-	err = retention.Ctl.DeleteRetentionByProject(ctx, event.ProjectID)
-	if err != nil {
+	if err := retention.Ctl.DeleteRetentionByProject(ctx, event.ProjectID); err != nil {
 		log.Errorf("failed to delete retention rule, error %v", err)
 	}
+	if err := member.Mgr.DeleteMemberByProjectID(ctx, event.ProjectID); err != nil {
+		log.Errorf("failed to delete project member, error %v", err)
+	}
 	return nil
 }
 

src/controller/event/handler/internal/project_test.go

@@ -21,19 +21,24 @@ import (
 	beegoorm "github.com/beego/beego/v2/client/orm"
 	"github.com/stretchr/testify/suite"
 
+	"github.com/goharbor/harbor/src/common"
 	common_dao "github.com/goharbor/harbor/src/common/dao"
+	commonmodels "github.com/goharbor/harbor/src/common/models"
 	"github.com/goharbor/harbor/src/controller/event"
 	"github.com/goharbor/harbor/src/controller/immutable"
 	"github.com/goharbor/harbor/src/lib/config"
 	"github.com/goharbor/harbor/src/lib/orm"
 	"github.com/goharbor/harbor/src/pkg"
 	"github.com/goharbor/harbor/src/pkg/artifact"
 	immutableModel "github.com/goharbor/harbor/src/pkg/immutable/model"
+	"github.com/goharbor/harbor/src/pkg/member"
+	memberModels "github.com/goharbor/harbor/src/pkg/member/models"
 	"github.com/goharbor/harbor/src/pkg/project"
 	"github.com/goharbor/harbor/src/pkg/project/models"
 	"github.com/goharbor/harbor/src/pkg/repository/model"
 	"github.com/goharbor/harbor/src/pkg/tag"
 	tagmodel "github.com/goharbor/harbor/src/pkg/tag/model/tag"
+	"github.com/goharbor/harbor/src/pkg/user"
 )
 
 // ProjectHandlerTestSuite is test suite for artifact handler.
@@ -81,6 +86,18 @@ func (suite *ProjectHandlerTestSuite) TestOnProjectDelete() {
 	projID, err := project.New().Create(suite.ctx, &models.Project{Name: "test-project", OwnerID: 1})
 	suite.Nil(err)
 
+	userID, err := user.Mgr.Create(suite.ctx, &commonmodels.User{Username: "test-user-event", Email: "[email protected]"})
+	defer user.Mgr.Delete(suite.ctx, userID)
+
+	// create project member
+	_, err = member.Mgr.AddProjectMember(suite.ctx, memberModels.Member{ProjectID: projID, EntityType: common.UserMember, EntityID: userID, Role: 1})
+	suite.Nil(err)
+
+	// verify project member
+	members, err := member.Mgr.SearchMemberByName(suite.ctx, projID, "test-user-event")
+	suite.Nil(err)
+	suite.Equal(1, len(members))
+
 	defer project.New().Delete(suite.ctx, projID)
 	immutableRule := &immutableModel.Metadata{
 		ProjectID: projID,
@@ -116,6 +133,11 @@ func (suite *ProjectHandlerTestSuite) TestOnProjectDelete() {
 	// check if immutable rule is deleted
 	_, err = immutable.Ctr.GetImmutableRule(suite.ctx, immutableID)
 	suite.NotNil(err)
+
+	// check if project member is deleted
+	mbs, err := member.Mgr.SearchMemberByName(suite.ctx, projID, "test-user-event")
+	suite.Nil(err)
+	suite.Equal(0, len(mbs))
 }
 
 // TestArtifactHandler tests ArtifactHandler.

src/pkg/member/dao/dao.go

@@ -44,6 +44,8 @@ type DAO interface {
 	DeleteProjectMemberByID(ctx context.Context, projectID int64, pmid int) error
 	// DeleteProjectMemberByUserID -- Delete project member by user id
 	DeleteProjectMemberByUserID(ctx context.Context, uid int) error
+	// DeleteProjectMemberByProjectID -- Delete project member by project id
+	DeleteProjectMemberByProjectID(ctx context.Context, projectID int64) error
 	// SearchMemberByName search members of the project by entity_name
 	SearchMemberByName(ctx context.Context, projectID int64, entityName string) ([]*models.Member, error)
 	// ListRoles lists the roles of user for the specific project
@@ -261,3 +263,15 @@ func (d *dao) ListRoles(ctx context.Context, user *models.User, projectID int64)
 	}
 	return roles, nil
 }
+
+func (d *dao) DeleteProjectMemberByProjectID(ctx context.Context, projectID int64) error {
+	o, err := orm.FromContext(ctx)
+	if err != nil {
+		return err
+	}
+	sql := "delete from project_member where project_id = ?"
+	if _, err := o.Raw(sql, projectID).Exec(); err != nil {
+		return err
+	}
+	return nil
+}

src/pkg/member/manager.go

@@ -44,6 +44,8 @@ type Manager interface {
 	SearchMemberByName(ctx context.Context, projectID int64, entityName string) ([]*models.Member, error)
 	// DeleteMemberByUserID delete project member by user id
 	DeleteMemberByUserID(ctx context.Context, uid int) error
+	// DeleteMemberByProjectID delete project member by project id
+	DeleteMemberByProjectID(ctx context.Context, projectID int64) error
 	// GetTotalOfProjectMembers get the total amount of project members
 	GetTotalOfProjectMembers(ctx context.Context, projectID int64, query *q.Query, roles ...int) (int, error)
 	// ListRoles list project roles
@@ -101,6 +103,9 @@ func (m *manager) Delete(ctx context.Context, projectID int64, memberID int) err
 func (m *manager) DeleteMemberByUserID(ctx context.Context, uid int) error {
 	return m.dao.DeleteProjectMemberByUserID(ctx, uid)
 }
+func (m *manager) DeleteMemberByProjectID(ctx context.Context, projectID int64) error {
+	return m.dao.DeleteProjectMemberByProjectID(ctx, projectID)
+}
 
 // NewManager ...
 func NewManager() Manager {