%% 启动 Mnesia
application:start(mnesia).

%% 创建 Schema（数据目录）
mnesia:create_schema([node()]).  %% 在当前节点创建

%% 启动后创建表
mnesia:create_table(person, [
    {attributes, record_info(fields, person)},
    {disc_copies, [node()]},  %% 磁盘副本
    {type, set}               %% 表类型
]).

%% 使用 record 定义表结构
-record(person, {
    id,         %% 主键（默认第一个字段）
    name,
    age,
    city
}).

%% 创建表
mnesia:create_table(person, [
    {attributes, record_info(fields, person)},
    {type, set},
    {ram_copies, [node()]}  %% 仅内存
]).

%% 写入
mnesia:transaction(fun() ->
    mnesia:write(#person{id = 1, name = "Alice", age = 25, city = "Beijing"})
end).

%% 读取
mnesia:transaction(fun() ->
    mnesia:read(person, 1)
end).
%% {atomic, [{person, 1, "Alice", 25, "Beijing"}]}

%% 删除
mnesia:transaction(fun() ->
    mnesia:delete({person, 1})
end).

%% 更新（先读再写）
mnesia:transaction(fun() ->
    [P] = mnesia:read(person, 1),
    mnesia:write(P#person{age = 26})
end).

%% dirty 操作不经过事务管理，性能更好
%% 但不保证一致性

mnesia:dirty_write(#person{id = 2, name = "Bob", age = 30, city = "Shanghai"}).
mnesia:dirty_read(person, 2).
mnesia:dirty_delete({person, 2}).

%% dirty 更新计数器
mnesia:dirty_update_counter(counter_table, my_counter, 1).

%% 事务保证 ACID
%% Atomicity: 原子性（全部成功或全部回滚）
%% Consistency: 一致性
%% Isolation: 隔离性
%% Durability: 持久性（disc_copies 表）

mnesia:transaction(fun() ->
    %% 在事务中可以执行多个操作
    mnesia:write(#person{id = 1, name = "Alice", age = 25}),
    mnesia:write(#person{id = 2, name = "Bob", age = 30}),
    
    %% 如果中间任何操作失败，所有操作都会回滚
    %% 例如：
    case some_condition() of
        true -> ok;
        false -> mnesia:abort(rollback)  %% 中止事务
    end,
    
    ok
end).

%% 成功
{atomic, Result} = mnesia:transaction(fun() ->
    mnesia:read(person, 1)
end).

%% 失败
{aborted, Reason} = mnesia:transaction(fun() ->
    mnesia:write(#person{id = 1, name = "Alice", age = 25}),
    mnesia:abort(some_reason)
end).

%% 默认超时 4 秒，可自定义
mnesia:transaction(fun() ->
    mnesia:read(person, 1)
end, 10000).  %% 10 秒超时

%% 使用 QLC（Query List Comprehension）查询
-include_lib("stdlib/include/qlc.hrl").

%% 查询所有年龄 > 20 的人
mnesia:transaction(fun() ->
    Q = qlc:q([P || P = #person{age = Age} <- mnesia:table(person), Age > 20]),
    qlc:e(Q)
end).

%% 查询特定城市的人
mnesia:transaction(fun() ->
    Q = qlc:q([{Name, Age} || #person{name = Name, age = Age, city = "Beijing"}
                                <- mnesia:table(person)]),
    qlc:e(Q)
end).

%% 排序
mnesia:transaction(fun() ->
    Q = qlc:q([P || P <- mnesia:table(person)],
              [{order_by, fun(#person{age = A}) -> A end}]),
    qlc:e(Q)
end).

%% 在 Node2 上加入集群
mnesia:change_config(extra_db_nodes, ['node2@host']).

%% 复制表到 Node2
mnesia:add_table_copy(person, 'node2@host', ram_copies).

%% 改变存储类型
mnesia:change_table_copy_type(person, 'node2@host', disc_copies).

%% 事务自动在所有节点上执行
mnesia:transaction(fun() ->
    mnesia:write(#person{id = 1, name = "Alice"})
end).
%% 如果表有多个副本，事务会在所有副本上提交

%% session_store.erl
-module(session_store).
-export([init/0, create/3, get/1, update/2, delete/1]).

-record(session, {
    id,          %% session ID
    user_id,     %% 用户 ID
    data,        %% 会话数据 (Map)
    expires_at   %% 过期时间
}).

init() ->
    mnesia:create_table(session, [
        {attributes, record_info(fields, session)},
        {type, set},
        {ram_copies, [node()]},
        {index, [user_id]}
    ]).

create(SessionId, UserId, Data) ->
    ExpiresAt = erlang:system_time(second) + 3600,  %% 1 小时后过期
    Session = #session{
        id = SessionId,
        user_id = UserId,
        data = Data,
        expires_at = ExpiresAt
    },
    mnesia:dirty_write(Session).

get(SessionId) ->
    case mnesia:dirty_read(session, SessionId) of
        [#session{expires_at = Expires} = Session] ->
            Now = erlang:system_time(second),
            if
                Now < Expires -> {ok, Session};
                true ->
                    delete(SessionId),
                    {error, expired}
            end;
        [] ->
            {error, not_found}
    end.

update(SessionId, NewData) ->
    case mnesia:dirty_read(session, SessionId) of
        [Session] ->
            mnesia:dirty_write(Session#session{data = NewData}),
            ok;
        [] ->
            {error, not_found}
    end.

delete(SessionId) ->
    mnesia:dirty_delete({session, SessionId}).