PHP Note Online - Thư viện xử lý database và session - Ứng dụng Note và Chat với PHP

<?php

class DB
{
	// Khai báo các biến dưới dạng private
	private $hostname = 'localhost',
			$username = 'root',
			$password = '',
			$dbname = 'inote';

	// Khai báo các biến kết nối toàn cục
	public $cn = NULL;
	public $rs = NULL;

	// Hàm kết nối
	public function connect()
	{
		// Kết nối
		$this->cn = mysqli_connect($this->hostname, $this->username, $this->password, $this->dbname);
	}

	// Hàm ngắt kết nối
	public function close(){
		// Nếu đã kết nối
	    if ($this->cn){
	    	// Ngắt kết nối
	        mysqli_close($this->cn);
	    }
	}

	// Hàm truy vấn
	public function query($sql = null) 
	{		
		// Nếu đã kết nối
	    if ($this->cn){
	    	// Truy vấn
			mysqli_query($this->cn, $sql);
	    }
	}

	// Hàm đếm hàng
	public function num_rows($sql = null) 
	{
		// Nếu đã kết nối
		if ($this->cn)
		{
			$query = mysqli_query($this->cn, $sql);
			$row = mysqli_num_rows($query);
			return $row;
		}
	}

	// Hàm lấy dữ liệu
	public function fetch_assoc($sql = null, $type)
	{
		// Nếu đã kết nối
		if ($this->cn)
		{
			// Thực thi truy vấn
			$query = mysqli_query($this->cn, $sql);
			// Nếu tham số type = 0
			if ($type == 0)
			{
				while ($row = mysqli_fetch_assoc($query))
				{
					$data[] = $row;
				}
				return $data;
			}
			// Nếu tham số type = 1
			else if ($type == 1)
			{
				$data = mysqli_fetch_assoc($query);
				return $data;
			}		
		}
	}

	// Hàm xử lý chuỗi dữ liệu truy vấn
	public function real_escape_string($string) {
		// Nếu đã kết nối
		if ($this->cn)
		{
			// Xử lý chuỗi dữ liệu truy vấn
			$string = mysqli_real_escape_string($this->cn, $string);
		}
		// Ngược lại chưa kết nối
		else
		{
			$string = $string;
		}
		return $string;
	}

	// Hàm lấy ID vừa insert
	public function insert_id() {
		// Nếu đã kết nối
		if ($this->cn)
		{
			// Lấy ID vừa insert
			return mysqli_insert_id($this->cn);
		}
	}
}

?>

<?php

class Session {
	// Hàm gửi dữ liệu
	public function send($user) {
		$_SESSION['user'] = $user;
	}

	// Hàm bắt đầu lưu session
	public function start() {
		session_start();
	}

	// Hàm lấy dữ liệu
	public function get() {
		// Nếu có tồn tại session đang lưu
		if (isset($_SESSION['user']))
		{
			// Gán $user với session
			$user = $_SESSION['user'];
		}
		// Ngược lại không tồn tại session
		else
		{
			$user = '';
		}
		return $user;
	}

	// Hàm xoá session
	public function destroy() {
		session_destroy();
	}
}

?>

<?php

// Include các thư viện
require_once 'classes/DB.php';
require_once 'classes/Session.php';

// Khởi tạo object DB
$db = new DB();
// Kết nối database
$db->connect();

// Khởi tạo object Session
$session = new Session();
// Bắt đầu session
$session->start();
// Lấy dữ liệu session
$user = $session->get();
// Múi giờ chung
date_default_timezone_set('Asia/Ho_Chi_Minh'); 
$date_current = '';
$date_current = date("Y-m-d H:i:sa");

// Nếu tồn tại $user
if ($user)
{
	// Lệnh truy vấn thông tin user
	$sql_get_data_user = "SELECT * FROM users WHERE username = '$user'";

	// Lấy thông tin user
	$data_user = $db->fetch_assoc($sql_get_data_user, 1);
}

?>