return error when uuid is not enough

pkg/services/accounts.go

@@ -163,12 +163,22 @@ func (s *AccountService) CreateAccounts(c *core.Context, mainAccount *models.Acc
 	var allInitTransactions []*models.Transaction
 
 	mainAccount.AccountId = s.GenerateUuid(uuid.UUID_TYPE_ACCOUNT)
+
+	if mainAccount.AccountId < 1 {
+		return errs.ErrSystemIsBusy
+	}
+
 	allAccounts[0] = mainAccount
 
 	if mainAccount.Type == models.ACCOUNT_TYPE_MULTI_SUB_ACCOUNTS {
 		for i := 0; i < len(childrenAccounts); i++ {
 			childAccount := childrenAccounts[i]
 			childAccount.AccountId = s.GenerateUuid(uuid.UUID_TYPE_ACCOUNT)
+
+			if childAccount.AccountId < 1 {
+				return errs.ErrSystemIsBusy
+			}
+
 			childAccount.ParentAccountId = mainAccount.AccountId
 			childAccount.Uid = mainAccount.Uid
 			childAccount.Type = models.ACCOUNT_TYPE_SINGLE_ACCOUNT
@@ -185,8 +195,14 @@ func (s *AccountService) CreateAccounts(c *core.Context, mainAccount *models.Acc
 		allAccounts[i].UpdatedUnixTime = now
 
 		if allAccounts[i].Balance != 0 {
+			transactionId := s.GenerateUuid(uuid.UUID_TYPE_TRANSACTION)
+
+			if transactionId < 1 {
+				return errs.ErrSystemIsBusy
+			}
+
 			newTransaction := &models.Transaction{
-				TransactionId:        s.GenerateUuid(uuid.UUID_TYPE_TRANSACTION),
+				TransactionId:        transactionId,
 				Uid:                  allAccounts[i].Uid,
 				Deleted:              false,
 				Type:                 models.TRANSACTION_DB_TYPE_MODIFY_BALANCE,

pkg/services/transaction_categories.go

@@ -163,6 +163,10 @@ func (s *TransactionCategoryService) CreateCategory(c *core.Context, category *m
 
 	category.CategoryId = s.GenerateUuid(uuid.UUID_TYPE_CATEGORY)
 
+	if category.CategoryId < 1 {
+		return errs.ErrSystemIsBusy
+	}
+
 	category.Deleted = false
 	category.CreatedUnixTime = time.Now().Unix()
 	category.UpdatedUnixTime = time.Now().Unix()
@@ -186,6 +190,10 @@ func (s *TransactionCategoryService) CreateCategories(c *core.Context, uid int64
 		primaryCategory := primaryCategories[i]
 		primaryCategory.CategoryId = s.GenerateUuid(uuid.UUID_TYPE_CATEGORY)
 
+		if primaryCategory.CategoryId < 1 {
+			return nil, errs.ErrSystemIsBusy
+		}
+
 		primaryCategory.Deleted = false
 		primaryCategory.CreatedUnixTime = time.Now().Unix()
 		primaryCategory.UpdatedUnixTime = time.Now().Unix()
@@ -197,6 +205,11 @@ func (s *TransactionCategoryService) CreateCategories(c *core.Context, uid int64
 		for j := 0; j < len(secondaryCategories); j++ {
 			secondaryCategory := secondaryCategories[j]
 			secondaryCategory.CategoryId = s.GenerateUuid(uuid.UUID_TYPE_CATEGORY)
+
+			if secondaryCategory.CategoryId < 1 {
+				return nil, errs.ErrSystemIsBusy
+			}
+
 			secondaryCategory.ParentCategoryId = primaryCategory.CategoryId
 
 			secondaryCategory.Deleted = false

pkg/services/transaction_tags.go

@@ -160,6 +160,10 @@ func (s *TransactionTagService) CreateTag(c *core.Context, tag *models.Transacti
 
 	tag.TagId = s.GenerateUuid(uuid.UUID_TYPE_TAG)
 
+	if tag.TagId < 1 {
+		return errs.ErrSystemIsBusy
+	}
+
 	tag.Deleted = false
 	tag.CreatedUnixTime = time.Now().Unix()
 	tag.UpdatedUnixTime = time.Now().Unix()

pkg/services/transactions.go

@@ -222,6 +222,11 @@ func (s *TransactionService) CreateTransaction(c *core.Context, transaction *mod
 	}
 
 	uuids := s.GenerateUuids(uuid.UUID_TYPE_TRANSACTION, uint8(needUuidCount))
+
+	if len(uuids) < needUuidCount {
+		return errs.ErrSystemIsBusy
+	}
+
 	transaction.TransactionId = uuids[0]
 
 	if transaction.Type == models.TRANSACTION_DB_TYPE_TRANSFER_OUT || transaction.Type == models.TRANSACTION_DB_TYPE_TRANSFER_IN {
@@ -237,8 +242,14 @@ func (s *TransactionService) CreateTransaction(c *core.Context, transaction *mod
 	transactionTagIndexs := make([]*models.TransactionTagIndex, len(tagIds))
 
 	for i := 0; i < len(tagIds); i++ {
+		tagIndexId := s.GenerateUuid(uuid.UUID_TYPE_TAG_INDEX)
+
+		if tagIndexId < 1 {
+			return errs.ErrSystemIsBusy
+		}
+
 		transactionTagIndexs[i] = &models.TransactionTagIndex{
-			TagIndexId:      s.GenerateUuid(uuid.UUID_TYPE_TAG_INDEX),
+			TagIndexId:      tagIndexId,
 			Uid:             transaction.Uid,
 			Deleted:         false,
 			TagId:           tagIds[i],
@@ -431,8 +442,14 @@ func (s *TransactionService) ModifyTransaction(c *core.Context, transaction *mod
 	transactionTagIndexs := make([]*models.TransactionTagIndex, len(addTagIds))
 
 	for i := 0; i < len(addTagIds); i++ {
+		tagIndexId := s.GenerateUuid(uuid.UUID_TYPE_TAG_INDEX)
+
+		if tagIndexId < 1 {
+			return errs.ErrSystemIsBusy
+		}
+
 		transactionTagIndexs[i] = &models.TransactionTagIndex{
-			TagIndexId:      s.GenerateUuid(uuid.UUID_TYPE_TAG_INDEX),
+			TagIndexId:      tagIndexId,
 			Uid:             transaction.Uid,
 			Deleted:         false,
 			TagId:           addTagIds[i],

pkg/services/users.go

@@ -153,6 +153,11 @@ func (s *UserService) CreateUser(c *core.Context, user *models.User) error {
 	}
 
 	user.Uid = s.GenerateUuid(uuid.UUID_TYPE_USER)
+
+	if user.Uid < 1 {
+		return errs.ErrSystemIsBusy
+	}
+
 	user.Password = utils.EncodePassword(user.Password, user.Salt)
 
 	user.Deleted = false