Add callback and update user role

internal/repository/orders/order.go

@@ -24,7 +24,7 @@ func (r *OrderRepository) Create(ctx context.Context, order *entity.Order) (*ent
 		logger.ContextLogger(ctx).Error("error when creating order", zap.Error(err))
 		return nil, err
 	}
-	return order, nil
+	return r.FindByID(ctx, order.ID)
 }
 
 func (r *OrderRepository) UpdateStatus(ctx context.Context, orderID int64, status string) (*entity.Order, error) {
@@ -43,13 +43,36 @@ func (r *OrderRepository) UpdateStatus(ctx context.Context, orderID int64, statu
 
 func (r *OrderRepository) FindByID(ctx context.Context, id int64) (*entity.Order, error) {
 	var order entity.Order
-	if err := r.db.WithContext(ctx).Preload("OrderItems").First(&order, id).Error; err != nil {
+
+	err := r.db.WithContext(ctx).Preload("OrderItems", func(db *gorm.DB) *gorm.DB {
+		return db.Preload("Product")
+	}).First(&order, id).Error
+
+	if err != nil {
 		logger.ContextLogger(ctx).Error("error when finding order by ID", zap.Error(err))
 		return nil, err
 	}
+
 	return &order, nil
 }
 
+func (r *OrderRepository) SetOrderStatus(ctx context.Context, db *gorm.DB, orderID int64, status string) error {
+	var order entity.Order
+	if err := db.WithContext(ctx).Preload("OrderItems").First(&order, orderID).Error; err != nil {
+		logger.ContextLogger(ctx).Error("error when finding order by ID", zap.Error(err))
+		return err
+	}
+
+	order.Status = status
+
+	if err := db.WithContext(ctx).Save(&order).Error; err != nil {
+		logger.ContextLogger(ctx).Error("error when updating order status", zap.Error(err))
+		return err
+	}
+
+	return nil
+}
+
 func (r *OrderRepository) Update(ctx context.Context, order *entity.Order) (*entity.Order, error) {
 	if err := r.db.WithContext(ctx).Save(order).Error; err != nil {
 		logger.ContextLogger(ctx).Error("error when updating order", zap.Error(err))