163 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			PHP
		
	
	
			
		
		
	
	
			163 lines
		
	
	
		
			5.6 KiB
		
	
	
	
		
			PHP
		
	
	
| <?php
 | |
| 
 | |
| /*
 | |
|  * This file is part of the Symfony package.
 | |
|  *
 | |
|  * (c) Fabien Potencier <fabien@symfony.com>
 | |
|  *
 | |
|  * For the full copyright and license information, please view the LICENSE
 | |
|  * file that was distributed with this source code.
 | |
|  */
 | |
| 
 | |
| namespace Symfony\Component\HttpFoundation;
 | |
| 
 | |
| /**
 | |
|  * StreamedJsonResponse represents a streamed HTTP response for JSON.
 | |
|  *
 | |
|  * A StreamedJsonResponse uses a structure and generics to create an
 | |
|  * efficient resource-saving JSON response.
 | |
|  *
 | |
|  * It is recommended to use flush() function after a specific number of items to directly stream the data.
 | |
|  *
 | |
|  * @see flush()
 | |
|  *
 | |
|  * @author Alexander Schranz <alexander@sulu.io>
 | |
|  *
 | |
|  * Example usage:
 | |
|  *
 | |
|  *     function loadArticles(): \Generator
 | |
|  *         // some streamed loading
 | |
|  *         yield ['title' => 'Article 1'];
 | |
|  *         yield ['title' => 'Article 2'];
 | |
|  *         yield ['title' => 'Article 3'];
 | |
|  *         // recommended to use flush() after every specific number of items
 | |
|  *     }),
 | |
|  *
 | |
|  *     $response = new StreamedJsonResponse(
 | |
|  *         // json structure with generators in which will be streamed
 | |
|  *         [
 | |
|  *             '_embedded' => [
 | |
|  *                 'articles' => loadArticles(), // any generator which you want to stream as list of data
 | |
|  *             ],
 | |
|  *         ],
 | |
|  *     );
 | |
|  */
 | |
| class StreamedJsonResponse extends StreamedResponse
 | |
| {
 | |
|     private const PLACEHOLDER = '__symfony_json__';
 | |
| 
 | |
|     /**
 | |
|      * @param mixed[]                        $data            JSON Data containing PHP generators which will be streamed as list of data or a Generator
 | |
|      * @param int                            $status          The HTTP status code (200 "OK" by default)
 | |
|      * @param array<string, string|string[]> $headers         An array of HTTP headers
 | |
|      * @param int                            $encodingOptions Flags for the json_encode() function
 | |
|      */
 | |
|     public function __construct(
 | |
|         private readonly iterable $data,
 | |
|         int $status = 200,
 | |
|         array $headers = [],
 | |
|         private int $encodingOptions = JsonResponse::DEFAULT_ENCODING_OPTIONS,
 | |
|     ) {
 | |
|         parent::__construct($this->stream(...), $status, $headers);
 | |
| 
 | |
|         if (!$this->headers->get('Content-Type')) {
 | |
|             $this->headers->set('Content-Type', 'application/json');
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     private function stream(): void
 | |
|     {
 | |
|         $jsonEncodingOptions = \JSON_THROW_ON_ERROR | $this->encodingOptions;
 | |
|         $keyEncodingOptions = $jsonEncodingOptions & ~\JSON_NUMERIC_CHECK;
 | |
| 
 | |
|         $this->streamData($this->data, $jsonEncodingOptions, $keyEncodingOptions);
 | |
|     }
 | |
| 
 | |
|     private function streamData(mixed $data, int $jsonEncodingOptions, int $keyEncodingOptions): void
 | |
|     {
 | |
|         if (\is_array($data)) {
 | |
|             $this->streamArray($data, $jsonEncodingOptions, $keyEncodingOptions);
 | |
| 
 | |
|             return;
 | |
|         }
 | |
| 
 | |
|         if (is_iterable($data) && !$data instanceof \JsonSerializable) {
 | |
|             $this->streamIterable($data, $jsonEncodingOptions, $keyEncodingOptions);
 | |
| 
 | |
|             return;
 | |
|         }
 | |
| 
 | |
|         echo json_encode($data, $jsonEncodingOptions);
 | |
|     }
 | |
| 
 | |
|     private function streamArray(array $data, int $jsonEncodingOptions, int $keyEncodingOptions): void
 | |
|     {
 | |
|         $generators = [];
 | |
| 
 | |
|         array_walk_recursive($data, function (&$item, $key) use (&$generators) {
 | |
|             if (self::PLACEHOLDER === $key) {
 | |
|                 // if the placeholder is already in the structure it should be replaced with a new one that explode
 | |
|                 // works like expected for the structure
 | |
|                 $generators[] = $key;
 | |
|             }
 | |
| 
 | |
|             // generators should be used but for better DX all kind of Traversable and objects are supported
 | |
|             if (\is_object($item)) {
 | |
|                 $generators[] = $item;
 | |
|                 $item = self::PLACEHOLDER;
 | |
|             } elseif (self::PLACEHOLDER === $item) {
 | |
|                 // if the placeholder is already in the structure it should be replaced with a new one that explode
 | |
|                 // works like expected for the structure
 | |
|                 $generators[] = $item;
 | |
|             }
 | |
|         });
 | |
| 
 | |
|         $jsonParts = explode('"'.self::PLACEHOLDER.'"', json_encode($data, $jsonEncodingOptions));
 | |
| 
 | |
|         foreach ($generators as $index => $generator) {
 | |
|             // send first and between parts of the structure
 | |
|             echo $jsonParts[$index];
 | |
| 
 | |
|             $this->streamData($generator, $jsonEncodingOptions, $keyEncodingOptions);
 | |
|         }
 | |
| 
 | |
|         // send last part of the structure
 | |
|         echo $jsonParts[array_key_last($jsonParts)];
 | |
|     }
 | |
| 
 | |
|     private function streamIterable(iterable $iterable, int $jsonEncodingOptions, int $keyEncodingOptions): void
 | |
|     {
 | |
|         $isFirstItem = true;
 | |
|         $startTag = '[';
 | |
| 
 | |
|         foreach ($iterable as $key => $item) {
 | |
|             if ($isFirstItem) {
 | |
|                 $isFirstItem = false;
 | |
|                 // depending on the first elements key the generator is detected as a list or map
 | |
|                 // we can not check for a whole list or map because that would hurt the performance
 | |
|                 // of the streamed response which is the main goal of this response class
 | |
|                 if (0 !== $key) {
 | |
|                     $startTag = '{';
 | |
|                 }
 | |
| 
 | |
|                 echo $startTag;
 | |
|             } else {
 | |
|                 // if not first element of the generic, a separator is required between the elements
 | |
|                 echo ',';
 | |
|             }
 | |
| 
 | |
|             if ('{' === $startTag) {
 | |
|                 echo json_encode((string) $key, $keyEncodingOptions).':';
 | |
|             }
 | |
| 
 | |
|             $this->streamData($item, $jsonEncodingOptions, $keyEncodingOptions);
 | |
|         }
 | |
| 
 | |
|         if ($isFirstItem) { // indicates that the generator was empty
 | |
|             echo '[';
 | |
|         }
 | |
| 
 | |
|         echo '[' === $startTag ? ']' : '}';
 | |
|     }
 | |
| }
 |