ZYZ/game-server/app/services/sdk/ta.ts

/**
 * 模仿thinkingdata-node直接写一个log的写入，而日志直接用pinus-logger实现
 */
import { taLogger } from "../../util/logger";

const KEY_NAME_MATCH_REGEX = /^[a-zA-Z#][a-zA-Z0-9_]+$/;

let util: any = {};
util.version = '1.2.2';
util.log = function(message: string) {
    console.log(message);
}
util.each = function (obj, iterator, context) {
    if (obj === null) {
        return;
    }
    if (Array.prototype.forEach && obj.forEach === Array.prototype.forEach) {
        obj.forEach(iterator, context);
    } else if (obj.length === +obj.length) {
        for (let i = 0, l = obj.length; i < l; i++) {
            // @ts-ignore
            if (i in obj && iterator.call(context, obj[i], i, obj)) {
                return;
            }
        }
    } else {
        for (let key in obj) {
            if (Object.prototype.hasOwnProperty.call(obj, key)) {
                // @ts-ignore
                if (iterator.call(context, obj[key], key, obj)) {
                    return;
                }
            }
        }
    }
};
util.formatDate = function (d) {
    function pad(n) {
        return n < 10 ? '0' + n : n;
    }

    function padMilliseconds(n) {
        if (n < 10) {
            return '00' + n;
        } else if (n < 100) {
            return '0' + n;
        } else {
            return n;
        }
    }
    return d.getFullYear() + '-' +
        pad(d.getMonth() + 1) + '-' +
        pad(d.getDate()) + ' ' +
        pad(d.getHours()) + ':' +
        pad(d.getMinutes()) + ':' +
        pad(d.getSeconds()) + '.' +
        padMilliseconds(d.getMilliseconds());
};

util.formatString = function (format) {
    const args = Array.prototype.slice.call(arguments, 1);
    return format.replace(/{(\d+)}/g, function (match, number) {
        return typeof args[number] !== 'undefined'
            ? args[number]
            : match
            ;
    });
};
util.isArray = Array.isArray;
util.isObject = function (obj) {
    return obj === Object(obj) && !util.isArray(obj) && typeof obj === 'object';
};
util.encodeURIComponent = function (str) {
    return encodeURIComponent(str);
}
util.trim = function (str) {
    return str.replace(/^\s+|\s+$/g, '');
}
util.toArray = function (iterable) {
    if (!iterable) {
        return [];
    }
    if (iterable.toArray) {
        return iterable.toArray();
    }
    return [].slice.call(iterable);
}
util.each(['Function', 'String', 'Object', 'Number', 'Boolean', 'Date', 'RegExp', 'Error'], function (name) {
    util['is' + name] = function (obj) {
        return Object.prototype.toString.call(obj) === '[object ' + name + ']';
    };
});
util.getProperty = function (obj, key) {
    return obj == null ? undefined : obj[key];
};
util.setProperty = function (obj, key, value) {
    obj[key] = value;
};
util.hasProperty = function (obj, key) {
    return obj != null && typeof obj === 'object' && key in obj;
};
util.createLog = function (category) {
    const log = function () {
    };
    log.debug = function () {
    };
    log.info = function () {
    };
    log.warn = function () {
    };
    log.error = function () {
    };
    return log;
};
const TABLE_NAME = "app_gold_record";

const EXTRA_CODE = {
    "#": "_j_",
    "ing": "_deng_",
    "&": "_he_",
};

function encodeKey(key) {
    let result = "";
    for (let i = 0; i < key.length; i++) {
        const c = key.charAt(i);
        result += EXTRA_CODE[c] || c;
    }
    return result;
}

function decodeKey(key) {
    let result = "";
    let i = 0;
    while (i < key.length) {
        const c = key.charAt(i);
        if (c === '_' && i + 2 < key.length) {
            const nextTwo = key.substring(i + 1, i + 3);
            let found = false;
            for (const [encoded, decoded] of Object.entries(EXTRA_CODE)) {
                if (nextTwo === encoded.substring(1)) {
                    result += decoded;
                    i += 3;
                    found = true;
                    break;
                }
            }
            if (!found) {
                result += c;
                i++;
            }
        } else {
            result += c;
            i++;
        }
    }
    return result;
}

function encode(data) {
    const str = JSON.stringify(data);
    return new Buffer(str).toString("base64");
}

function decode(str) {
    return JSON.parse(new Buffer(str, "base64").toString());
}

function encodeId(id) {
    return encodeKey(encode({
        "appid": 1,
        "id": id,
    }));
}

function decodeId(id) {
    const decoded = decode(decodeKey(id));
    return decoded["id"];
}

function encodeData(data) {
    return encode({
        "data": data,
        "time": Date.now(),
    });
}

function decodeData(data) {
    const decoded = decode(data);
    return decoded["data"];
}

let instance = null;

export class TA {
    private appId: string;
    private worker: any;
    private receiver: any;

    constructor(appId: string, token: string) {
        this.appId = appId;
        this.worker = null;
        this.receiver = null;
    }

    userSet(userId: string, properties: any) {
        const data = {
            "#type": "user_set",
            "#time": util.formatDate(new Date()),
            "#account_id": userId,
        };
        util.each(properties, function (val, key) {
            data[key] = val;
        });
        this.upload(data);
    }

    userSetOnce(userId: string, properties: any) {
        const data = {
            "#type": "user_setOnce",
            "#time": util.formatDate(new Date()),
            "#account_id": userId,
        };
        util.each(properties, function (val, key) {
            data[key] = val;
        });
        this.upload(data);
    }

    userAdd(userId: string, properties: any) {
        const data = {
            "#type": "user_add",
            "#time": util.formatDate(new Date()),
            "#account_id": userId,
        };
        util.each(properties, function (val, key) {
            data[key] = val;
        });
        this.upload(data);
    }

    userUnset(userId: string, keys: string[]) {
        const data: any = {
            "#type": "user_unset",
            "#time": util.formatDate(new Date()),
            "#account_id": userId,
        };
        for (let i = 0; i < keys.length; i++) {
            data[keys[i]] = 0;
        }
        this.upload(data);
    }

    userDelete(userId: string) {
        const data = {
            "#type": "user_del",
            "#time": util.formatDate(new Date()),
            "#account_id": userId,
        };
        this.upload(data);
    }

    track(userId: string, eventName: string, properties: any) {
        const data = {
            "#type": "track",
            "#time": util.formatDate(new Date()),
            "#account_id": userId,
            "#event_name": eventName,
        };
        util.each(properties, function (val, key) {
            if (!KEY_NAME_MATCH_REGEX.test(key)) {
                return;
            }
            data[key] = val;
        });
        this.upload(data);
    }

    upload(data) {
        if (!instance) {
            taLogger.info("ta data:", JSON.stringify(data));
        } else {
            taLogger.info("ta data:", JSON.stringify(data));
        }
    }
}

let ta: TA = null;
let initCount = 0;

export function initTa(appId: string, token: string) {
    initCount++;
    if (ta == null || initCount <= 1) {
        ta = new TA(appId, token);
    }
    return ta;
}

export function getTa() {
    if (!ta) {
        console.error("ta未初始化");
        return null;
    }
    return ta;
}

export function reportTAEvent(uid: string, event: string, data: any) {
    const ta = getTa();
    if (ta) {
        ta.track(uid, event, data);
    }
}

export function reportTAUserSet(uid: string, data: any) {
    const ta = getTa();
    if (ta) {
        ta.userSet(uid, data);
    }
}

export function initTaLoggingMode(isInit: boolean) {
    if (isInit) {
        if (!instance) {
            instance = new TA("AABBCC", "");
        }
    }
}