使用Redis的Bitmap统计一周连续登录的用户

const redis = require('redis');
const client = redis.createClient();

// 设置用户登录
function userLogin(userId, day) {
    client.setbit(`login:${userId}`, day, 1, (err, reply) => {
        if (err) {
            console.error('Error:', err);
        } else {
            console.log(`User ${userId} logged in on day ${day}`);
        }
    });
}

// 统计连续登录用户数量
function countConsecutiveLogins(userIds) {
    const dayBits = 7; // 一周7天
    let consecutiveCount = 0;
    for (const userId of userIds) {
        let consecutive = true;
        for (let day = 0; day < dayBits; day++) {
            client.getbit(`login:${userId}`, day, (err, bit) => {
                if (err) {
                    console.error('Error:', err);
                } else {
                    if (bit === 0) {
                        consecutive = false;
                    }
                    if (day === dayBits - 1 && consecutive) {
                        consecutiveCount++;
                        console.log(`User ${userId} logged in consecutively for a week`);
                    }
                }
            });
        }
    }
}

// 示例：设置用户登录和统计连续登录
const userIds = ['user1', 'user2', 'user3'];
for (let i = 0; i < userIds.length; i++) {
    const userId = userIds[i];
    for (let day = 0; day < 7; day++) {
        userLogin(userId, day);
    }
}
countConsecutiveLogins(userIds);

// 省略前文的初始化代码

// 时间窗口滑动统计
function slidingWindowCount(userId, windowSize) {
    const today = 6; // 当天索引
    const maxDays = 7; // 一周7天
    let consecutiveCount = 0;
    for (let startDay = 0; startDay <= today - windowSize + 1; startDay++) {
        let consecutive = true;
        for (let day = startDay; day < startDay + windowSize; day++) {
            client.getbit(`login:${userId}`, day, (err, bit) => {
                if (err) {
                    console.error('Error:', err);
                } else {
                    if (bit === 0) {
                        consecutive = false;
                    }
                    if (day === startDay + windowSize - 1 && consecutive) {
                        consecutiveCount++;
                        console.log(`User ${userId} logged in consecutively for ${windowSize} days from day ${startDay}`);
                    }
                }
            });
        }
    }
}

// 示例：时间窗口滑动统计
slidingWindowCount('user1', 3); // 统计连续登录用户数量，时间窗口大小为3天