Question

在我的并发项目中，我需要在redis中获取一个值然后更新它并在redis中设置。像下面的代码一样，我预期的结果应该是3000，但我无法得到正确的结果。序列可能有误，可以GET GET SET SET或GET SET SET GET等。如何获得正确的序列和正确的结果？我应该使用一些锁？

import * as redis from 'redis';

let config: redis.ClientOpts = {
    host: '127.0.0.1',
    port: 6379
};

let redisClient: redis.RedisClient = new redis.RedisClient(config);
redisClient.set('num', '0');

(async () => {
    for (let i = 0; i < 1000; i++) {
        await add ();
    }
})();

(async () => {
    for (let i = 0; i < 1000; i++) {
        await add ();
    }
})();

(async () => {
    for (let i = 0; i < 1000; i++) {
        await add ();
    }
})();

// I know incr command, this is just an example.
async function add () { 
    let numStr: string = await get('num');
    let num: number = Number(numStr);
    num++;
    await set('num', String(num));
    console.log(num);
}

async function get (key: string): Promise<string> {
    return new Promise<string>((resovle, reject) => {
        redisClient.get(key, (err: Error, reply: string) => {
            if (err) {
                console.error(err);
                reject(err);
            }
            resovle(reply);
        })
    });
}

async function set (key: string, value: string): Promise<string> {  
    return new Promise<string>((resovle, reject) => {
        redisClient.set(key, value, (err: Error, reply: string) => {
            if (err) {
                console.error(err);
                reject(err);
            }
            resovle(reply);
        })
    });
}

Answer 1

我已经解决了，谢谢大家。这是我的代码。参考：https://redis.io/topics/distlock

import * as redis from 'redis';
import * as crypto from 'crypto';

const lockScript: string = 'return redis.call("set", KEYS[1], ARGV[1], "NX", "PX", ARGV[2])';
const unlockScript: string = 'if redis.call("get", KEYS[1]) == ARGV[1] then return redis.call("del", KEYS[1]) else return 0 end';

let config: redis.ClientOpts = {
    host: '127.0.0.1',
    port: 6379
};

let redisClient: redis.RedisClient = new redis.RedisClient(config);
redisClient.set('num', '0');

(async () => {
    for (let i = 0; i < 1000; i++) {
        await add ();
    }
})();

(async () => {
    for (let i = 0; i < 1000; i++) {
        await add ();
    }
})();

(async () => {
    for (let i = 0; i < 1000; i++) {
        add ();
    }
})();

async function add () {
    let resourse: string = 'lock:num';
    let uniqueStr: string = crypto.randomBytes(10).toString('hex');
    await attemptLock(resourse, uniqueStr);
    let num: number = Number(await get('num'));
    num++;
    console.log(num);
    await set('num', String(num));
    await unlock(resourse, uniqueStr);
}

async function loop (resourse: string, uniqueStr: string, ttl?: string) {
    return new Promise<any>((resolve, reject) => {
        setTimeout(async () => {
            let result: any = await lock(resourse, uniqueStr, ttl);
            resolve(!!result);
        }, 10);
    });
}

async function attemptLock (resourse: string, uniqueStr: string, ttl?: string) {
    let result = await loop(resourse, uniqueStr, ttl);
    if (result) {
        return true;
    } else {
        return attemptLock(resourse, uniqueStr, ttl);
    }
}

async function lock (resourse: string, uniqueStr: string, ttl?: string) {
    ttl = ttl ? ttl : '30000';

    return new Promise<any>((resolve, reject) => {
        redisClient.eval(lockScript, 1, resourse, uniqueStr, ttl, (err: Error, reply: any) =>{
            if (err) {
                console.error(err);
                reject(err);                
            }

            // reply will be nil when the key exists, on the contrary it will be "OK"
            resolve(reply);
        });
    });
}

async function unlock (resourse: string, uniqueStr: string) {
    return new Promise<any>((resolve, reject) => {
        redisClient.eval(unlockScript, 1, resourse, uniqueStr, (err: Error, reply: any) =>{
            if (err) {
                console.error(err);
                reject(err);
            }

            resolve(reply);
        });
    });
}

async function get (key: string): Promise<string> {
    return new Promise<string>((resovle, reject) => {
        redisClient.get(key, (err: Error, reply: string) => {
            if (err) {
                console.error(err);
                reject(err);
            }
            resovle(reply);
        })
    });
}

async function set (key: string, value: string): Promise<string> {
    return new Promise<string>((resovle, reject) => {
        redisClient.set(key, value, (err: Error, reply: string) => {
            if (err) {
                console.error(err);
                reject(err);
            }
            resovle(reply);
        })
    });
}

Answer 2

您需要解决的问题是执行事务中add（）函数执行的所有操作（获取，增量，设置）以实现原子性 - 这可以通过以下方式在Redis中实现：

Redis事务（MULTI / EXEC / WATCH），具有乐观锁定（重试）
Redis Lua脚本：创建一个将执行添加功能并执行它的lua脚本

Answer 3

Redis提供了一种更灵活的方式，可以使用MULTI-EXEC方法“原子地”执行事务。

想象一下，您正在开发银行系统，并且客户要求进行货币转移。您需要先从客户的账户中扣除资金，然后将其存入另一个账户。在这些操作之间，其中一个可能会失败。所以，我们使用交易。

事务是原子的，这意味着要么所有操作都发生，所有操作都不会发生。数据库具有锁定和并发管理实现，可确保实现此目的，同时不会严重影响系统性能。

示例：

var redis  = require("redis"),
    client = redis.createClient(), multi;

async function add () { 
    multi = client.multi();
    let numStr: string = await multi.get('num');
    let num: number = Number(numStr);
    num++;
    await multi.set('num', String(num));
    multi.exec(function (err, replies) {
        console.log(replies);});
    }
}

Answer 4

您正面临此问题，因为您没有等待每个IFFE功能完成。因此，他们并行运行，不会给你你想要的结果。

要解决此问题，您需要在每个IFFE函数上await。现在，只有await中的函数，我们才能async。所以我将IFFE功能转换为普通功能。这是实现

async function run() {
  for (let i = 0; i < 1000; i++) {
    await add();
  }
}

// I know incr command, this is just an example.
async function add() {
  let numStr = await get('num');
  let num = Number(numStr);
  num++;
  await set('num', String(num));
  console.log(num);
}

function get(key) {
  return new Promise((resovle, reject) => {
    redisClient.get(key, (err, reply) => {
      if (err) {
        console.error(err);
        reject(err);
      }
      resovle(reply);
    });
  });
}

async function set(key, value) {
  return new Promise((resovle, reject) => {
    redisClient.set(key, value, (err, reply) => {
      if (err) {
        console.error(err);
        reject(err);
      }
      resovle(reply);
    })
  });
}

async function runner() {
  await run();
  await run();
  await run();

}
runner();

如何在并发下正确使用nodejs中的redis的GET SET命令？

4 个答案: