File: Store.php

package info (click to toggle)
simplesamlphp 1.16.3-1%2Bdeb10u2
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 21,036 kB
  • sloc: php: 73,175; ansic: 875; sh: 83; perl: 82; xml: 52; makefile: 46
file content (111 lines) | stat: -rw-r--r-- 3,013 bytes parent folder | download
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
110
111
<?php

/*
 * Copyright (c) 2012 The University of Queensland
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

/*
 * Written by David Gwynne <dlg@uq.edu.au> as part of the IT
 * Infrastructure Group in the Faculty of Engineering, Architecture
 * and Information Technology.
 */

class sspmod_riak_Store_Store extends SimpleSAML\Store
{
    public $client;
    public $bucket;

	protected function __construct()
    {
		$config = SimpleSAML_Configuration::getConfig('module_riak.php');

		$path = $config->getString('path', 'riak-php-client/riak.php');
		$host = $config->getString('host', 'localhost');
		$port = $config->getString('port', 8098);
		$bucket = $config->getString('bucket', 'simpleSAMLphp');

		require_once($path);
		$this->client = new \RiakClient($host, $port);
		$this->bucket = $this->client->bucket($bucket);
	}

	/**
	 * Retrieve a value from the datastore.
	 *
	 * @param string $type  The datatype.
	 * @param string $key  The key.
	 * @return mixed|NULL  The value.
	 */
	public function get($type, $key)
    {
		assert(is_string($type));
		assert(is_string($key));

		$v = $this->bucket->getBinary("$type.$key");
		if (!$v->exists()) {
			return null;
		}

		$expires = $v->getIndex('Expires', 'int');
		if (sizeof($expires) && (int)array_shift($expires) <= time()) {
			$v->delete();
			return null;
		}

		return (unserialize($v->getData()));
	}


	/**
	 * Save a value to the datastore.
	 *
	 * @param string $type  The datatype.
	 * @param string $key  The key.
	 * @param mixed $value  The value.
	 * @param int|NULL $expire  The expiration time (unix timestamp), or NULL if it never expires.
	 */
	public function set($type, $key, $value, $expire = null)
    {
		assert(is_string($type));
		assert(is_string($key));
		assert($expire === null || (is_int($expire) && $expire > 2592000));

		$v = $this->bucket->newBinary("$type.$key", serialize($value), 'application/php');
		if (!is_null($expire)) {
			$v->addIndex("Expires", "int", $expire);
		}

		$v->store();
	}

	/**
	 * Delete a value from the datastore.
	 *
	 * @param string $type  The datatype.
	 * @param string $key  The key.
	 */
	public function delete($type, $key)
    {
		assert(is_string($type));
		assert(is_string($key));

		$v = $this->bucket->getBinary("$type.$key");
		if (!$v->exists()) {
			return;
		}

		$v->delete();
	}
}