¿Cómo puedo limitar el tiempo de consulta de la base de datos durante las solicitudes web?

Question

Tenemos una aplicación django bastante típica ejecutándose en postgresql 9.0. Recientemente descubrimos algunas consultas de DNS que se han ejecutado durante más de 4 horas, debido a búsquedas ineficientes en la interfaz de administración. Si bien planeamos solucionar estas consultas, como medida de seguridad nos gustaría restringir artificialmente el tiempo de consulta de la base de datos a 15 segundos, pero solo en el contexto de una solicitud web; los trabajos por lotes y las tareas de apio no deben estar limitados por esta restricción.

¿Cómo podemos hacer eso? ¿O es una idea terrible?

Answer 1

La mejor manera de hacer esto sería la creación de un papel/usuario que sólo se utiliza para ejecutar las peticiones web, a continuación, la statement_timeout ese rol .

ALTER ROLE role_name SET statement_timeout = 15000 

Todas las demás funciones utilizarán la configuración global de statement_timeout (que está deshabilitada en una instalación común).

Answer 2

Por lo que el tiempo se va, usted podría pasar todas sus consultas a través de una clase que, en la creación de instancias, genera dos hilos: uno para la consulta, y uno para un temporizador. Si el temporizador llega a 15 segundos, entonces mata el hilo con la consulta.

En cuanto a averiguar si la consulta se crea una instancia de una solicitud web, no sé lo suficiente sobre Django para poder ayudarte. De manera simplista, diría que en su clase que maneja sus llamadas a la base de datos, un parámetro opcional para el constructor podría ser algo como context, que podría ser http en el caso de una solicitud web y "" para cualquier otra cosa.

Answer 3

Tendrá que manejar esto manualmente. Eso es verificar la regla de 15 segundos y eliminar las consultas que la violan.

Consulta pg_stat_activity y encuentra a los infractores y llama a pg_terminate_backend (procpid) para matar a los infractores.

Algo como esto en un bucle:

SELECT pg_terminate_backend(pg_stat_activity.procpid) 
FROM pg_stat_activity 
WHERE pg_stat_activity.datname = 'TARGET_DB' 
    AND usename = 'WEBUSERNAME' 
    AND (now()-query_start) > '00:00:15';