sql: small fixes to nameconcatoid

There was an edge case if the result was exactly 64 characters where the
input would not be truncated correctly. This also makes the
information_schema.role_routine_grants table use the same logic for
computing the specific_name.

Release note: None

Author: rafiss

pkg/sql/information_schema.go

@@ -1774,7 +1774,7 @@ var informationSchemaRoleRoutineGrantsTable = virtualSchemaTable{
 
 				_, overloads := builtinsregistry.GetBuiltinProperties(name)
 				for _, o := range overloads {
-					fnSpecificName := tree.NewDString(fmt.Sprintf("%s_%d", fnNameStr, o.Oid))
+					fnSpecificName := tree.NewDString(nameConcatOid(fnNameStr, o.Oid))
 					for _, grantee := range roleNameForBuiltins {
 						if err := addRow(
 							tree.DNull, // grantor
@@ -1816,7 +1816,8 @@ var informationSchemaRoleRoutineGrantsTable = virtualSchemaTable{
 						return err
 					}
 					scNameStr := tree.NewDString(sc.GetName())
-					fnSpecificName := tree.NewDString(fmt.Sprintf("%s_%d", fn.GetName(), catid.FuncIDToOID(fn.GetID())))
+
+					fnSpecificName := tree.NewDString(nameConcatOid(fn.GetName(), catid.FuncIDToOID(fn.GetID())))
 					fnName := tree.NewDString(fn.GetName())
 					for _, u := range privs {
 						userNameStr := tree.NewDString(u.User.Normalized())
@@ -2955,3 +2956,16 @@ func userCanSeeDescriptor(
 func descriptorIsVisible(desc catalog.Descriptor, allowAdding bool) bool {
 	return desc.Public() || (allowAdding && desc.Adding())
 }
+
+// nameConcatOid is a Go version of the nameconcatoid builtin function. The
+// result is the same as fmt.Sprintf("%s_%d", s, o) except that, if it would not
+// fit in 63 characters, we make it do so by truncating the name input (not the
+// oid).
+func nameConcatOid(s string, o oid.Oid) string {
+	const maxLen = 63
+	oidStr := strconv.Itoa(int(o))
+	if len(s)+1+len(oidStr) <= maxLen {
+		return s + "_" + oidStr
+	}
+	return s[:maxLen-1-len(oidStr)] + "_" + oidStr
+}

pkg/sql/logictest/testdata/logic_test/pg_builtins

@@ -882,4 +882,9 @@ select nameconcatoid(repeat('a', 58) || 'bbbbbbbbbb', 2);
 ----
 aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaabbb_2
 
+query TI
+select nameconcatoid(repeat('a', 62), 2), length(nameconcatoid(repeat('a', 62), 2))
+----
+aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa_2  63
+
 subtest end

pkg/sql/sem/builtins/pg_builtins.go

@@ -2191,7 +2191,7 @@ var pgBuiltins = map[string]builtinDefinition{
 			ReturnType: tree.FixedReturnType(types.Name),
 			Body: `
 SELECT
-  CASE WHEN length($1::text || '_' || $2::text) > 64
+  CASE WHEN length($1::text || '_' || $2::text) > 63
 	THEN (substring($1 from 1 for 63 - length($2::text) - 1) || '_' || $2::text)::name
 	ELSE ($1::text || '_' || $2::text)::name
 	END