MOD Avancement sur la mise à jour de la BD pour la 1.1 et le système de mise à jour

[euphorik.git] / modules / erl / euphorik_bd_update.erl

diff --git a/modules/erl/euphorik_bd_update.erl b/modules/erl/euphorik_bd_update.erl
new file mode 100644 (file)
index 0000000..a0456bc
--- /dev/null
+++ b/modules/erl/euphorik_bd_update.erl
@@ -0,0 +1,124 @@
+% coding: utf-8
+% Copyright 2008 Grégory Burri
+%
+% This file is part of Euphorik.
+%
+% Euphorik is free software: you can redistribute it and/or modify
+% it under the terms of the GNU General Public License as published by
+% the Free Software Foundation, either version 3 of the License, or
+% (at your option) any later version.
+%
+% Euphorik is distributed in the hope that it will be useful,
+% but WITHOUT ANY WARRANTY; without even the implied warranty of
+% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+% GNU General Public License for more details.
+%
+% You should have received a copy of the GNU General Public License
+% along with Euphorik.  If not, see <http://www.gnu.org/licenses/>.
+% 
+% Module permettant la mise à jour de la base de données.
+% Offre aussi des possibilités de backup et de reprise.
+% @author G.Burri
+
+-module(euphorik_bd_update).
+-export([
+   version_bd/0,
+   update/0,
+   restore/1
+]).
+-import(qlc, [e/2, q/1, cursor/2]).
+-include("../include/euphorik_bd.hrl").
+-include("../include/euphorik_defines.hrl").
+-include_lib("stdlib/include/qlc.hrl").
+
+
+% Renvoie la version courante de la BD.
+version_bd() ->
+   euphorik_bd:resultat_transaction(mnesia:transaction(
+      fun() ->
+         mnesia:read({proprietes, version})
+      end
+   )).
+   
+
+% Met à jour la bd, compare ?VERSION_BD avec la version dans la table 'proprietes'
+% et exécute les patchs nécessaires.
+update() ->
+   case mnesia:dirty_read({proprietes, version}) of
+      [#proprietes{valeur = Version}] ->
+         update(Version);
+      _ ->
+         erreur
+   end.
+   
+
+% Mise à jour de la BD.
+% attention : il est nécessaire de se trouver dans une transaction.
+update(?VERSION_BD) -> fini;
+update(Version) ->
+   case mnesia:backup(mnesia:system_info(directory) ++ "/backup" ++ integer_to_list(Version)) of
+      ok ->
+         case patch(Version) of
+            ok ->
+               mnesia:dirty_write(#proprietes{nom = version, valeur = Version + 1}),
+               update(Version + 1);
+            Erreur ->
+               Erreur
+         end;
+      {error, Raison} -> {error, lists:flatten(io_lib:format("Erreur de création du backup de la version ~w : ~w", [Version, Raison]))}
+   end.
+   
+   
+% Applique une modification de la BD pour passer d'une version à la suivante.
+% crée un backup avant l'application du patch
+% dans BD/backups nommé "backup<num>" où <num> et le numéro de la version.
+% 1 -> 2
+patch(1) ->
+   % traitement des users   
+   mnesia:transform_table(
+      user,
+      fun({user, Id, Cookie, Pseudo, Login, Password, Email, Date_creation, Date_derniere_connexion, Css, Nick_format, View_times, View_tooltips, Indice_flood, _Page_principale, Conversations, Ek_master, Last_ip}) ->
+            {user, Id, Cookie, Pseudo, Login, Password, Email, Date_creation, Date_derniere_connexion, Css, Nick_format, View_times, View_tooltips, reverse, Indice_flood, lists:map(fun({C, _}) -> C end, Conversations), Ek_master, Last_ip}
+      end,
+      record_info(fields, user),
+      user
+   ),
+   mnesia:transform_table(
+      minichat,
+      fun({minichat, Id, Auteur_id, Date, Pseudo, Contenu, Troll_id}) ->
+            {minichat, Id, Auteur_id, Date, Pseudo, Contenu, Troll_id, Id}
+      end,
+      record_info(fields, minichat),
+      minichat
+   ),
+   case mnesia:transaction(
+      fun() ->
+         % met à jour les enfants des racines
+         % idéalement : utiliser un cursor mais je crois qu'il n'est pas possible de faire des modifs en itérant en même temps avec un cursor, a voir..
+         Messages = e(q([M || M <- mnesia:table(minichat), euphorik_bd:parents(M#minichat.id) =:= []]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),
+         lists:foreach(fun(M) -> patch_1_traiter_message(M, M#minichat.id) end, Messages)
+      end
+   ) of
+      {aborted, Raison} -> {erreur, Raison};
+      {atomic, _} -> ok
+   end.
+% 2 -> 3
+%patch(2) ->
+
+
+patch_1_traiter_message(M, Racine) ->
+   % met à jour la racine de chaque message qui répond à M
+   lists:foreach(
+      fun(M2) ->
+         mnesia:write(M2#minichat{racine_id = Racine}),
+         patch_1_traiter_message(M2, Racine)
+      end,
+      euphorik_bd:enfants(M#minichat.id)
+   ).
+
+
+% Reviens à une version précédente de la base de données
+% (les données insérées durant les versions plus récentes sont perdues)
+restore(N) ->
+   mnesia:restore(mnesia:system_info(directory) ++ "/backup" ++ integer_to_list(N)).
+