Function extractStatement

backend/plugin/parser/tsql/restore.go:383–412

(statement string, backupItem *storepb.PriorBackupDetail_Item)

}

func extractStatement(statement string, backupItem *storepb.PriorBackupDetail_Item) (string, error) {
	if backupItem == nil {
		return "", errors.Errorf("backup item is nil")
	}

	parseResults, err := parseTSQLStatements(statement)
	if err != nil {
		return "", errors.Wrapf(err, "failed to parse statement")
	}

	var originalSQL []string
	for _, parsedStatement := range parseResults {
		node, ok := GetOmniNode(parsedStatement.AST)
		if !ok || node == nil || !isRestorableDML(node) {
			continue
		}
		start, end := statementPositions(parsedStatement.Start, parsedStatement.Text, dmlNodeLoc(node))
		if !inRange(start, end, backupItem.StartPosition, backupItem.EndPosition) {
			continue
		}
		sql := strings.TrimSuffix(sourceFromLoc(parsedStatement.Text, dmlNodeLoc(node)), ";")
		originalSQL = append(originalSQL, sql)
	}
	if len(originalSQL) == 0 {
		return "", nil
	}

	// Join with semicolons and add a trailing semicolon
	return strings.Join(originalSQL, ";\n") + ";", nil
}

func isRestorableDML(node ast.Node) bool {
	switch node.(type) {