Tarantool: Хороший, Плохой, Злой06.02.2017 17:18

local function bootstrap()

    box.schema.user.create('good', { password = 'secret' })
    box.schema.user.grant('good', 'read,write,execute', 'universe')
   
   -----------------------------------
    -- Пространство стикеров
    local stickers = box.schema.create_space('stickers')
        -- Индекс для файла
    stickers:create_index('primary', { 
        type = 'TREE', parts = {1, 'unsigned'}
    })

    -- Индекс для рейтинга
    stickers:create_index('secondary', {
        type = 'TREE',
        unique = false,
        parts = {2, 'integer'}
    })

    -- Индекс для названий стикеров
    stickers:create_index('ternary', {
        type ='HASH', parts = {3, 'string', 4, 'string'}
    })

    -----------------------------------
    -- Пространство стикер-пака
    local packs = box.schema.create_space('packs')
    -- Индекс для стикер-пака
    packs:create_index('primary', {
        type = 'HASH', parts = {1, 'string'}
    })
    
    -- Индекс для рейтинга пака
    packs:create_index('secondary', {
        type = 'TREE',
        unique = false, 
        parts = {2, 'integer'}
    })
    
    -----------------------------------
    -- Пространство секретных ссылок
    local secret = box.schema.create_space('secret')
    
    -- Индекс для секретного ключа
    secret:create_index('primary', {
        type = 'HASH', parts = {1, 'string'}
    })
    
    -- Индекс для времени создания ключа
    secret:create_index('secondary', {
        type = 'TREE',
        unique = false, 
        parts = {2, 'integer'}
    })
    
    -----------------------------------
    -- Пространство сессий пользователей
    local sessions = box.schema.create_space('sessions')

    -- Индекс для уникального пользователя   
        sessions:create_index('primary', {
        type = 'HASH', parts = {1, 'string'}
    })
    
    -- Индекс для времени создания сессии
    sessions:create_index('secondary', {
        type = 'TREE',
        unique = false, 
        parts = {2, 'integer'}
    })
    
    -----------------------------------
    -- Пространство статистики сервера
    local server = box.schema.create_space('server')

    -- Просто индекс   
        server:create_index('primary', {
        type = 'TREE', parts = {1, 'unsigned'}
    })
    -- Создадим запись
    server:insert{1, 0, 0, 0}
   
end

function create_random_vote() {
# Get random sticker id

    global $tarantool;
    $tuple = $tarantool->call("box.space.stickers.index.primary:random", array(rand()));
    $id = $tuple[0][0];
    $url = $tuple[0][4];

# Create random sticker token
     $token = $tarantool->evaluate(
        "digest = require('digest')
        return ( digest.md5_hex(digest.urandom(32)))"
    )[0];

$time = time();

# Set secure token to protect post action
##################################################################
#
# box.space.secret:insert({key', 0, 456, 'bla-bla'})
#
##################################################################
    $tarantool->insert('secret', array ($token, $time, $id, $url));

    return array (
        $url,
        $token
    );
}

function update_votes($id, $plus, $minus) {
    global $tarantool;

###########################################################
#
# box.space.stickers:update(1, {{'+', 6, 1}, {'+', 7, -1}})
#
###########################################################
   $tarantool->update("stickers", $id,
        array (
            array(
                "field" => 5,
                "op" => "+",
                "arg" => $plus
            ),
            array(
                "field" => 6,
                "op" => "+",
                "arg" => $minus
            )
        )
    );
}

ping();
} catch (Exception $e) {
        echo "Exception: ", $e->getMessage(), "\n";
}

const MIN_VOTES = 20;           // Number of votes to show the ugly
const UPDATE_PLUS = 1;          // Increment for positive update
const UPDATE_MINUS = -1;        // Increment for negative update
const NO_UPDATE = 0;
const COOKIE = 'uuid';          // Cookie name
const HIDDEN = '/img/Question.svg';// Picture for hidden element

# Get routes from request
$route = isset($_REQUEST['q']) ? $_REQUEST['q'] : '/';
$vote_plus = isset($_REQUEST['vote_plus']) ? $_REQUEST['vote_plus'] : '';
$vote_minus = isset($_REQUEST['vote_minus']) ? $_REQUEST['vote_minus'] : '';

