data management page shows all user data statistics

pkg/api/data_managements.go

@@ -118,6 +118,47 @@ func (a *DataManagementsApi) ExportDataHandler(c *core.Context) ([]byte, string,
 	return result, fileName, nil
 }
 
+// DataStatisticsHandler returns user data statistics
+func (a *DataManagementsApi) DataStatisticsHandler(c *core.Context) (interface{}, *errs.Error) {
+	uid := c.GetCurrentUid()
+	totalAccountCount, err := a.accounts.GetTotalAccountCountByUid(uid)
+
+	if err != nil {
+		log.ErrorfWithRequestId(c, "[data_managements.DataStatisticsHandler] failed to get total account count for user \"uid:%d\", because %s", uid, err.Error())
+		return nil, errs.ErrOperationFailed
+	}
+
+	totalTransactionCategoryCount, err := a.categories.GetTotalCategoryCountByUid(uid)
+
+	if err != nil {
+		log.ErrorfWithRequestId(c, "[data_managements.DataStatisticsHandler] failed to get total transaction category count for user \"uid:%d\", because %s", uid, err.Error())
+		return nil, errs.ErrOperationFailed
+	}
+
+	totalTransactionTagCount, err := a.tags.GetTotalTagCountByUid(uid)
+
+	if err != nil {
+		log.ErrorfWithRequestId(c, "[data_managements.DataStatisticsHandler] failed to get total transaction tag count for user \"uid:%d\", because %s", uid, err.Error())
+		return nil, errs.ErrOperationFailed
+	}
+
+	totalTransactionCount, err := a.transactions.GetTotalTransactionCountByUid(uid)
+
+	if err != nil {
+		log.ErrorfWithRequestId(c, "[data_managements.DataStatisticsHandler] failed to get total transaction count for user \"uid:%d\", because %s", uid, err.Error())
+		return nil, errs.ErrOperationFailed
+	}
+
+	dataStatisticsResp := &models.DataStatisticsResponse{
+		TotalAccountCount:             totalAccountCount,
+		TotalTransactionCategoryCount: totalTransactionCategoryCount,
+		TotalTransactionTagCount:      totalTransactionTagCount,
+		TotalTransactionCount:         totalTransactionCount,
+	}
+
+	return dataStatisticsResp, nil
+}
+
 // ClearDataHandler deletes all user data
 func (a *DataManagementsApi) ClearDataHandler(c *core.Context) (interface{}, *errs.Error) {
 	var clearDataReq models.ClearDataRequest

pkg/models/data_management.go

@@ -4,3 +4,11 @@ package models
 type ClearDataRequest struct {
 	Password string `json:"password" binding:"omitempty,min=6,max=128"`
 }
+
+// DataStatisticsResponse represents a view-object of user data statistic
+type DataStatisticsResponse struct {
+	TotalAccountCount             int64 `json:"totalAccountCount,string"`
+	TotalTransactionCategoryCount int64 `json:"totalTransactionCategoryCount,string"`
+	TotalTransactionTagCount      int64 `json:"totalTransactionTagCount,string"`
+	TotalTransactionCount         int64 `json:"totalTransactionCount,string"`
+}

pkg/services/accounts.go

@@ -30,6 +30,17 @@ var (
 	}
 )
 
+// GetTotalAccountCountByUid returns total account count of user
+func (s *AccountService) GetTotalAccountCountByUid(uid int64) (int64, error) {
+	if uid <= 0 {
+		return 0, errs.ErrUserIdInvalid
+	}
+
+	count, err := s.UserDataDB(uid).Where("uid=? AND deleted=?", uid, false).Count(&models.Account{})
+
+	return count, err
+}
+
 // GetAllAccountsByUid returns all account models of user
 func (s *AccountService) GetAllAccountsByUid(uid int64) ([]*models.Account, error) {
 	if uid <= 0 {

pkg/services/transaction_categories.go

@@ -29,6 +29,17 @@ var (
 	}
 )
 
+// GetTotalCategoryCountByUid returns total category count of user
+func (s *TransactionCategoryService) GetTotalCategoryCountByUid(uid int64) (int64, error) {
+	if uid <= 0 {
+		return 0, errs.ErrUserIdInvalid
+	}
+
+	count, err := s.UserDataDB(uid).Where("uid=? AND deleted=?", uid, false).Count(&models.TransactionCategory{})
+
+	return count, err
+}
+
 // GetAllCategoriesByUid returns all transaction category models of user
 func (s *TransactionCategoryService) GetAllCategoriesByUid(uid int64, categoryType models.TransactionCategoryType, parentCategoryId int64) ([]*models.TransactionCategory, error) {
 	if uid <= 0 {

pkg/services/transaction_tags.go

@@ -29,6 +29,17 @@ var (
 	}
 )
 
+// GetTotalTagCountByUid returns total tag count of user
+func (s *TransactionTagService) GetTotalTagCountByUid(uid int64) (int64, error) {
+	if uid <= 0 {
+		return 0, errs.ErrUserIdInvalid
+	}
+
+	count, err := s.UserDataDB(uid).Where("uid=? AND deleted=?", uid, false).Count(&models.TransactionTag{})
+
+	return count, err
+}
+
 // GetAllTagsByUid returns all transaction tag models of user
 func (s *TransactionTagService) GetAllTagsByUid(uid int64) ([]*models.TransactionTag, error) {
 	if uid <= 0 {

pkg/services/transactions.go

@@ -32,6 +32,17 @@ var (
 	}
 )
 
+// GetTotalTransactionCountByUid returns total transaction count of user
+func (s *TransactionService) GetTotalTransactionCountByUid(uid int64) (int64, error) {
+	if uid <= 0 {
+		return 0, errs.ErrUserIdInvalid
+	}
+
+	count, err := s.UserDataDB(uid).Where("uid=? AND deleted=?", uid, false).Count(&models.Transaction{})
+
+	return count, err
+}
+
 // GetAllTransactions returns all transactions
 func (s *TransactionService) GetAllTransactions(uid int64, pageCount int32, noDuplicated bool) ([]*models.Transaction, error) {
 	maxTransactionTime := utils.GetMaxTransactionTimeFromUnixTime(time.Now().Unix())