agrs-sequelize-sdk 1.2.89 → 1.2.91

package/models/TokenRotationState.js

@@ -0,0 +1,121 @@
+// models/TokenRotationState.js
+module.exports = (sequelize, DataTypes) => {
+  const TokenRotationState = sequelize.define(
+    "TokenRotationState",
+    {
+      id: {
+        type: DataTypes.INTEGER,
+        primaryKey: true,
+        autoIncrement: true,
+      },
+      tokenType: {
+        type: DataTypes.STRING,
+        allowNull: false,
+        unique: true,
+        defaultValue: "FACEBOOK_ACCESS_TOKEN",
+        comment: "Type of token being rotated (e.g., FACEBOOK_ACCESS_TOKEN)",
+      },
+      currentIndex: {
+        type: DataTypes.INTEGER,
+        allowNull: false,
+        defaultValue: 0,
+        comment: "Current index in the token rotation (0-based)",
+      },
+      totalTokens: {
+        type: DataTypes.INTEGER,
+        allowNull: false,
+        defaultValue: 10,
+        comment: "Total number of tokens in rotation",
+      },
+      lastUsedAt: {
+        type: DataTypes.DATE,
+        allowNull: true,
+        comment: "Timestamp of last token usage",
+      },
+      totalUsageCount: {
+        type: DataTypes.BIGINT,
+        allowNull: false,
+        defaultValue: 0,
+        comment: "Total number of times tokens have been used",
+      },
+      rotationStrategy: {
+        type: DataTypes.ENUM("round_robin", "least_used", "random"),
+        allowNull: false,
+        defaultValue: "round_robin",
+        comment: "Strategy for token rotation",
+      },
+      metadata: {
+        type: DataTypes.JSONB,
+        allowNull: true,
+        comment: "Additional metadata (per-token usage stats, health, etc.)",
+      },
+    },
+    {
+      tableName: "TokenRotationState",
+      timestamps: true, // Adds createdAt and updatedAt
+      indexes: [
+        {
+          fields: ["tokenType"],
+          unique: true,
+          name: "idx_token_rotation_type",
+        },
+        {
+          fields: ["lastUsedAt"],
+          name: "idx_token_rotation_last_used",
+        },
+      ],
+    }
+  );
+
+  // Instance methods
+  TokenRotationState.prototype.incrementUsage = async function () {
+    this.totalUsageCount = (
+      BigInt(this.totalUsageCount) + BigInt(1)
+    ).toString();
+    this.lastUsedAt = new Date();
+    await this.save();
+  };
+
+  TokenRotationState.prototype.rotateToNext = async function () {
+    this.currentIndex = (this.currentIndex + 1) % this.totalTokens;
+    await this.save();
+  };
+
+  TokenRotationState.prototype.getCurrentTokenNumber = function () {
+    return this.currentIndex + 1; // Return 1-based index
+  };
+
+  // Static methods
+  TokenRotationState.getOrCreate = async function (
+    tokenType = "FACEBOOK_ACCESS_TOKEN"
+  ) {
+    const [state, created] = await this.findOrCreate({
+      where: { tokenType },
+      defaults: {
+        tokenType,
+        currentIndex: 0,
+        totalTokens: 10,
+        rotationStrategy: "round_robin",
+      },
+    });
+    return state;
+  };
+
+  TokenRotationState.getStats = async function (
+    tokenType = "FACEBOOK_ACCESS_TOKEN"
+  ) {
+    const state = await this.findOne({ where: { tokenType } });
+    if (!state) return null;
+
+    return {
+      currentTokenNumber: state.getCurrentTokenNumber(),
+      totalTokens: state.totalTokens,
+      totalUsageCount: state.totalUsageCount.toString(),
+      lastUsedAt: state.lastUsedAt,
+      rotationStrategy: state.rotationStrategy,
+      metadata: state.metadata,
+    };
+  };
+
+  return TokenRotationState;
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agrs-sequelize-sdk",
-  "version": "1.2.89",
+  "version": "1.2.91",
   "main": "index.js",
   "scripts": {
     "start": "node index.js",