# Get cookie from request or create new value
$cookie = isset($_COOKIE[COOKIE]) ? $_COOKIE[COOKIE] : update_user('');

switch ($route) {
    case '/good':
        action_good();
        break;
    case '/bad':
        action_bad();
        break;
    case '/ugly':
        action_ugly($cookie);
        break;
    case '/about':
        action_about();
        break;
    default:
# This is post request:
        if (!empty($vote_plus) && !empty($vote_minus)) {
           sleep (1);

           $cookie = update_user($cookie);
           do_post($vote_plus, $vote_minus);

        }
        setcookie(COOKIE, $cookie, time() + (86400 * 30), "/");
        action_main();
}

exit();

function action_main() {
    global $tarantool;

# Get crypto safe random seed from Tarantool LUA module
# https://tarantool.org/doc/reference/reference_lua/digest.html
    $r = $tarantool->evaluate(
        "digest = require('digest')
        return (digest.urandom(4))"
    );

    $seed = unpack('L', $r[0])[1];
    srand($seed);

    list ($left_url, $left_token_plus) = create_random_vote();
    list ($right_url, $right_token_plus) = create_random_vote();

    $left_token_minus = $right_token_plus;
    $right_token_minus = $left_token_plus;

    update_stats(UPDATE_PLUS, NO_UPDATE);

    $title = 'Хорошие и плохие стикеры Telegram';
    include_once('main.html');
}

function action_good() {
    $title = 'ТОП лучших стикеров Telegram';
    $top = get_top(10,Tarantool::ITERATOR_LE);
    $active_good ='class="active"';
    $active_bad ='';

    include_once('top.html');
}

function action_bad () {
    $title = 'ТОП худших стикеров Telegram';
    $active_bad ='class="active"';
    $active_good ='';
    $top = get_top(10,Tarantool::ITERATOR_GE);
# Hide the ugly
    $top[0][4] = HIDDEN;
    include_once('top.html');
}

function action_ugly($user) {
    $title = 'Худший стикер Telegram';
    $top = get_top(1,Tarantool::ITERATOR_GE);
    $votes = get_session($user);
# Hide the ugly until getting enough votes
    if ($votes < MIN_VOTES) {
        $ugly_message = "Голосуй " . MIN_VOTES . " раз чтобы увидеть результат
";
        $ugly_message .= "Осталось " . (MIN_VOTES - $votes) . " голосований";
        $ugly_img = HIDDEN;
    } else {
        $ugly_img = $top[0][4];
    }
    include_once('ugly.html');
}

function action_about() {
    $title = 'Как это сделано?';
    list($stickers, $shows, $votes, $visitors) = get_server_stats();
    include_once('about.html');
}

function do_post($vote_plus, $vote_minus) {
    global $tarantool;

    $tuple_plus = $tarantool->select("secret", $vote_plus);
    $tuple_minus = $tarantool->select("secret", $vote_minus);

    $id_plus = $tuple_plus[0][2];
    $id_minus = $tuple_minus[0][2];

# Clean up used tokens
    if (!empty($vote_plus) && !empty($vote_minus)) {
        $tarantool->delete("secret", $vote_plus);
        $tarantool->delete("secret", $vote_minus);
    }

# Get actual tuple data
    if (!empty($id_plus) && !empty($id_minus)) {

        $raiting = +1;
        update_rating($id_plus, $raiting);

        $raiting = -1;
        update_rating($id_minus, $raiting);

        update_votes($id_plus,  UPDATE_PLUS, NO_UPDATE);
        update_votes($id_minus, NO_UPDATE,  UPDATE_MINUS);

        update_stats(NO_UPDATE, UPDATE_PLUS);
    }
}

function create_random_vote() {
# Get random sticker id

    global $tarantool;
    $tuple = $tarantool->call("box.space.stickers.index.primary:random", array(rand()));
    $id = $tuple[0][0];
    $url = $tuple[0][4];

# Create random sticker token
     $token = $tarantool->evaluate(
        "digest = require('digest')
        return ( digest.md5_hex(digest.urandom(32)))"
    )[0];

$time = time();

# Set secure token to protect post action
##################################################################
#
# box.space.secret:insert({key', 0, 456, 'bla-bla'})
#
##################################################################
    $tarantool->insert('secret', array ($token, $time, $id, $url));

    return array (
        $url,
        $token
    );
}

