Add replace_args method for rustc_errors::diagnostic::Diagnostic

Author: SLASHLogin

compiler/rustc_codegen_ssa/src/back/write.rs

@@ -1792,9 +1792,7 @@ impl SharedEmitterMain {
                     if let Some(code) = diag.code {
                         d.code(code);
                     }
-                    for (name, arg) in diag.args {
-                        d.set_arg(name, arg);
-                    }
+                    d.replace_args(diag.args);
                     handler.emit_diagnostic(&mut d);
                 }
                 Ok(SharedEmitterMessage::InlineAsmError(cookie, msg, level, source)) => {

compiler/rustc_errors/src/diagnostic.rs

@@ -940,6 +940,10 @@ impl Diagnostic {
         self
     }
 
+    pub fn replace_args(&mut self, args: FxHashMap<DiagnosticArgName<'static>, DiagnosticArgValue<'static>>) {
+        self.args = args;
+    }
+
     pub fn styled_message(&self) -> &[(DiagnosticMessage, Style)] {
         &self.message
     }