prisma-sharding 0.0.2 → 0.0.4

package/dist/index.mjs

@@ -1,22 +1,23 @@
-// src/constants/index.ts
-var DEFAULTS = {
-  POOL_MAX_CONNECTIONS: 10,
-  POOL_IDLE_TIMEOUT_MS: 1e4,
-  POOL_CONNECTION_TIMEOUT_MS: 5e3,
-  HEALTH_CHECK_INTERVAL_MS: 3e4,
-  CIRCUIT_BREAKER_THRESHOLD: 3,
-  CONSISTENT_HASH_VIRTUAL_NODES: 150
-};
-var ERROR_MESSAGES = {
-  NO_SHARDS: "At least one shard must be configured",
-  SHARD_NOT_FOUND: (id) => `Shard "${id}" not found`,
-  NO_HEALTHY_SHARDS: "No healthy shards available",
-  INVALID_STRATEGY: (s) => `Invalid routing strategy: "${s}". Use "modulo" or "consistent-hash"`,
-  NOT_CONNECTED: "Sharding not connected. Call connect() first",
-  ALREADY_CONNECTED: "Sharding already connected",
-  MISSING_CLIENT_FACTORY: "createClient function is required",
-  INVALID_SHARD_URL: (id) => `Invalid or missing URL for shard "${id}"`
-};
+// src/utils/index.ts
+function hashString(str) {
+  let hash = 0;
+  for (let i = 0; i < str.length; i++) {
+    const char = str.charCodeAt(i);
+    hash = (hash << 5) - hash + char;
+    hash = hash & hash;
+  }
+  return Math.abs(hash);
+}
+function validateUrl(url) {
+  return url.startsWith("postgresql://") || url.startsWith("postgres://");
+}
+function createDefaultLogger() {
+  return {
+    info: (msg) => console.log(`[PrismaSharding] ${msg}`),
+    warn: (msg) => console.warn(`[PrismaSharding] ${msg}`),
+    error: (msg) => console.error(`[PrismaSharding] ${msg}`)
+  };
+}
 
 // src/core/errors.ts
 var ShardingError = class _ShardingError extends Error {
@@ -50,6 +51,87 @@ var RoutingError = class _RoutingError extends ShardingError {
   }
 };
 