function update_rating($id, $update) {
    global $tarantool;

#################################################
#
# box.space.stickers:update(7856, {{'+', 2, 10}})
#
#################################################
   $tarantool->update("stickers", $id, array (
        array(
            "field" => 1,
            "op" => "+",
            "arg" => $update
        )
    ));
}

function update_votes($id, $plus, $minus) {
    global $tarantool;

###########################################################
#
# box.space.stickers:update(1, {{'+', 6, 1}, {'+', 7, -1}})
#
###########################################################
   $tarantool->update("stickers", $id,
        array (
            array(
                "field" => 5,
                "op" => "+",
                "arg" => $plus
            ),
            array(
                "field" => 6,
                "op" => "+",
                "arg" => $minus
            )
        )
    );
}

function update_user($cookie) {
    global $tarantool;

# Create uuid if first time user
    if (empty($cookie)) {

##################################
#
# uuid = require('uuid')
# uuid()
#
##################################
        $uuid = $tarantool->evaluate(
           "uuid = require('uuid')
            return (uuid.str())"
        )[0];
    } else {
       $uuid = $cookie;
    }
$time = time();
$ip = isset($_SERVER['REMOTE_ADDR']) ? $_SERVER['REMOTE_ADDR'] : '';
$agent = isset($_SERVER['HTTP_USER_AGENT']) ? $_SERVER['HTTP_USER_AGENT'] : '';

# Create session or update user stat inside
###########################################################
#
# box.space.sessions:upsert({'111222333', 123456, 0, 'ip', 'agent'},
# {{'=', 2, 1}, {'+', 3, 1}, {'=', 4, 'ip'}, {'=', 5, 'agent'}})
#
###########################################################
  # Please check https://github.com/tarantool/tarantool-php/issues/111
   $tarantool->upsert("sessions", array($uuid, $time, 0, $ip, $agent),
        array (
            array(
                "field" => 1
                "op" => "=",
                "arg" => $time
            ),
            array(
                "field" => 2
                "op" => "+",
                "arg" => 1
            ),
            array(
                "field" => 3
                "op" => "=",
                "arg" => $ip
            ),
            array(
                "field" => 4,
                "op" => "=",
                "arg" => $agent
            )
        )
    );
    return($uuid);
}

function update_stats($vote, $click) {
    global $tarantool;

########################################################
#
# box.space.server:update(1, {{'+', 3, 1}, {'+', 4, 1}})
#
########################################################
   $tarantool->update("server",1,
        array (
            array(
                "field" => 2,
                "op" => "+",
                "arg" => $vote
            ),
            array(
                "field" => 3,
                "op" => "+",
                "arg" => $click
            )
        )
    );
}

function get_session($sid) {
    global $tarantool;

##########################################
#
# box.space.sessions:select('id')
#
#########################################
if (strlen($sid) > 16) {
    return $tarantool->select("sessions", $sid)[0][2];
    } else {
        return 0;
    }
}

function get_top($limit, $iterator) {
    global $tarantool;

######################################################################################
#
# box.space.stickers.index.secondary:select({primary}, {iterator = box.index.GE, offset=0, limit=10})
#
######################################################################################
    $result = $tarantool->select("stickers", null, 'secondary', $limit, 0, $iterator);
    return $result;
}
function get_server_stats() {
    global $tarantool;

    $time = time() - 30*86400; // one month before
    $stickers = $tarantool->call('box.space.stickers:count')[0][0];

    $tuple = $tarantool->select('server',1);

    $shows = $tuple[0][2];
    $votes = $tuple[0][3];

    $visitors = $tarantool->call('box.space.sessions.index.secondary:count',
       array($time, array('iterator' => Tarantool::ITERATOR_GE))
    )[0][0];

# $shows, $votes, $visitors) = get_server_stats();

    return array($stickers, $shows, $votes, $visitors);
}
?>