JavaScript 异步迭代器辅助资源池：异步流资源管理

            
async function* numberGenerator(max) {
  for (let i = 0; i <= max; i++) {
    await delay(100); // Simulate asynchronous operation
    yield i;
  }
}

function delay(ms) {
  return new Promise(resolve => setTimeout(resolve, ms));
}

(async () => {
  for await (const number of numberGenerator(5)) {
    console.log(number);
  }
})();

            

              
                Copy
              
            
          

            
class ResourcePool {
  constructor(resourceFactory, resourceDestroyer, poolSize) {
    this.resourceFactory = resourceFactory;
    this.resourceDestroyer = resourceDestroyer;
    this.poolSize = poolSize;
    this.availableResources = [];
    this.acquiredResources = new Set();
    this.waitingQueue = [];

    // Pre-populate the pool with initial resources
    for (let i = 0; i < poolSize; i++) {
      this.availableResources.push(resourceFactory());
    }
  }

  async acquire() {
    if (this.availableResources.length > 0) {
      const resource = this.availableResources.pop();
      this.acquiredResources.add(resource);
      return resource;
    } else {
      return new Promise(resolve => {
        this.waitingQueue.push(resolve);
      });
    }
  }

  release(resource) {
    if (this.acquiredResources.has(resource)) {
      this.acquiredResources.delete(resource);
      this.availableResources.push(resource);

      if (this.waitingQueue.length > 0) {
        const resolve = this.waitingQueue.shift();
        resolve(this.availableResources.pop());
      }
    } else {
      console.warn("Releasing a resource that wasn't acquired from this pool.");
    }
  }

  async destroy() {
    for (const resource of this.availableResources) {
      await this.resourceDestroyer(resource);
    }
    this.availableResources = [];

    for (const resource of this.acquiredResources) {
      await this.resourceDestroyer(resource);
    }
    this.acquiredResources.clear();
  }
}

// Example usage with a hypothetical database connection
async function createDatabaseConnection() {
  // Simulate creating a database connection
  await delay(50);
  return { id: Math.random(), status: 'connected' };
}

async function closeDatabaseConnection(connection) {
  // Simulate closing a database connection
  await delay(50);
  console.log(`Closing connection ${connection.id}`);
}

(async () => {
  const poolSize = 5;
  const dbPool = new ResourcePool(createDatabaseConnection, closeDatabaseConnection, poolSize);

  async function processData(data) {
    const connection = await dbPool.acquire();
    console.log(`Processing data ${data} with connection ${connection.id}`);
    await delay(100); // Simulate database operation
    dbPool.release(connection);
  }

  const dataToProcess = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
  const promises = dataToProcess.map(data => processData(data));
  await Promise.all(promises);

  await dbPool.destroy();
})();

            

              
                Copy
              
            
          

            
async function* processStream(dataStream, resourcePool) {
  for await (const data of dataStream) {
    const resource = await resourcePool.acquire();
    try {
      // Process the data using the acquired resource
      const result = await processData(data, resource);
      yield result;
    } finally {
      resourcePool.release(resource);
    }
  }
}

async function processData(data, resource) {
  // Simulate processing data with the resource
  await delay(50);
  return `Processed ${data} with resource ${resource.id}`;
}

(async () => {
    const poolSize = 3;
    const dbPool = new ResourcePool(createDatabaseConnection, closeDatabaseConnection, poolSize);

    async function* generateData() {
        for (let i = 1; i <= 10; i++) {
            await delay(20);
            yield i;
        }
    }

    const dataStream = generateData();
    const results = [];
    for await (const result of processStream(dataStream, dbPool)) {
        results.push(result);
        console.log(result);
    }

    await dbPool.destroy();
})();

            

              
                Copy
              
            
          

            
class ResourcePool {
  // ... (previous code) ...

  async acquire() {
    while (true) {
      if (this.availableResources.length > 0) {
        const resource = this.availableResources.pop();
        if (await this.isValidResource(resource)) {
          this.acquiredResources.add(resource);
          return resource;
        } else {
          console.warn("Invalid resource detected, destroying and acquiring a new one.");
          await this.resourceDestroyer(resource);
          // Attempt to acquire another resource (loop continues)
        }
      } else {
        return new Promise(resolve => {
          this.waitingQueue.push(resolve);
        });
      }
    }
  }

  async isValidResource(resource) {
    // Implement your resource validation logic here
    // For example, check if a database connection is still active
    try {
        // Simulate a check
        await delay(10);
        return true; // Assume valid for this example
    } catch (error) {
        console.error("Resource is invalid:", error);
        return false;
    }
  }

  // ... (rest of the code) ...
}

            

              
                Copy
              
            
          

            
class ResourcePool {
    // ... (previous code) ...

    async acquire(timeout = 5000) { // Default timeout of 5 seconds
        return new Promise((resolve, reject) => {
            let timeoutId;

            const acquireResource = () => {
                if (this.availableResources.length > 0) {
                    const resource = this.availableResources.pop();
                    this.acquiredResources.add(resource);
                    clearTimeout(timeoutId);
                    resolve(resource);
                } else {
                    // Resource not immediately available, try again after a short delay
                    setTimeout(acquireResource, 50);
                }
            };

            timeoutId = setTimeout(() => {
                reject(new Error("Timeout acquiring resource from pool."));
            }, timeout);

            acquireResource(); // Start trying to acquire immediately
        });
    }

    // ... (rest of the code) ...
}

(async () => {
    const poolSize = 2;
    const dbPool = new ResourcePool(createDatabaseConnection, closeDatabaseConnection, poolSize);

    try {
        const connection = await dbPool.acquire(2000); // Acquire with a 2-second timeout
        console.log("Acquired connection:", connection.id);
        dbPool.release(connection);
    } catch (error) {
        console.error("Error acquiring connection:", error.message);
    }

    await dbPool.destroy();
})();

            

              
                Copy