mirror of
https://github.com/Abdulazizzn/n8n-enterprise-unlocked.git
synced 2025-12-17 10:02:05 +00:00
feat(Redis Node): Update node-redis (no-changelog) (#8269)
Co-authored-by: Michael Kret <michael.k@radency.com>
This commit is contained in:
committed by
GitHub
parent
3734c89cf6
commit
ab52aaf7e9
168
packages/nodes-base/nodes/Redis/utils.ts
Normal file
168
packages/nodes-base/nodes/Redis/utils.ts
Normal file
@@ -0,0 +1,168 @@
|
||||
import type {
|
||||
ICredentialDataDecryptedObject,
|
||||
ICredentialTestFunctions,
|
||||
ICredentialsDecrypted,
|
||||
IDataObject,
|
||||
IExecuteFunctions,
|
||||
INodeCredentialTestResult,
|
||||
} from 'n8n-workflow';
|
||||
import { NodeOperationError } from 'n8n-workflow';
|
||||
|
||||
import { createClient } from 'redis';
|
||||
export type RedisClientType = ReturnType<typeof createClient>;
|
||||
|
||||
export function setupRedisClient(credentials: ICredentialDataDecryptedObject): RedisClientType {
|
||||
const redisOptions = {
|
||||
socket: {
|
||||
host: credentials.host as string,
|
||||
port: credentials.port as number,
|
||||
},
|
||||
database: credentials.database as number,
|
||||
password: (credentials.password as string) || undefined,
|
||||
};
|
||||
|
||||
return createClient(redisOptions);
|
||||
}
|
||||
|
||||
export async function redisConnectionTest(
|
||||
this: ICredentialTestFunctions,
|
||||
credential: ICredentialsDecrypted,
|
||||
): Promise<INodeCredentialTestResult> {
|
||||
const credentials = credential.data as ICredentialDataDecryptedObject;
|
||||
|
||||
try {
|
||||
const client = setupRedisClient(credentials);
|
||||
await client.connect();
|
||||
await client.ping();
|
||||
return {
|
||||
status: 'OK',
|
||||
message: 'Connection successful!',
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
status: 'Error',
|
||||
message: error.message,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
/** Parses the given value in a number if it is one else returns a string */
|
||||
function getParsedValue(value: string): string | number {
|
||||
if (value.match(/^[\d\.]+$/) === null) {
|
||||
// Is a string
|
||||
return value;
|
||||
} else {
|
||||
// Is a number
|
||||
return parseFloat(value);
|
||||
}
|
||||
}
|
||||
|
||||
/** Converts the Redis Info String into an object */
|
||||
export function convertInfoToObject(stringData: string): IDataObject {
|
||||
const returnData: IDataObject = {};
|
||||
|
||||
let key: string, value: string;
|
||||
for (const line of stringData.split('\n')) {
|
||||
if (['#', ''].includes(line.charAt(0))) {
|
||||
continue;
|
||||
}
|
||||
[key, value] = line.split(':');
|
||||
if (key === undefined || value === undefined) {
|
||||
continue;
|
||||
}
|
||||
value = value.trim();
|
||||
|
||||
if (value.includes('=')) {
|
||||
returnData[key] = {};
|
||||
let key2: string, value2: string;
|
||||
for (const keyValuePair of value.split(',')) {
|
||||
[key2, value2] = keyValuePair.split('=');
|
||||
(returnData[key] as IDataObject)[key2] = getParsedValue(value2);
|
||||
}
|
||||
} else {
|
||||
returnData[key] = getParsedValue(value);
|
||||
}
|
||||
}
|
||||
|
||||
return returnData;
|
||||
}
|
||||
|
||||
export async function getValue(client: RedisClientType, keyName: string, type?: string) {
|
||||
if (type === undefined || type === 'automatic') {
|
||||
// Request the type first
|
||||
type = await client.type(keyName);
|
||||
}
|
||||
|
||||
if (type === 'string') {
|
||||
return client.get(keyName);
|
||||
} else if (type === 'hash') {
|
||||
return client.hGetAll(keyName);
|
||||
} else if (type === 'list') {
|
||||
return client.lRange(keyName, 0, -1);
|
||||
} else if (type === 'sets') {
|
||||
return client.sMembers(keyName);
|
||||
}
|
||||
}
|
||||
|
||||
export async function setValue(
|
||||
this: IExecuteFunctions,
|
||||
client: RedisClientType,
|
||||
keyName: string,
|
||||
value: string | number | object | string[] | number[],
|
||||
expire: boolean,
|
||||
ttl: number,
|
||||
type?: string,
|
||||
valueIsJSON?: boolean,
|
||||
) {
|
||||
if (type === undefined || type === 'automatic') {
|
||||
// Request the type first
|
||||
if (typeof value === 'string') {
|
||||
type = 'string';
|
||||
} else if (Array.isArray(value)) {
|
||||
type = 'list';
|
||||
} else if (typeof value === 'object') {
|
||||
type = 'hash';
|
||||
} else {
|
||||
throw new NodeOperationError(
|
||||
this.getNode(),
|
||||
'Could not identify the type to set. Please set it manually!',
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
if (type === 'string') {
|
||||
await client.set(keyName, value.toString());
|
||||
} else if (type === 'hash') {
|
||||
if (valueIsJSON) {
|
||||
let values: unknown;
|
||||
if (typeof value === 'string') {
|
||||
try {
|
||||
values = JSON.parse(value);
|
||||
} catch {
|
||||
// This is how we originally worked and prevents a breaking change
|
||||
values = value;
|
||||
}
|
||||
} else {
|
||||
values = value;
|
||||
}
|
||||
for (const key of Object.keys(values as object)) {
|
||||
await client.hSet(keyName, key, (values as IDataObject)[key]!.toString());
|
||||
}
|
||||
} else {
|
||||
const values = value.toString().split(' ');
|
||||
await client.hSet(keyName, values);
|
||||
}
|
||||
} else if (type === 'list') {
|
||||
for (let index = 0; index < (value as string[]).length; index++) {
|
||||
await client.lSet(keyName, index, (value as IDataObject)[index]!.toString());
|
||||
}
|
||||
} else if (type === 'sets') {
|
||||
//@ts-ignore
|
||||
await client.sAdd(keyName, value);
|
||||
}
|
||||
|
||||
if (expire) {
|
||||
await client.expire(keyName, ttl);
|
||||
}
|
||||
return;
|
||||
}
|
||||
Reference in New Issue
Block a user