sql >> Base de Datos >  >> RDS >> PostgreSQL

Cómo encontrar las primeras horas de inicio gratuitas de las reservas en Postgres

Esquema adaptado

CREATE EXTENSION btree_gist;
CREATE TYPE timerange AS RANGE (subtype = time);  -- create type once

-- Workers
CREATE TABLE worker(
   worker_id serial PRIMARY KEY
 , worker text NOT NULL
);
INSERT INTO worker(worker) VALUES ('JOHN'), ('MARY');

-- Holidays
CREATE TABLE pyha(pyha date PRIMARY KEY);

-- Reservations
CREATE TABLE reservat (
   reservat_id serial PRIMARY KEY
 , worker_id   int NOT NULL REFERENCES worker ON UPDATE CASCADE
 , day         date NOT NULL CHECK (EXTRACT('isodow' FROM day) < 7)
 , work_from   time NOT NULL -- including lower bound
 , work_to     time NOT NULL -- excluding upper bound
 , CHECK (work_from >= '10:00' AND work_to <= '21:00'
      AND work_to - work_from BETWEEN interval '15 min' AND interval '4 h'
      AND EXTRACT('minute' FROM work_from) IN (0, 15, 30, 45)
      AND EXTRACT('minute' FROM work_from) IN (0, 15, 30, 45)
    )
 , EXCLUDE USING gist (worker_id WITH =, day WITH =
                     , timerange(work_from, work_to) WITH &&)
);
INSERT INTO reservat (worker_id, day, work_from, work_to) VALUES 
   (1, '2014-10-28', '10:00', '11:30')  -- JOHN
 , (2, '2014-10-28', '11:30', '13:00'); -- MARY

-- Trigger for volatile checks
CREATE OR REPLACE FUNCTION holiday_check()
  RETURNS trigger AS
$func$
BEGIN
   IF EXISTS (SELECT 1 FROM pyha WHERE pyha = NEW.day) THEN
      RAISE EXCEPTION 'public holiday: %', NEW.day;
   ELSIF NEW.day < now()::date OR NEW.day > now()::date + 31 THEN
      RAISE EXCEPTION 'day out of range: %', NEW.day;
   END IF;

   RETURN NEW;
END
$func$ LANGUAGE plpgsql STABLE; -- can be "STABLE"

CREATE TRIGGER insupbef_holiday_check
BEFORE INSERT OR UPDATE ON reservat
FOR EACH ROW EXECUTE PROCEDURE holiday_check();

Puntos principales

  • No uses char(n) . Más bien varchar(n) , o mejor aún, varchar o simplemente text .

  • No utilice el nombre de un trabajador como clave principal. No es necesariamente único y puede cambiar. Use una clave principal sustituta en su lugar, mejor una serial . También realiza entradas en reservat más pequeños, índices más pequeños, consultas más rápidas, ...

  • Actualización: Para un almacenamiento más económico (8 bytes en lugar de 22) y un manejo más simple, guardo el inicio y el final como time ahora y construya un rango sobre la marcha para la restricción de exclusión:

    EXCLUDE USING gist (worker_id WITH =, day WITH =
                  , timerange(work_from, work_to) WITH &&)

  • Dado que sus rangos nunca pueden cruzar el límite de fechas por definición, sería más eficiente tener una date separada columna (day en mi implementación) y un intervalo de tiempo . El tipo timerange no se incluye en las instalaciones predeterminadas, pero se crea fácilmente. De esta manera, puede simplificar en gran medida sus restricciones de verificación.

  • Utilice EXTRACT('isodow', ...) para simplificar excluyendo los domingos

  • Supongo que desea permitir el borde superior de '21:00'.

  • Se supone que los bordes incluyen el límite inferior y excluyen el límite superior.

  • La verificación de si los días nuevos/actualizados se encuentran dentro de un mes a partir de "ahora" no es IMMUTABLE . Lo movió de CHECK restricción al activador; de lo contrario, podría tener problemas con el volcado/restauración. Detalles:

Aparte
Además de simplificar la entrada y verificar las restricciones, esperaba timerange para ahorrar 8 bytes de almacenamiento en comparación con tsrange desde time solo ocupa 4 bytes. Pero resulta que timerange ocupa 22 bytes en disco (25 en RAM), al igual que tsrange (o tstzrange ). Así que podrías ir con tsrange también. El principio de consulta y la restricción de exclusión son los mismos.

Consulta

Envuelto en una función SQL para un manejo conveniente de parámetros:

CREATE OR REPLACE FUNCTION f_next_free(_start timestamp, _duration interval)
  RETURNS TABLE (worker_id int, worker text, day date
               , start_time time, end_time time) AS
$func$
   SELECT w.worker_id, w.worker
        , d.d AS day
        , t.t AS start_time
        ,(t.t + _duration) AS end_time
   FROM  (
      SELECT _start::date + i AS d
      FROM   generate_series(0, 31) i
      LEFT   JOIN pyha p ON p.pyha = _start::date + i
      WHERE  p.pyha IS NULL   -- eliminate holidays
      ) d
   CROSS  JOIN (
      SELECT t::time
      FROM   generate_series (timestamp '2000-1-1 10:00'
                            , timestamp '2000-1-1 21:00' - _duration
                            , interval '15 min') t
      ) t  -- times
   CROSS  JOIN worker w
   WHERE  d.d + t.t > _start  -- rule out past timestamps
   AND    NOT EXISTS (
      SELECT 1
      FROM   reservat r
      WHERE  r.worker_id = w.worker_id
      AND    r.day = d.d
      AND    timerange(r.work_from, r.work_to) && timerange(t.t, t.t + _duration)
      )
   ORDER  BY d.d, t.t, w.worker, w.worker_id
   LIMIT  30  -- could also be parameterized
$func$ LANGUAGE sql STABLE;

Llamar:

SELECT * FROM f_next_free('2014-10-28 12:00'::timestamp, '1.5 h'::interval);

SQL Fiddle en Postgres 9.3 ahora.

Explicar

  • La función toma un _start timestamp como hora mínima de inicio e _duration interval . Tenga cuidado de descartar solo tiempos anteriores en el inicio día, no los días siguientes. Más simple simplemente agregando día y hora:t + d > _start .
    Para reservar una reserva a partir de "ahora", simplemente ingrese now()::timestamp :

    SELECT * FROM f_next_free(`now()::timestamp`, '1.5 h'::interval);

  • Subconsulta d genera días a partir del valor de entrada _day . Días festivos excluidos.

  • Los días se combinan con posibles rangos de tiempo generados en la subconsulta t .
  • Eso se une a todos los trabajadores disponibles w .
  • Finalmente elimine todos los candidatos que colisionen con las reservas existentes usando un NOT EXISTS anti-semi-join, y en particular el operador de superposiciones && .

Relacionado: