80 lines
3.2 KiB
Python
80 lines
3.2 KiB
Python
# -*- coding:utf-8 -*-
|
|
# Copyright (c) 2020 Huawei Technologies Co.,Ltd.
|
|
#
|
|
# openGauss is licensed under Mulan PSL v2.
|
|
# You can use this software according to the terms
|
|
# and conditions of the Mulan PSL v2.
|
|
# You may obtain a copy of Mulan PSL v2 at:
|
|
#
|
|
# http://license.coscl.org.cn/MulanPSL2
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OF ANY KIND,
|
|
# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
# See the Mulan PSL v2 for more details.
|
|
# ----------------------------------------------------------------------------
|
|
import os
|
|
from gspylib.inspection.common import SharedFuncs
|
|
from gspylib.inspection.common.CheckItem import BaseItem
|
|
from gspylib.inspection.common.CheckResult import ResultStatus
|
|
|
|
|
|
class CheckNextvalInDefault(BaseItem):
|
|
def __init__(self):
|
|
super(CheckNextvalInDefault, self).__init__(self.__class__.__name__)
|
|
|
|
def doCheck(self):
|
|
sql1 = """select distinct rt.relname from PG_ATTRDEF ad,
|
|
(
|
|
select c.oid,c.relname from pg_class c, pgxc_class xc
|
|
where
|
|
c.oid = xc.pcrelid and
|
|
c.relkind = 'r' and
|
|
xc.pclocatortype = 'R'
|
|
) as rt(oid,relname)
|
|
where ad.adrelid = rt.oid
|
|
and ad.adsrc like '%nextval%';
|
|
"""
|
|
sql2 = """select relname from pg_class c, pg_namespace n
|
|
where relkind = 'S' and c.relnamespace = n.oid
|
|
and n.nspname like 'pg_temp%';
|
|
"""
|
|
sqldb = "select datname from pg_database;"
|
|
output = SharedFuncs.runSqlCmd(sqldb, self.user, "", self.port,
|
|
self.tmpPath, "postgres",
|
|
self.mpprcFile)
|
|
dbList = output.split("\n")
|
|
dbList.remove("template0")
|
|
result = ""
|
|
for db in dbList:
|
|
output1 = SharedFuncs.runSqlCmd(sql1, self.user, "", self.port,
|
|
self.tmpPath, db, self.mpprcFile)
|
|
tmptablist = []
|
|
if (output1):
|
|
for tab in output1.splitlines():
|
|
tmpsql = "select * from %s limit 1" % tab
|
|
tmpout = SharedFuncs.runSqlCmd(tmpsql, self.user, "",
|
|
self.port, self.tmpPath, db,
|
|
self.mpprcFile)
|
|
if (tmpout):
|
|
tmptablist.append(tab)
|
|
else:
|
|
pass
|
|
output2 = SharedFuncs.runSqlCmd(sql2, self.user, "", self.port,
|
|
self.tmpPath, db, self.mpprcFile)
|
|
if (output2):
|
|
for tab in output2.splitlines():
|
|
if (tab not in tmptablist):
|
|
tmptablist.append(tab)
|
|
if (tmptablist):
|
|
result += "%s:\n%s\n" % (db, "\n".join(tmptablist))
|
|
if (result):
|
|
self.result.val = "there is some default expression " \
|
|
"contains nextval(sequence):\n%s" % result
|
|
self.result.rst = ResultStatus.NG
|
|
else:
|
|
self.result.val = "no default expression " \
|
|
"contains nextval(sequence)"
|
|
self.result.rst = ResultStatus.OK
|