queue/Queue_Manager.php at 1.x · mantle-framework/queue · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
<?php
/**
 * Queue_Manager class file.
 *
 * @package Mantle
 */

namespace Mantle\Queue;

use InvalidArgumentException;
use Mantle\Contracts\Container;
use Mantle\Contracts\Queue\Provider;
use Mantle\Contracts\Queue\Queue_Manager as Queue_Manager_Contract;

/**
 * Queue Manager
 */
class Queue_Manager implements Queue_Manager_Contract {
	/**
	 * Provider class map.
	 *
	 * @var class-string<Provider>[]|Provider[]
	 */
	protected array $providers = [];

	/**
	 * Provider connections.
	 *
	 * @var Provider[]
	 */
	protected $connections = [];

	/***
	 * Constructor.
	 *
	 * @param Container $container Container instance.
	 */
	public function __construct( protected Container $container ) {}

	/**
	 * Get a queue provider instance.
	 *
	 * @param string $name Provider name, optional.
	 */
	public function get_provider( ?string $name = null ): Provider {
		$name = $name ?: $this->get_default_driver();

		if ( ! isset( $this->connections[ $name ] ) ) {
			$this->connections[ $name ] = $this->resolve( $name );
		}

		return $this->connections[ $name ];
	}

	/**
	 * Add a provider for the queue manager.
	 *
	 * @param string                          $name Provider name.
	 * @param Provider|class-string<Provider> $provider Provider class name/instance.
	 *
	 * @throws InvalidArgumentException Thrown invalid provider.
	 */
	public function add_provider( string $name, string|Provider $provider ): static {
		if ( is_string( $provider ) && ( ! class_exists( $provider ) || ! in_array( Provider::class, class_implements( $provider ), true ) ) ) {
			throw new InvalidArgumentException( "Provider does not implement Provider contract: [{$provider}]" );
		}

		$this->providers[ $name ] = $provider;

		return $this;
	}

	/**
	 * Get the default queue driver in queue.
	 */
	protected function get_default_driver(): string {
		if ( ! isset( $this->container['config'] ) ) {
			return 'wordpress';
		}

		return $this->container['config']['queue.default'] ?? 'wordpress';
	}

	/**
	 * Resolve a connection to a queue provider.
	 *
	 * @param string $provider Provider name.
	 *
	 * @throws InvalidArgumentException Thrown on invalid provider name.
	 * @throws InvalidArgumentException Thrown on invalid provider instance resolved.
	 */
	protected function resolve( string $provider ): Provider {
		if ( ! isset( $this->providers[ $provider ] ) ) {
			throw new InvalidArgumentException( "No provider found for [{$provider}]." );
		}

		if ( ! is_object( $this->providers[ $provider ] ) ) {
			$this->connections[ $provider ] = $this->container->make( $this->providers[ $provider ] );
		} else {
			$this->connections[ $provider ] = $this->providers[ $provider ];
		}

		if ( ! ( $this->connections[ $provider ] instanceof Provider ) ) {
			throw new InvalidArgumentException( "Unknown provider instance resolved for [{$provider}]: " . $this->connections[ $provider ]::class );
		}

		return $this->connections[ $provider ];
	}
}