MOD amélioration de la vitesse de euphorik_bd:messages (mais toujours en O(n)). Cette...
[euphorik.git] / modules / erl / euphorik_bd.erl
index 1cd534e..4fecb73 100755 (executable)
 \r
 -module(euphorik_bd).\r
 -export([   \r
+   % texte :\r
+   get_texte/1,\r
+   get_texte/2,\r
+   \r
    % users :\r
    nouveau_user/2,\r
    nouveau_user/4,\r
    css_from_user_cookie/1,\r
    is_ek_master_from_cookie/1,\r
    \r
-   % messages :e\r
+   % messages :\r
    nouveau_message/3,\r
    nouveau_message_sys/1,\r
-   nouveau_message_sys/2,\r
    messages/1,\r
    messages/2,\r
    messages/3,\r
@@ -74,7 +77,6 @@
    troll_by_id/1,\r
    current_troll/0,\r
    elire_troll/0,\r
-   message_id_associe/1,\r
    \r
    % utiles :\r
    resultat_transaction/1\r
 -include_lib("stdlib/include/qlc.hrl").\r
 \r
 \r
+get_texte(Id) ->\r
+   get_texte(Id, fr).\r
+\r
+\r
+% TODO : généraliser la langue\r
+get_texte(Id, _Lang = fr) ->\r
+   resultat_transaction(mnesia:transaction(fun() ->\r
+      case mnesia:read({texte, Id}) of\r
+         [#texte{fr = Texte}] -> Texte;\r
+         _ -> "Message " ++ integer_to_list(Id) ++ " unknown"\r
+      end\r
+   end)).\r
+\r
+\r
 % Ajoute un nouveau user et le renvoie\r
 nouveau_user(Cookie, Profile) ->\r
    F = fun() ->\r
@@ -179,7 +195,7 @@ update_pseudo_user(UserId, Pseudo) ->
 user_by_cookie(Cookie) ->\r
    resultat_transaction(mnesia:transaction(\r
       fun() ->\r
-         case e(q([E || E <- mnesia:table(user), E#user.cookie =:= Cookie]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+         case mnesia:index_read(user, Cookie, #user.cookie) of\r
             [User] -> {ok, User};\r
             _ -> erreur\r
          end\r
@@ -258,14 +274,20 @@ user_by_mess(Id) ->
 % retourne soit l'id du message soit {erreur, <raison>}.\r
 nouveau_message(Mess, Auteur_id, Repond_A_ids) ->\r
    % regarde si les id 'Repond_A' existent\r
-   F = fun() ->   \r
-      Repond_a = e(\r
-         q([M || M <- mnesia:table(minichat), lists:member(M#minichat.id, Repond_A_ids)]),\r
-         [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
-      ), \r
+   F = fun() ->         \r
+      Repond_a = lists:foldr(\r
+         fun(Repond_a_id, Acc) ->\r
+            case mnesia:read({minichat, Repond_a_id}) of
+               [M] -> [M | Acc];\r
+               _ -> Acc % le message n'est pas trouvé
+            end\r
+         end,
+         [],\r
+         Repond_A_ids\r
+      ),\r
       Racine_id = case Repond_a of\r
          [] -> undefined;\r
-         [M | _ ] -> \r
+         [M | _] -> \r
             Une_racine = M#minichat.racine_id,\r
             % vérification que tout les messages de Repond_a possède la même racine (même conversation)\r
             case lists:all(fun(R) -> R#minichat.racine_id =:= Une_racine end, Repond_a) of\r
@@ -274,47 +296,48 @@ nouveau_message(Mess, Auteur_id, Repond_A_ids) ->
                _ ->\r
                   {erreur, "Les messages ne font pas partie de la même conversation"}\r
             end\r
-      end,      \r
-      case Racine_id of\r
-         {erreur, E} -> {erreur, E};\r
-         _ ->\r
-            % est-ce que l'auteur existe ?\r
-            case e(q([E || E <- mnesia:table(user), E#user.id =:= Auteur_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
-               [#user{profile = Profile} = Auteur] ->\r
-                  if length(Repond_a) =/= length(Repond_A_ids) ->\r
-                        {erreur, "Un ou plusieurs messages introuvable"};\r
-                     true ->\r
+      end,      
+      if length(Repond_a) =/= length(Repond_A_ids) ->
+            {erreur, "Un ou plusieurs messages introuvable"};
+         true ->\r
+            case Racine_id of\r
+               {erreur, E} -> {erreur, E};\r
+               _ ->\r
+                  % est-ce que l'auteur existe ?\r
+                  case mnesia:wread({user, Auteur_id}) of\r
+                     [#user{profile = Profile} = Auteur] ->\r
                         % comparaison entre la date du dernier poste et maintenant (gestion du flood)\r
-                        Delta = delta_date_ms(Auteur#user.date_derniere_connexion, now()),\r
+                        Now = now(),\r
+                        Delta = euphorik_common:delta_date_ms(Auteur#user.date_derniere_connexion, Now),\r
                         Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end,\r
                         Auteur_maj = Auteur#user{\r
                            indice_flood = if Nouvel_indice_flood > ?INDICE_SPAM_MAX -> ?INDICE_SPAM_MAX; Nouvel_indice_flood < 0 -> 0; true -> Nouvel_indice_flood end,\r
-                           date_derniere_connexion = now()\r
+                           date_derniere_connexion = Now\r
                         },\r
                         % est-ce que l'auteur à trop floodé ?\r
                         if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->\r
-                           mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}),\r
-                           nouveau_message_sys("''" ++ Profile#profile.pseudo ++ if Auteur#user.login =/= [] -> " (" ++ Auteur#user.login ++ ")"; true -> "" end ++ "'' est bloqué pour " ++ integer_to_list(trunc(?DUREE_BLOCAGE_SPAM / 1000)) ++ " secondes pour cause de flood.");\r
-                        Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->\r
-                           {erreur, "Bloqué pour cause de flood"};\r
-                        true ->     \r
-                           mnesia:write(Auteur_maj),\r
-                           Id = nouvel_id(minichat),\r
-                           inserer_reponses(Id, Repond_A_ids),\r
-                           mnesia:write(#minichat{\r
-                              id = Id,\r
-                              auteur_id = Auteur#user.id,\r
-                              date = now(),\r
-                              pseudo = Profile#profile.pseudo,\r
-                              contenu = Mess,\r
-                              racine_id = if Racine_id =:= undefined -> Id; true -> Racine_id end\r
-                           }),\r
-                           Id\r
-                        end\r
-                  end;\r
-               _ ->\r
-                  {erreur, "L'auteur du message est introuvable"}\r
-            end\r
+                              mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}),\r
+                              nouveau_message_sys("\"" ++ Profile#profile.pseudo ++ if Auteur#user.login =/= [] -> " (" ++ Auteur#user.login ++ ")"; true -> "" end ++ "\" est bloqué pour " ++ integer_to_list(trunc(?DUREE_BLOCAGE_SPAM / 1000)) ++ " secondes pour cause de flood.");\r
+                           Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->\r
+                              {erreur, "Bloqué pour cause de flood"};\r
+                           true ->     \r
+                              mnesia:write(Auteur_maj),\r
+                              Id = nouvel_id(minichat),\r
+                              inserer_reponses(Id, Repond_A_ids),\r
+                              mnesia:write(#minichat{\r
+                                 id = Id,\r
+                                 auteur_id = Auteur#user.id,\r
+                                 date = Now,\r
+                                 pseudo = Profile#profile.pseudo,\r
+                                 contenu = Mess,\r
+                                 racine_id = if Racine_id =:= undefined -> Id; true -> Racine_id end\r
+                              }),\r
+                              Id\r
+                        end;\r
+                     _ ->\r
+                        {erreur, "L'auteur du message est introuvable"}\r
+                  end\r
+            end
       end\r
    end,\r
    resultat_transaction(mnesia:transaction(F)).\r
@@ -330,16 +353,11 @@ inserer_reponses(_, []) ->
 % Permet de créer un message système.\r
 % Renvoie l'id du message système\r
 nouveau_message_sys(Mess) ->\r
-   nouveau_message_sys(Mess, undefined).\r
-   \r
-\r
-% Création d'un message système lié à un troll.\r
-nouveau_message_sys(Mess, Troll_id) ->\r
    {ok, #user{profile = Profile}} = user_by_id(0),\r
    resultat_transaction(mnesia:transaction(\r
       fun() ->\r
          Id = nouvel_id(minichat),\r
-         mnesia:write(#minichat{id = Id, auteur_id = 0, date = now(), pseudo = Profile#profile.pseudo, contenu = Mess, troll_id = Troll_id, racine_id = Id}),\r
+         mnesia:write(#minichat{id = Id, auteur_id = 0, date = now(), pseudo = Profile#profile.pseudo, contenu = Mess, racine_id = Id}),\r
          Id\r
       end\r
    )).\r
@@ -350,26 +368,29 @@ messages(N) ->
    messages(N, 1).\r
 \r
 \r
-% Renvoie N messages se trouvant sur la page P\r
+% Renvoie N messages se trouvant sur la page P
+% TODO FIXME : fonction en O(N * P) !\r
 messages(N, P) ->\r
-   F = fun() ->\r
-      C = cursor(\r
-         qlc:keysort(\r
-            #minichat.id, \r
-            q([E#minichat{contenu = contenu_message(E)} || E <- mnesia:table(minichat)]),\r
-            [{order, descending}]\r
-         ),\r
-         [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
-      ),\r
-      if P > 1 -> qlc:next_answers(C, N * (P - 1));\r
-         true -> ok\r
-      end,\r
-      R = qlc:next_answers(C, N),\r
-      qlc:delete_cursor(C),\r
-      R\r
+   F = fun() ->
+      % % #minichat{contenu = contenu_message(E)}
+      get_tuples_avant(minichat, reculer(minichat, mnesia:last(minichat), N * (P - 1)), N)     \r
    end,\r
-   lists:reverse(resultat_transaction(mnesia:transaction(F))).\r
-\r
+   resultat_transaction(mnesia:transaction(F)).
+   
+get_tuples_avant(Table, Id, N) ->
+   get_tuples_avant(Table, Id, N, []).
+get_tuples_avant(_, '$end_of_table', _, Tuples) -> Tuples;
+get_tuples_avant(_, _, 0, Tuples) ->
+   Tuples;
+get_tuples_avant(Table, Id, N, Tuples) ->
+   [T] = mnesia:read({Table, Id}),
+   get_tuples_avant(Table, mnesia:prev(Table, Id), N - 1, [T | Tuples]).
+
+reculer(_, '$end_of_table' = Fin, _) -> Fin;
+reculer(_, Id, 0) -> Id;
+reculer(Table, Id, N) ->
+   reculer(Table, mnesia:prev(Table, Id), N - 1).
+   \r
 \r
 % Renvoie les messages manquants pour la page P en sachant qu'il y a N message\r
 % par page et que le dernier message que l'on possède est Id\r
@@ -390,9 +411,9 @@ message_by_id(Id) ->
    \r
    \r
 % Renvoie le contenu d'un message donnée en fonction du troll associé, à utiliser à l'intérieur d'une transaction.\r
-% TODO : Cette fonction pourrait être remplacé par un "outer-join", est-ce possible avec qlc ?\r
+% TODO : Cette fonction pourrait être remplacée par un "outer-join", est-ce possible avec qlc ?\r
 contenu_message(E) ->\r
-   case mnesia:read({troll, E#minichat.troll_id}) of\r
+   case mnesia:index_read(troll, E#minichat.id, #troll.id_minichat) of\r
       [] -> E#minichat.contenu;\r
       [T] -> E#minichat.contenu ++ T#troll.content\r
    end.\r
@@ -455,33 +476,36 @@ enfants(M_id) ->
    )).\r
    \r
    \r
-% Renvoie les parents d'un message M (les messages auquels répond M)\r
+% Renvoie les id des parents d'un message M (les messages auquels répond M)\r
 % ordrés du plus petit au plus grand..\r
+% On évite d'utiliser qlc pour des raisons de performance\r
 % @spec parents_id(integer()) -> [integer()]\r
-parents_id(M) ->\r
+parents_id(M_id) ->\r
    resultat_transaction(mnesia:transaction(fun() ->\r
-      e(\r
-         qlc:sort(\r
-            q([E#reponse_minichat.cible || E <- mnesia:table(reponse_minichat), E#reponse_minichat.repondant =:= M]),\r
-            [{order, ascending}]\r
-         ),\r
-         [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
-      )\r
+      case mnesia:read({reponse_minichat, M_id}) of\r
+         Parents when is_list(Parents) ->\r
+            lists:sort(lists:map(\r
+               fun(#reponse_minichat{cible = Cible}) -> Cible end,\r
+               Parents\r
+            ));\r
+         _ -> []            \r
+      end\r
    end)).\r
    \r
    \r
 % Renvoie les id des enfants d'un message M (les messages qui répondent à M)\r
 % ordrés du plus petit au plus grand.\r
 % @spec enfants_id(integer()) -> [integer()]\r
-enfants_id(M) ->\r
+enfants_id(M_id) ->\r
    resultat_transaction(mnesia:transaction(fun() ->\r
-      e(\r
-         qlc:sort(\r
-            q([E#reponse_minichat.repondant || E <- mnesia:table(reponse_minichat), E#reponse_minichat.cible =:= M]),\r
-            [{order, ascending}]\r
-         ),\r
-         [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
-      )\r
+      case mnesia:index_read(reponse_minichat, M_id, #reponse_minichat.cible) of\r
+         Enfants when is_list(Enfants) ->\r
+            lists:sort(lists:map(\r
+               fun(#reponse_minichat{repondant = Repondant}) -> Repondant end,\r
+               Enfants\r
+            ));\r
+         _ -> []            \r
+      end\r
    end)).\r
      \r
 \r
@@ -512,7 +536,7 @@ est_une_reponse_a_user(Id_user, Id_mess) ->
 a_repondu_a_message(Id_user, Id_mess) ->\r
    resultat_transaction(mnesia:transaction(\r
       fun() ->\r
-         case mnesia:match_object({reponse_minichat, '_', Id_mess}) of\r
+         case mnesia:index_read(reponse_minichat, Id_mess, #reponse_minichat.cible) of\r
             [] -> false;\r
             Reponses ->\r
                lists:any(\r
@@ -551,7 +575,7 @@ list_ban() ->
          e(qlc:keysort(1, q([\r
             {\r
                IP#ip_table.ip,\r
-               delta_date_minute(date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration), Now),\r
+               euphorik_common:delta_date_minute(date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration), Now),\r
                e(q([{Profile#profile.pseudo, U#user.login} || #user{profile = Profile} = U <- mnesia:table(user), U#user.last_ip =:= IP#ip_table.ip]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
             } ||\r
             IP <- mnesia:table(ip_table),\r
@@ -598,7 +622,7 @@ est_banni(User_id) ->
                if Echeance < Now -> % l'échéance est passée\r
                      false;\r
                   true ->\r
-                     {true, delta_date_minute(Echeance, Now)}\r
+                     {true, euphorik_common:delta_date_minute(Echeance, Now)}\r
                end;\r
             _ ->\r
                false\r
@@ -627,11 +651,11 @@ can_register(IP) ->
                mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}),\r
                true;\r
             [T] ->\r
-               Delta = delta_date_ms(T#ip_table.date_last_try_register, now()),\r
+               Delta = euphorik_common:delta_date_ms(T#ip_table.date_last_try_register, now()),\r
                if T#ip_table.nb_try_register =:= ?NB_MAX_FLOOD_REGISTER, Delta < ?TEMPS_BAN_FLOOD_REGISTER ->\r
                      false;\r
                   true ->\r
-                     mnesia:write(#ip_table{\r
+                     mnesia:write(T#ip_table{\r
                         ip = IP,\r
                         date_last_try_register = now(),\r
                         nb_try_register = T#ip_table.nb_try_register + if Delta < ?TEMPS_FLOOD_REGISTER -> 1; T#ip_table.nb_try_register > 0 -> -1; true -> 0 end\r
@@ -771,39 +795,17 @@ elire_troll() ->
                plus_de_trolls;\r
             Trolls ->\r
                Troll = lists:nth(random:uniform(length(Trolls)), Trolls),\r
-               Troll2 = Troll#troll{date_post = now()},\r
-               mnesia:write(Troll2),\r
-               nouveau_message_sys("Troll de la semaine : ", Troll2#troll.id)\r
+               Id_message = nouveau_message_sys("Troll de la semaine : "),\r
+               Troll2 = Troll#troll{date_post = now(), id_minichat = Id_message},\r
+               mnesia:write(Troll2)\r
          end\r
       end\r
    ).\r
    \r
-\r
-% Renvoie l'id du message associé au troll dont l'id est donnée.\r
-% Renvoie undefined si il n'y en a pas.\r
-message_id_associe(Troll_id) ->\r
-   resultat_transaction(mnesia:transaction(\r
-      fun() ->\r
-         case e(q([M#minichat.id || M <- mnesia:table(minichat), M#minichat.troll_id =:= Troll_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
-            [Id] -> Id;\r
-            _ -> undefined\r
-         end\r
-      end\r
-   )).\r
-\r
    \r
 % Renvoie le résultat d'une transaction (en décomposant le tuple fournit)\r
 resultat_transaction({_, T}) ->\r
    T.\r
-   \r
-\r
-% Retourne la difference entre deux timestamp (erlang:now()) en miliseconde\r
-delta_date_ms(D1, D2) ->\r
-   1000000000 * abs(element(1, D1) - element(1, D2)) + 1000 * abs(element(2, D1) - element(2, D2)) + trunc(abs(element(3, D1) - element(3, D2)) / 1000).\r
-\r
-% Retourne la différence entre deux timestamp (erlang:now) en minutes\r
-delta_date_minute(D1, D2) ->\r
-   trunc(delta_date_ms(D1, D2) / 1000 / 60).\r
 \r
 \r
 % Renvoie un nouvel id pour une table donnée\r