PostgreSQL用户登录失败自动锁定的解决办法_postgresql the account has been locked(1)

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

session_preload_libraries='session_exec'
session_exec.login_name='login'

注意：上面第一个变量是设置session_preload_libraries而不是通常设置的shared_preload_libraries。
第二个变量是需要自定义实现的登录函数。

重启数据库服务。

$ sudo systemctl restart postgresql-12

二、自定义登录函数篇

创建t_login表用于存储提取自数据库日志中登录失败的信息。

create table t_login
(
login_time timestamp(3) with time zone --插入时间,
user_name text --数据库登录用户,
flag int4 --标志位，0代表过期数据，1代表正常状态数据
);

使用file_fdw外部表记录数据库日志信息。
file_fdw如果未配置过，参见下面步骤。

$ cd /opt/postgresql-12.5/contrib/file_fdw
$ make && make install

create extension file_fdw;
CREATE SERVER pglog FOREIGN DATA WRAPPER file_fdw;

建立外部表postgres_log，关联数据库日志中登录失败的信息。

CREATE FOREIGN TABLE postgres_log(  
  log_time timestamp(3) with time zone,  
  user_name text,  
  database_name text,  
  process_id integer,
  connection_from text,
  session_id text,  
  session_line_num bigint,  
  command_tag text,  
  session_start_time timestamp with time zone,  
  virtual_transaction_id text,  
  transaction_id bigint,  
  error_severity text,  
  sql_state_code text,  
  message text,  
  detail text,  
  hint text,  
  internal_query text,  
  internal_query_pos integer,  
  context text,  
  query text,  
  query_pos integer,  
  location text,  
  application_name text
) SERVER pglog  
OPTIONS ( program 'find /opt/pg_log_5432 -type f -name "*.csv" -mtime -1 -exec cat {} \;', format 'csv' );

注意：
1./opt/pg_log_5432需要修改为实际环境日志目录。
2. 不同PG版本csv日志格式可能有所差异，参考PG官网文档runtime-config-logging章节（http://postgres.cn/docs/12/runtime-config-logging.html）。

此时连接数据库因未创建登录函数会出现下面的警告信息。

$ psql -Upostgres
WARNING:  function "login()" does not exist
psql (12.5)
Type "help" for help.

创建登录函数login。

create or replace function login() returns void as $$
declare
res text;
c1 timestamp(3) with time zone;
begin

--获取当前日志中最新时间
select login_time 
from public.t_login 
where flag = 0 
order by login_time 
desc limit 1 
into c1; 

 --将最新的数据插入t_login表
insert into public.t_login  
select log_time,user_name 
from public.postgres_log 
where command_tag='authentication' 
and error_severity= 'FATAL' 
and log_time > c1;

update public.t_login set flag = 1 where login_time > c1; 

--检查登录失败次数是否大于3，若大于3则锁定用户
for res in select user_name from public.t_login where flag = 1 group by user_name having count(*) >=3 
loop
--锁定用户
EXECUTE format('alter user %I nologin',res); 
--断开当前被锁定用户会话
EXECUTE 'select pg_catalog.pg_terminate_backend(pid) from pg_catalog.pg_stat_activity where usename=$1' using res; 
raise notice 'Account % is locked!',res;
end loop;
end;
$$ language plpgsql strict security definer set search_path to 'public';

测试使用篇

创建测试用户。

create user test1 encrypted password 'XXX';

模拟test1用户登录失败，输入错误密码。

$ psql -h192.168.137.11 -Utest1 postgres
Password for user test1: 
psql: error: FATAL:  password authentication failed for user "test1"

通过外部表查看登录失败的日志。

select * from postgres_log where command_tag='authentication' and error_severity= 'FATAL';

可以看到1条数据，手工插入一条登录失败的信息到t_login表。

insert into t_login  select log_time,user_name,0
    from postgres_log 
    where command_tag='authentication' 
    and error_severity= 'FATAL';

参考上面登录失败测试，接着再测试2次。

然后使用postgres用户登录数据库，观察t_login表数据。

postgres=# select * from t_login;
       login_time        | user_name | flag 
-------------------------+-----------+------
 2021-02-08 06:24:47.101 | test1     |    0
 2021-02-08 06:25:16.581 | test1     |    1
 2021-02-08 06:25:18.429 | test1     |    1


![img](https://img-blog.csdnimg.cn/img_convert/bb278c7eb6abbc51bce618422f97b403.png)
![img](https://img-blog.csdnimg.cn/img_convert/561f7a2d93df6fd9db7cc4d72c47401f.png)
![img](https://img-blog.csdnimg.cn/img_convert/ba6bd4a5621cf04e6b711786608d2265.png)

**既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上软件测试知识点，真正体系化！**

**由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新**

**[需要这份系统化的资料的朋友，可以戳这里获取](https://bbs.csdn.net/topics/618631832)**

6606211)]

**既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上软件测试知识点，真正体系化！**

**由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新**

**[需要这份系统化的资料的朋友，可以戳这里获取](https://bbs.csdn.net/topics/618631832)**