feat:support export api impl

Author: taolx0

internal/apiserver/service/dms_controller.go

@@ -2586,7 +2586,28 @@ func (d *DMSController) ListCBOperationLogs(c echo.Context) error {
 //	  200: ExportCBOperationLogsReply
 //	  default: body:GenericResp
 func (d *DMSController) ExportCBOperationLogs(c echo.Context) error {
-	return nil
+	req := &aV1.ExportCBOperationLogsReq{}
+	err := bindAndValidateReq(c, req)
+	if nil != err {
+		return NewErrResp(c, err, apiError.BadRequestErr)
+	}
+
+	currentUserUid, err := jwt.GetUserUidStrFromContext(c)
+	if err != nil {
+		return NewErrResp(c, err, apiError.DMSServiceErr)
+	}
+
+	content, err := d.DMS.ExportCBOperationLogs(c.Request().Context(), req, currentUserUid)
+	if err != nil {
+		return NewErrResp(c, err, apiError.APIServerErr)
+	}
+
+	fileName := fmt.Sprintf("CBoperation_%s.csv", time.Now().Format("20060102150405.000"))
+	c.Response().Header().Set(echo.HeaderContentDisposition,
+		mime.FormatMediaType("attachment", map[string]string{"filename": fileName}))
+
+	return c.Blob(http.StatusOK, "text/csv", content)
+
 }
 
 // swagger:route GET /v1/dms/projects/{project_uid}/cb_operation_logs/tips dms GetCBOperationLogTips

internal/dms/biz/cb_operation_log.go

@@ -12,6 +12,8 @@ type CbOperationLogType string
 
 const (
 	CbOperationLogTypeSql CbOperationLogType = "SQL"
+
+	CbExecOpSuccess = "Success"
 )
 
 // CbOperationLogRepo 定义操作日志的存储接口
@@ -21,6 +23,7 @@ type CbOperationLogRepo interface {
 	UpdateCbOperationLog(ctx context.Context, log *CbOperationLog) error
 	ListCbOperationLogs(ctx context.Context, opt *ListCbOperationLogOption) ([]*CbOperationLog, int64, error)
 	CleanCbOperationLogOpTimeBefore(ctx context.Context, t time.Time) (int64, error)
+	CountOperationLogs(ctx context.Context, opt *ListCbOperationLogOption) (int64, error)
 }
 
 // CbOperationLog 代表操作日志记录
@@ -42,6 +45,7 @@ type CbOperationLog struct {
 
 	User      *User
 	DbService *DBService
+	Project   *Project
 }
 
 func (c CbOperationLog) GetOpTime() time.Time {
@@ -58,6 +62,28 @@ func (c CbOperationLog) GetSessionID() string {
 	return ""
 }
 
+func (c CbOperationLog) GetUserName() string {
+	if c.User != nil {
+		return c.User.Name
+	}
+	return ""
+}
+
+func (c CbOperationLog) GetProjectName() string {
+	if c.Project != nil {
+		return c.Project.Name
+	}
+	return ""
+}
+
+func (c CbOperationLog) GetDbServiceName() string {
+	if c.DbService != nil {
+		return c.DbService.Name
+	}
+	return ""
+
+}
+
 // ListCbOperationLogOption 用于查询操作日志的选项
 type ListCbOperationLogOption struct {
 	PageNumber   uint32

internal/dms/service/cb_operation_log.go

@@ -13,3 +13,7 @@ func (d *DMSService) ListCBOperationLogs(ctx context.Context, req *dmsV1.ListCBO
 func (d *DMSService) GetCBOperationLogTips(ctx context.Context, req *dmsV1.GetCBOperationLogTipsReq, uid string) (reply *dmsV1.GetCBOperationLogTipsReply, err error) {
 	return d.getCBOperationLogTips(ctx, req, uid)
 }
+
+func (d *DMSService) ExportCBOperationLogs(ctx context.Context, req *dmsV1.ExportCBOperationLogsReq, uid string) ([]byte, error) {
+	return d.exportCbOperationLogs(ctx, req, uid)
+}

internal/dms/service/cb_operation_log_ce.go

@@ -18,3 +18,7 @@ func (d *DMSService) listCBOperationLogs(ctx context.Context, req *dmsV1.ListCBO
 func (d *DMSService) getCBOperationLogTips(ctx context.Context, req *dmsV1.GetCBOperationLogTipsReq, currentUid string) (*dmsV1.GetCBOperationLogTipsReply, error) {
 	return nil, errNotSupportCBOperationLog
 }
+
+func (d *DMSService) exportCbOperationLogs(ctx context.Context, req *dmsV1.ExportCBOperationLogsReq, uid string) ([]byte, error) {
+	return nil, errNotSupportCBOperationLog
+}

internal/dms/storage/cb_operation_log.go

@@ -69,7 +69,7 @@ func (d *CbOperationLogRepo) ListCbOperationLogs(ctx context.Context, opt *biz.L
 	if err := transaction(d.log, ctx, d.db, func(tx *gorm.DB) error {
 		// find models
 		{
-			db := tx.WithContext(ctx).Preload("User").Preload("DbService")
+			db := tx.WithContext(ctx).Preload("Project").Preload("User").Preload("DbService")
 			if opt.OrderBy != "" {
 				db = db.Order(fmt.Sprintf("%s DESC", opt.OrderBy))
 			}
@@ -116,3 +116,19 @@ func (d *CbOperationLogRepo) CleanCbOperationLogOpTimeBefore(ctx context.Context
 	})
 	return
 }
+
+func (d *CbOperationLogRepo) CountOperationLogs(ctx context.Context, opt *biz.ListCbOperationLogOption) (int64, error) {
+	var total int64
+	if err := transaction(d.log, ctx, d.db, func(tx *gorm.DB) error {
+		db := tx.WithContext(ctx).Model(&model.CbOperationLog{})
+		db = gormWheres(ctx, db, opt.FilterBy)
+		if err := db.Count(&total).Error; err != nil {
+			return fmt.Errorf("failed to count cb operation logs: %v", err)
+		}
+		return nil
+	}); err != nil {
+		return 0, err
+	}
+
+	return total, nil
+}

internal/dms/storage/convert.go

@@ -1188,6 +1188,11 @@ func convertModelCbOperationLog(model *model.CbOperationLog) (*biz.CbOperationLo
 		return nil, err
 	}
 
+	project, err := convertModelProject(model.Project)
+	if err != nil {
+		return nil, err
+	}
+
 	return &biz.CbOperationLog{
 		UID:               model.UID,
 		ProjectID:         model.ProjectID,
@@ -1205,5 +1210,6 @@ func convertModelCbOperationLog(model *model.CbOperationLog) (*biz.CbOperationLo
 		ResultSetRowCount: model.ResultSetRowCount,
 		User:              user,
 		DbService:         dbService,
+		Project:           project,
 	}, nil
 }

internal/dms/storage/model/model.go

@@ -531,4 +531,5 @@ type CbOperationLog struct {
 
 	User      *User      `json:"user" gorm:"foreignKey:OpPersonUID"`
 	DbService *DBService `json:"db_service" gorm:"foreignKey:DBServiceUID"`
+	Project   *Project   `json:"project" gorm:"foreignKey:ProjectID"`
 }