05 — QueryRunner 심화

1. acquire connection
2. execute query
3. release connection

const qr = AppDataSource.createQueryRunner();
//   ↑ 객체 생성 — 아직 풀에서 연결을 빌리지 않음
 
await qr.connect();
//   ↑ 풀에서 연결 한 개를 acquire — 이 시점부터 풀의 idle 카운트 -1
 
await qr.startTransaction('REPEATABLE READ');
//   ↑ 연결에 BEGIN 전송
 
try {
  await qr.manager.save(User, { /* ... */ });
  //   ↑ 같은 연결에서 INSERT
  await qr.manager.save(Order, { /* ... */ });
  //   ↑ 같은 연결에서 INSERT
 
  await qr.commitTransaction();
  //   ↑ 연결에 COMMIT 전송
} catch (err) {
  await qr.rollbackTransaction();
  //   ↑ 연결에 ROLLBACK 전송
  throw err;
} finally {
  await qr.release();
  //   ↑ 연결을 풀로 반환 — 이 시점부터 풀의 idle 카운트 +1
}

class PostgresQueryRunner {
  databaseConnection: pg.Client;     // ← 빌려온 물리 연결
  isReleased: boolean;               // ← release 후 다시 못 씀
  isTransactionActive: boolean;      // ← BEGIN 후 COMMIT/ROLLBACK 전까지 true
  manager: EntityManager;            // ← 이 QR에 묶인 EntityManager
  // ...
}

// ❌
const qr = ds.createQueryRunner();
await qr.connect();
const user = await qr.manager.findOne(User, { where: { id } });
if (!user) throw new NotFoundError(); // ← release 없음!
await qr.startTransaction();
try {
  // ...
} finally {
  await qr.release();
}

// ❌
const qr = ds.createQueryRunner();
await qr.connect();
validateInput(); // ← throw 가능. release 없음.
await qr.startTransaction();
try { /* ... */ }
finally { await qr.release(); }

// ❌ Floating promise
try {
  qr.manager.save(user); // ← await 없음. throw가 try 밖으로 샌다
  await qr.commitTransaction();
} finally {
  await qr.release();
}

// ❌
try { /* ... */ }
finally {
  qr.release(); // ← await 없음. Promise가 떠다님.
}
// 함수가 빨리 return하면 release가 실행되기 전에
// 다음 요청이 풀에서 빌리려다 timeout

// ❌ double release
try { /* ... */ }
catch (err) {
  await qr.release(); // ← 1번
  throw err;
} finally {
  await qr.release(); // ← 2번 — 에러 또는 미정의 동작
}

// ✅ release를 의식할 필요가 없다
await ds.transaction(async (manager) => {
  // ...
});

async function withQueryRunner<T>(
  ds: DataSource,
  fn: (qr: QueryRunner) => Promise<T>,
): Promise<T> {
  const qr = ds.createQueryRunner();
  await qr.connect();
  try {
    return await fn(qr);
  } finally {
    await qr.release();
  }
}
 
// 사용
await withQueryRunner(AppDataSource, async (qr) => {
  await qr.startTransaction();
  try {
    await qr.manager.save(/* ... */);
    await qr.commitTransaction();
  } catch (err) {
    await qr.rollbackTransaction();
    throw err;
  }
});

// using 문법 (ES proposal — Stage 3, TS 5.2+)
{
  await using qr = ds.createDisposableQueryRunner();
  // 블록 끝에서 자동으로 release
}

// .eslintrc
{
  "rules": {
    "no-restricted-syntax": [
      "error",
      {
        "selector": "CallExpression[callee.property.name='createQueryRunner']",
        "message": "직접 호출 금지 — withQueryRunner() 헬퍼를 사용하세요."
      }
    ]
  }
}

// 개발 환경에서만
let activeRunners = new WeakSet<QueryRunner>();
const originalCreate = AppDataSource.createQueryRunner.bind(AppDataSource);
AppDataSource.createQueryRunner = (...args) => {
  const qr = originalCreate(...args);
  activeRunners.add(qr);
  const stack = new Error().stack;
  setTimeout(() => {
    if (activeRunners.has(qr) && !qr.isReleased) {
      console.error('LEAK: QueryRunner not released after 30s', stack);
    }
  }, 30_000);
  return qr;
};

// ❌ 안티패턴 — manager를 콜백 밖으로 빼낸다
let savedManager: EntityManager;
await AppDataSource.transaction(async (manager) => {
  savedManager = manager;
});
// 콜백이 끝나면 release됨
await savedManager.save(user); // ← 이미 release된 연결을 쓰려 함 — 에러

await ds.transaction(async (manager) => {
  // 메인 트랜잭션 안에서 부분 롤백 가능한 savepoint
  await manager.queryRunner!.query('SAVEPOINT sp1');
 
  try {
    await manager.save(user);
    await manager.save(order);
  } catch (err) {
    await manager.queryRunner!.query('ROLLBACK TO SAVEPOINT sp1');
    // 메인 트랜잭션은 살아 있음
  }
 
  // 다른 작업
  await manager.save(auditLog);
});

-- PostgreSQL: idle in transaction 찾기
SELECT pid, application_name, state, query_start, NOW() - query_start AS duration, query
FROM pg_stat_activity
WHERE state = 'idle in transaction'
  AND NOW() - query_start > interval '1 minute'
ORDER BY query_start;

// Java — 같은 문제
try (Connection conn = ds.getConnection();
     PreparedStatement ps = conn.prepareStatement("...");
     ResultSet rs = ps.executeQuery()) {
    // ...
} // try-with-resources가 자동으로 close — Java 7+