+// src/constants/index.ts
+var DEFAULTS = {
+  POOL_MAX_CONNECTIONS: 10,
+  POOL_IDLE_TIMEOUT_MS: 1e4,
+  POOL_CONNECTION_TIMEOUT_MS: 5e3,
+  HEALTH_CHECK_INTERVAL_MS: 3e4,
+  CIRCUIT_BREAKER_THRESHOLD: 3,
+  CONSISTENT_HASH_VIRTUAL_NODES: 150
+};
+var ERROR_MESSAGES = {
+  NO_SHARDS: "At least one shard must be configured",
+  SHARD_NOT_FOUND: (id) => `Shard "${id}" not found`,
+  NO_HEALTHY_SHARDS: "No healthy shards available",
+  INVALID_STRATEGY: (s) => `Invalid routing strategy: "${s}". Use "modulo" or "consistent-hash"`,
+  NOT_CONNECTED: "Sharding not connected. Call connect() first",
+  ALREADY_CONNECTED: "Sharding already connected",
+  MISSING_CLIENT_FACTORY: "createClient function is required",
+  INVALID_SHARD_URL: (id) => `Invalid or missing URL for shard "${id}"`
+};
+
+// src/core/router.ts
+var ShardRouter = class {
+  constructor(config) {
+    this.consistentHashRing = /* @__PURE__ */ new Map();
+    this.virtualNodes = DEFAULTS.CONSISTENT_HASH_VIRTUAL_NODES;
+    this.strategy = config.strategy;
+    this.shardIds = config.shardIds;
+    this.logger = config.logger;
+    if (this.strategy === "consistent-hash") {
+      this.initializeConsistentHashRing();
+    }
+  }
+  initializeConsistentHashRing() {
+    for (const shardId of this.shardIds) {
+      for (let i = 0; i < this.virtualNodes; i++) {
+        const hash = hashString(`${shardId}:${i}`);
+        this.consistentHashRing.set(hash, shardId);
+      }
+    }
+  }
+  getShardIndex(key) {
+    const shardCount = this.shardIds.length;
+    if (shardCount === 0) {
+      throw new RoutingError("No shards available");
+    }
+    if (this.strategy === "consistent-hash") {
+      return this.getIndexConsistentHash(key);
+    }
+    return this.getIndexModulo(key, shardCount);
+  }
+  getIndexModulo(key, shardCount) {
+    const hash = hashString(key);
+    return hash % shardCount;
+  }
+  getIndexConsistentHash(key) {
+    const hash = hashString(key);
+    const sortedHashes = Array.from(this.consistentHashRing.keys()).sort((a, b) => a - b);
+    for (const ringHash of sortedHashes) {
+      if (hash <= ringHash) {
+        const shardId = this.consistentHashRing.get(ringHash);
+        const match2 = shardId.match(/shard_(\d+)/);
+        return match2 ? parseInt(match2[1], 10) - 1 : 0;
+      }
+    }
+    const firstShardId = this.consistentHashRing.get(sortedHashes[0]);
+    const match = firstShardId.match(/shard_(\d+)/);
+    return match ? parseInt(match[1], 10) - 1 : 0;
+  }
+  getShardId(key) {
+    const index = this.getShardIndex(key);
+    return this.shardIds[index] || `shard_${index + 1}`;
+  }
+  getRandomShardIndex() {
+    return Math.floor(Math.random() * this.shardIds.length);
+  }
+  getRandomShardId() {
+    const index = this.getRandomShardIndex();
+    return this.shardIds[index];
+  }
+};
+
 // src/core/manager.ts
 var ShardManager = class {
   constructor(config) {
@@ -217,88 +299,6 @@ var ShardManager = class {
   }
 };
 
-// src/utils/index.ts
-function hashString(str) {
-  let hash = 0;
-  for (let i = 0; i < str.length; i++) {
-    const char = str.charCodeAt(i);
-    hash = (hash << 5) - hash + char;
-    hash = hash & hash;
-  }
-  return Math.abs(hash);
-}
-function validateUrl(url) {
-  return url.startsWith("postgresql://") || url.startsWith("postgres://");
-}
-function createDefaultLogger() {
-  return {
-    info: (msg) => console.log(`[PrismaSharding] ${msg}`),
-    warn: (msg) => console.warn(`[PrismaSharding] ${msg}`),
-    error: (msg) => console.error(`[PrismaSharding] ${msg}`)
-  };
-}
-
-// src/core/router.ts
-var ShardRouter = class {
-  constructor(config) {
-    this.consistentHashRing = /* @__PURE__ */ new Map();
-    this.virtualNodes = DEFAULTS.CONSISTENT_HASH_VIRTUAL_NODES;
-    this.strategy = config.strategy;
-    this.shardIds = config.shardIds;
-    this.logger = config.logger;
-    if (this.strategy === "consistent-hash") {
-      this.initializeConsistentHashRing();
-    }
-  }
-  initializeConsistentHashRing() {
-    for (const shardId of this.shardIds) {
-      for (let i = 0; i < this.virtualNodes; i++) {
-        const hash = hashString(`${shardId}:${i}`);
-        this.consistentHashRing.set(hash, shardId);
-      }
-    }
-  }
-  getShardIndex(key) {
-    const shardCount = this.shardIds.length;
-    if (shardCount === 0) {
-      throw new RoutingError("No shards available");
-    }
-    if (this.strategy === "consistent-hash") {
-      return this.getIndexConsistentHash(key);
-    }
-    return this.getIndexModulo(key, shardCount);
-  }
-  getIndexModulo(key, shardCount) {
-    const hash = hashString(key);
-    return hash % shardCount;
-  }
-  getIndexConsistentHash(key) {
-    const hash = hashString(key);
-    const sortedHashes = Array.from(this.consistentHashRing.keys()).sort((a, b) => a - b);
-    for (const ringHash of sortedHashes) {
-      if (hash <= ringHash) {
-        const shardId = this.consistentHashRing.get(ringHash);
-        const match2 = shardId.match(/shard_(\d+)/);
-        return match2 ? parseInt(match2[1], 10) - 1 : 0;
-      }
-    }
-    const firstShardId = this.consistentHashRing.get(sortedHashes[0]);
-    const match = firstShardId.match(/shard_(\d+)/);
-    return match ? parseInt(match[1], 10) - 1 : 0;
-  }
-  getShardId(key) {
-    const index = this.getShardIndex(key);
-    return this.shardIds[index] || `shard_${index + 1}`;
-  }
-  getRandomShardIndex() {
-    return Math.floor(Math.random() * this.shardIds.length);
-  }
-  getRandomShardId() {
-    const index = this.getRandomShardIndex();
-    return this.shardIds[index];
-  }
-};
-
 // src/core/sharding.ts
 var PrismaSharding = class {
   constructor(config) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prisma-sharding",
-  "version": "0.0.2",
+  "version": "0.0.4",
   "description": "Lightweight database sharding library for Prisma with connection pooling and health monitoring",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -13,9 +13,16 @@
     "sharding",
     "database",
     "postgresql",
+    "db-sharding",
     "horizontal-scaling",
-    "connection-pool"
+    "connection-pool",
+    "cli"
   ],
+  "bin": {
+    "prisma-sharding-migrate": "./dist/cli/migrate.js",
+    "prisma-sharding-studio": "./dist/cli/studio.js",
+    "prisma-sharding-test": "./dist/cli/test.js"
+  },
   "files": [
     "dist"
   ],
@@ -27,7 +34,9 @@
     }
   },
   "scripts": {
-    "build": "tsup src/index.ts --format cjs,esm --dts --clean",
+    "build": "tsup src/index.ts --format cjs,esm --dts --clean && tsup src/cli/migrate.ts src/cli/studio.ts src/cli/test.ts --format cjs --outDir dist/cli --clean",
+    "build:lib": "tsup src/index.ts --format cjs,esm --dts --clean",
+    "build:cli": "tsup src/cli/migrate.ts src/cli/studio.ts src/cli/test.ts --format cjs --outDir dist/cli",
     "dev": "tsup src/index.ts --format cjs,esm --dts --watch",
     "lint": "eslint src --ext .ts",
     "typecheck": "tsc --noEmit",
@@ -40,8 +49,13 @@
   "peerDependencies": {
     "@prisma/client": ">=5.0.0"
   },
+  "dependencies": {
+    "dotenv": "^16.3.1",
+    "pg": "^8.18.0"
+  },
   "devDependencies": {
     "@types/node": "^20.10.0",
+    "@types/pg": "^8.16.0",
     "@typescript-eslint/eslint-plugin": "^6.13.0",
     "@typescript-eslint/parser": "^6.13.0",
     "eslint": "^8.55.0",