Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 | 1x 1x 1x 1x 1x 312x 312x 312x 312x 312x 312x 312x 312x 312x 312x 312x 81x 81x 81x 81x 81x 81x 1x 1x 1x 80x 5x 5x 5x 75x 75x 75x 75x 75x 75x 81x 80x | import {Logger} from "../../logger";
import {IRequest} from "../../server/request";
import {ICmdReq, Session} from "../../server/session";
import {IRespCommand} from "../command/resp-command";
import {DataType} from "../data/data-type";
import {Database} from "../data/database";
import {RedisToken} from "../protocol/redis-token";
export class CommandWrapper extends IRespCommand {
public maxParams: number = -1;
public minParams: number = -1;
public dataType: DataType = DataType.NONE;
public pubSubAllowed: boolean;
private logger: Logger = new Logger(module.id);
constructor(private command: IRespCommand) {
super();
this.logger.info(`Instantiating command ${command.name}`);
Iif (command.dataType) {
this.dataType = command.dataType;
}
this.pubSubAllowed = Boolean(command.pubSubAllowed);
// We have to set these values
this.minParams = Number(command.minParams);
this.maxParams = Number(command.maxParams);
}
public isPubSubAllowed(): boolean {
return this.pubSubAllowed;
}
public execSync(request: IRequest): RedisToken | Promise<RedisToken> {
this.logger.debug(
`${request.getCommand()}.execute(%s)`,
...request.getParams()
);
this.logger.debug(`db id is ${request.getSession().getCurrentDb()}`);
const db: Database = this.getCurrentDB(request);
this.logger.debug(`COMMAND is ${request.getCommand()}`);
this.logger.debug(
"PARAMS Is/Are",
...request.getParams()
);
switch (true) {
case request.getLength() < this.minParams || this.maxParams > -1 && request.getLength() > this.maxParams:
Iif (request.getSession().inTransaction()) {
request.getSession().setError(); // Send EXECABORT if EXEC is called later
}
this.logger.debug(`getLength() is ${request.getLength()}, minParams is ${this.minParams}, maxParams is ${this.maxParams}`);
return RedisToken.error(`ERR wrong number of arguments for '${request.getCommand()}' command`);
break;
case this.dataType !== DataType.NONE &&
!db.isType(
request.getParam(0),
this.dataType
):
if (request.getSession().inTransaction()) {
request.getSession().setError(); // Send EXECABORT if EXEC is called later
}
this.logger.debug(`Check type for ${request.getParam(0)} against ${this.dataType}`);
return RedisToken.error("WRONGTYPE Operation against a key holding the wrong kind of value");
break;
case this.isSubscribed(request) && !this.pubSubAllowed:
return RedisToken.error("ERR only (P)SUBSCRIBE / (P)UNSUBSCRIBE / QUIT allowed in this context");
break;
default:
if (
request.getSession().inTransaction() &&
"exec discard multi".indexOf(request.getCommand().toLowerCase()) === -1) {
this.logger.debug(
`Adding command to transaction for command "${request.getCommand()}" with params [%s]`,
...request.getParams()
);
request.getSession().queueRequest({"command": this.command,
request});
return RedisToken.string("QUEUED");
} else Eif (this.dataType) {
this.logger.debug(
`Executing DB Command "${request.getCommand()}" with params [%s]`,
...request.getParams()
);
const retVal: RedisToken | Promise<RedisToken> = this.executeDBCommand(
request,
db
);
this.logger.debug(
"DBCommand returned %j",
`${retVal}`
);
return retVal;
}
if (this.command.name) {
this.logger.debug(
`Executing RESP Command "${request.getCommand()}" with params [%s]`,
...request.getParams()
);
return this.executeCommand(request);
}
this.logger.warn(
`Could not execute command ${request.getCommand()} with params [%s]`,
...request.getParams()
);
return RedisToken.error(`invalid command type: ${this.command.constructor.name}`);
}
}
private executeCommand(request: IRequest): RedisToken | Promise<RedisToken> {
return this.command.execSync(request);
}
private executeDBCommand(request: IRequest, db: Database): RedisToken | Promise<RedisToken> {
return this.command.execSync(
request,
db
);
}
private getCurrentDB(request: IRequest): Database {
return request.getServerContext().getDatabase(request.getSession().getCurrentDb());
}
private isSubscribed(request: IRequest): boolean {
return request.getSession().getValue("subscribed") === true;
}
}
|