Fixed nil panic

dolthub · Sep 12, 2024 · 968726d · 968726d
1 parent f47dfea
commit 968726d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/go/libraries/doltcore/sqle/dprocedures/dolt_reset.go b/go/libraries/doltcore/sqle/dprocedures/dolt_reset.go
@@ -91,21 +91,20 @@ func doDoltReset(ctx *sql.Context, args []string) (int, error) {
 			return 1, err
 		}
 	} else {
-		firstArg := apr.Arg(0)
-		if apr.NArg() != 1 || (apr.NArg() == 1 && firstArg == ".") {
+		if apr.NArg() != 1 || (apr.NArg() == 1 && apr.Arg(0) == ".") {
 			err := resetSoftTables(ctx, apr, roots, dSess, dbName)
 			if err != nil {
 				return 1, err
 			}
 		} else {
 			// check if the input is a table name or commit ref
-			if isTableInRoots(ctx, roots, firstArg) {
+			if isTableInRoots(ctx, roots, apr.Arg(0)) {
 				err := resetSoftTables(ctx, apr, roots, dSess, dbName)
 				if err != nil {
 					return 1, err
 				}
 			} else {
-				err := resetSoftToRef(ctx, dbData, firstArg, dSess, dbName)
+				err := resetSoftToRef(ctx, dbData, apr.Arg(0), dSess, dbName)
 				if err != nil {
 					return 1, err
 				}