Example Needed (Async MySQL + Ratchet)

[KristienJones]

Hi. I need some examples of using Async MySQL with an IoServer. I'll be using https://github.com/friends-of-reactphp/mysql most likely, but I can't get it to work correctly with Ratchet.

My Scenario: I need to authenticate connections using tokens that are stored on a database. The user connects with a token. Then once connected, the server authenticates and allows the connection to continue.

I'm entirely new to Ratchet and using composer also, so i'm clearly a little lost. I've spent the last day trying to figure it out but to no avail.

Below is my working code (Non-Async) server.php:

      use Ratchet\Server\IoServer;
	use Ratchet\Http\HttpServer;
	use Ratchet\WebSocket\WsServer;
	use MyApp\Chat;
	require dirname(__DIR__) . '/vendor/autoload.php';
	
	$server = IoServer::factory(
		new HttpServer(
			new WsServer(
				new Chat()
			)
		),
		8080
	);
	
	$server->run();

Chat.php

	namespace MyApp;
	use Ratchet\MessageComponentInterface;
	use Ratchet\ConnectionInterface;
	use mysqli;

	class Chat implements MessageComponentInterface {
		protected $clients;
		protected $publicKey;
		protected $privateKey;
		protected $db;

		public function __construct() {
			$this->publicKey = openssl_pkey_get_public(file_get_contents(dirname(__DIR__).'/certificates/webSocket/webSocket.pem'));
			$this->privateKey = openssl_pkey_get_private(file_get_contents(dirname(__DIR__).'/certificates/webSocket/webSocket.pem'));
			
			$this->clients = new \SplObjectStorage;
			echo "Server Started.";
		}

		public function onOpen(ConnectionInterface $conn) {
			$this->clients->attach($conn);
			echo "New connection! ({$conn->resourceId})\n";
			$authFail = false;
			
			$this->db = new mysqli("localhost","root","","lewisjones"); 
			if($this->db){
				$querystring = $conn->httpRequest->getUri()->getQuery();
				parse_str($querystring,$queryarray);
				
				if(!isset($queryarray['token'])){
					$authFail = true;
				} else {
					$getSession = $this->query("select token from wsSessions where userID = ? limit 1",array(1));
					if(!is_array($getSession)){
						$authFail = true;
					} else {
						$verify = openssl_verify($getSession[0]["token"], base64_decode(base64_decode($queryarray['token'])), $this->publicKey, OPENSSL_ALGO_SHA256);
						if($verify != 1) {
							$authFail = true;
						}
					}
					$this->query("delete from wsSessions where userID = ?",array(1));
				}
				$this->db->close();
			} else {
				$authFail = true;
			}
			
			if($authFail){
				foreach ($this->clients as $client) {
					if ($conn->resourceId !== $client) {
						$client->send("Authentication failed! Connection Closed.");
						$conn->close();
					}
				}
			}
		}

		public function onMessage(ConnectionInterface $from, $msg) {
			$numRecv = count($this->clients) - 1;
			echo sprintf('Connection %d sending message "%s" to %d other connection%s' . "\n"
				, $from->resourceId, $msg, $numRecv, $numRecv == 1 ? '' : 's');

			foreach ($this->clients as $client) {
				if ($from !== $client) {
					$client->send($msg);
				}
			}
		}

		public function onClose(ConnectionInterface $conn) {
			$this->clients->detach($conn);
			echo "Connection {$conn->resourceId} has disconnected\n";
		}

		public function onError(ConnectionInterface $conn, \Exception $e) {
			echo "An error has occurred: {$e->getMessage()}\n";
			$conn->close();
		}
		
		private function query($sql,$params = null){
			$types = null;
			if(is_array($params)){
				$types = "";
				foreach($params as $key => $value){
					if(is_int($value)){
						$types .= "i";
					} else {
						if(is_float($value)){
							$types .= "d";
						} else {
							$types .= "s";
						}
					}
				}
			}
			
			if($stmt = $this->db->prepare($sql)){
				if($types && $params){
					$bind_names[] = $types;
					for ($i=0; $i<count($params);$i++){
						$bind_name = 'bind' . $i;
						$$bind_name = $params[$i];
						$bind_names[] = &$$bind_name;
					}
					call_user_func_array(array($stmt,'bind_param'),$bind_names);
				}
				
				$stmt->execute();
				$result = $stmt->get_result();
				
				if (!empty($result) && $result->num_rows > 0) {
					$returnArray = array();
					while ($row = $result->fetch_assoc()) {
						$returnArray[] = $row;
					}
					return $returnArray;
				}
				
				if($stmt){
					return true;
				} 
				return false;
				
				$stmt->free_result();
				$stmt->close();
			}
			return false;
		}
	}

If anyone can point me in the right direction, that would be appreciated. Thanks.

[KristienJones]

OK. Update, I've managed to get it working but i've hit another wall.

Due to the nature of it being asynchronous, I cannot seem to manage variables correctly (Which I need in order to close a connection if the auth fails)

This is what I have so far:

public function onOpen(ConnectionInterface $conn) {
			$this->clients->attach($conn);
			echo "New connection! ({$conn->resourceId})\n";
			
			$connection = $this->factory->createLazyConnection($this->uri);
			$connection->query("select token from wsSessions where userID = ? limit 1",[1])->then(function(ConnectionInterface $conn,QueryResult $command) {
				//logic here, if not authorized, close connection
				$conn->close();
			});
			$connection->quit();
		}

The problem I have, is that I cannot close the connection. $conn doesn't seem to parse correctly. And I can't update global variables and do the logic outside of the query function because it's asynchronous.

Any advice?

[FeIix]

Hi, I think you just need to add use($conn) to use it in your anonymous function scope.

Like this

$connection->query("select token from wsSessions where userID = ? limit 1", [1])
    ->then(function(ConnectionInterface $conn,QueryResult $command) use($conn) { // <- here
        //logic here, if not authorized, close connection
        $conn->close();
    });