mirror of
https://github.com/mongodb/mongo.git
synced 2024-11-29 08:42:18 +01:00
159 lines
4.5 KiB
Python
Executable File
159 lines
4.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
import re
|
|
import utils
|
|
|
|
|
|
assertNames = [ "uassert" , "massert" ]
|
|
|
|
def assignErrorCodes():
|
|
cur = 10000
|
|
for root in assertNames:
|
|
for x in utils.getAllSourceFiles():
|
|
print( x )
|
|
didAnything = False
|
|
fixed = ""
|
|
for line in open( x ):
|
|
s = line.partition( root + "(" )
|
|
if s[1] == "" or line.startswith( "#define " + root):
|
|
fixed += line
|
|
continue
|
|
fixed += s[0] + root + "( " + str( cur ) + " , " + s[2]
|
|
cur = cur + 1
|
|
didAnything = True
|
|
if didAnything:
|
|
out = open( x , 'w' )
|
|
out.write( fixed )
|
|
out.close()
|
|
|
|
|
|
codes = []
|
|
|
|
def readErrorCodes( callback, replaceZero = False ):
|
|
ps = [ re.compile( "(([umsg]asser(t|ted))) *\(( *)(\d+)" ) ,
|
|
re.compile( "((User|Msg|MsgAssertion)Exceptio(n))\(( *)(\d+)" ) ,
|
|
re.compile( "(((verify))) *\(( *)(\d+)" )
|
|
]
|
|
|
|
for x in utils.getAllSourceFiles():
|
|
|
|
needReplace = [False]
|
|
lines = []
|
|
lastCodes = [0]
|
|
lineNum = 1
|
|
|
|
for line in open( x ):
|
|
|
|
for p in ps:
|
|
|
|
def repl( m ):
|
|
m = m.groups()
|
|
|
|
start = m[0]
|
|
spaces = m[3]
|
|
code = m[4]
|
|
if code == '0' and replaceZero :
|
|
code = getNextCode( lastCodes )
|
|
lastCodes.append( code )
|
|
code = str( code )
|
|
needReplace[0] = True
|
|
|
|
print( "Adding code " + code + " to line " + x + ":" + str( lineNum ) )
|
|
|
|
else :
|
|
codes.append( ( x , lineNum , line , code ) )
|
|
callback( x , lineNum , line , code )
|
|
|
|
return start + "(" + spaces + code
|
|
|
|
line = re.sub( p, repl, line )
|
|
|
|
if replaceZero : lines.append( line )
|
|
lineNum = lineNum + 1
|
|
|
|
if replaceZero and needReplace[0] :
|
|
print( "Replacing file " + x )
|
|
of = open( x + ".tmp", 'w' )
|
|
of.write( "".join( lines ) )
|
|
of.close()
|
|
os.remove(x)
|
|
os.rename( x + ".tmp", x )
|
|
|
|
|
|
def getNextCode( lastCodes = [0] ):
|
|
highest = [max(lastCodes)]
|
|
def check( fileName , lineNum , line , code ):
|
|
code = int( code )
|
|
if code > highest[0]:
|
|
highest[0] = code
|
|
readErrorCodes( check )
|
|
return highest[0] + 1
|
|
|
|
def checkErrorCodes():
|
|
seen = {}
|
|
errors = []
|
|
def checkDups( fileName , lineNum , line , code ):
|
|
if code in seen:
|
|
print( "DUPLICATE IDS" )
|
|
print( "%s:%d:%s %s" % ( fileName , lineNum , line.strip() , code ) )
|
|
print( "%s:%d:%s %s" % seen[code] )
|
|
errors.append( seen[code] )
|
|
seen[code] = ( fileName , lineNum , line , code )
|
|
readErrorCodes( checkDups, True )
|
|
return len( errors ) == 0
|
|
|
|
def getBestMessage( err , start ):
|
|
err = err.partition( start )[2]
|
|
if not err:
|
|
return ""
|
|
err = err.partition( "\"" )[2]
|
|
if not err:
|
|
return ""
|
|
err = err.rpartition( "\"" )[0]
|
|
if not err:
|
|
return ""
|
|
return err
|
|
|
|
def genErrorOutput():
|
|
|
|
if os.path.exists( "docs/errors.md" ):
|
|
i = open( "docs/errors.md" , "r" )
|
|
|
|
|
|
out = open( "docs/errors.md" , 'wb' )
|
|
out.write( "MongoDB Error Codes\n==========\n\n\n" )
|
|
|
|
prev = ""
|
|
seen = {}
|
|
|
|
codes.sort( key=lambda x: x[0]+"-"+x[3] )
|
|
for f,l,line,num in codes:
|
|
if num in seen:
|
|
continue
|
|
seen[num] = True
|
|
|
|
if f.startswith( "./" ):
|
|
f = f[2:]
|
|
|
|
if f != prev:
|
|
out.write( "\n\n" )
|
|
out.write( f + "\n----\n" )
|
|
prev = f
|
|
|
|
url = "http://github.com/mongodb/mongo/blob/master/" + f + "#L" + str(l)
|
|
|
|
out.write( "* " + str(num) + " [code](" + url + ") " + getBestMessage( line , str(num) ) + "\n" )
|
|
|
|
out.write( "\n" )
|
|
out.close()
|
|
|
|
if __name__ == "__main__":
|
|
ok = checkErrorCodes()
|
|
print( "ok:" + str( ok ) )
|
|
print( "next: " + str( getNextCode() ) )
|
|
if ok:
|
|
genErrorOutput()
|
|
|