← Wszystkie zadaniaNode.js · Senior

Circuit breaker

Gdy zależna usługa leży, każde żądanie do niej czeka na timeout i zajmuje zasoby. Napisz createBreaker(fn, { failureThreshold, resetMs, now }) z metodami call(...args) i state().

- closed: wywołania idą do fn; po failureThreshold błędach z rzędu bezpiecznik przechodzi w open (sukces zeruje licznik). - open: call od razu odrzuca z Circuit open, a fn nie jest wywoływane. - Po resetMs od otwarcia stan to half-open: przechodzi jedno wywołanie próbne (pozostałe równoległe dostają Circuit open). Sukces zamyka bezpiecznik, błąd znowu otwiera go na resetMs.

function createBreaker(fn, { failureThreshold, resetMs, now = Date.now }) {
  return {
    async call(...args) {
      // ...
      return fn(...args);
    },
    state() {
      return "closed";
    },
  };
}

Ctrl+Enter uruchamia testy. Tab robi wcięcie; aby wyjść z edytora klawiaturą, naciśnij Esc, a potem Tab. Kod zapisuje się w tej przeglądarce.

Testy (6)

let t = 0;
const b = createBreaker(async (x) => x * 2, { failureThreshold: 3, resetMs: 1000, now: () => t });
expect(await b.call(21)).toBe(42);
expect(b.state()).toBe('closed');

let t = 0;
const f = fn(async () => { throw new Error('timeout'); });
const b = createBreaker(f, { failureThreshold: 3, resetMs: 1000, now: () => t });
for (let i = 0; i < 3; i++) await expect(b.call()).rejects.toThrow('timeout');
expect(b.state()).toBe('open');
await expect(b.call()).rejects.toThrow('Circuit open');
expect(f).toHaveBeenCalledTimes(3);

let t = 0;
let fail = true;
const b = createBreaker(async () => { if (fail) throw new Error('down'); return 'ok'; }, { failureThreshold: 2, resetMs: 1000, now: () => t });
await expect(b.call()).rejects.toThrow('down');
await expect(b.call()).rejects.toThrow('down');
t = 999;
expect(b.state()).toBe('open');
t = 1000;
expect(b.state()).toBe('half-open');
fail = false;
expect(await b.call()).toBe('ok');
expect(b.state()).toBe('closed');

let t = 0;
const b = createBreaker(async () => { throw new Error('down'); }, { failureThreshold: 1, resetMs: 500, now: () => t });
await expect(b.call()).rejects.toThrow('down');
t = 500;
await expect(b.call()).rejects.toThrow('down');
expect(b.state()).toBe('open');
t = 900;
await expect(b.call()).rejects.toThrow('Circuit open');

let t = 0;
let ok = false;
const b = createBreaker(async () => { if (!ok) throw new Error('down'); await wait(30); return 'ok'; }, { failureThreshold: 1, resetMs: 100, now: () => t });
await expect(b.call()).rejects.toThrow('down');
t = 100;
ok = true;
const first = b.call();
await expect(b.call()).rejects.toThrow('Circuit open');
expect(await first).toBe('ok');

let t = 0;
let n = 0;
const b = createBreaker(async () => { n++; if (n % 2) throw new Error('flaky'); return n; }, { failureThreshold: 2, resetMs: 1000, now: () => t });
await expect(b.call()).rejects.toThrow('flaky');
expect(await b.call()).toBe(2);
await expect(b.call()).rejects.toThrow('flaky');
expect(b.state()).toBe('closed');