root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Merge branch '5.3.x'

This commit is contained in:
Arjen Poutsma 2022-08-30 12:18:51 +02:00
parent 0ebc86c1af 4c0ece944a
commit b5b51e73fc
1 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
spring-web/src/main/java/org/springframework/http/codec/multipart/PartGenerator.java
View File

@ -688,6 +688,16 @@ final class PartGenerator extends BaseSubscriber<MultipartParser.Token> {
@Override @Override
public void onComplete() { public void onComplete() {
this.completed = true; this.completed = true;
public void partComplete(boolean finalPart) {
State state = PartGenerator.this.state.get();
// writeComplete might have changed our state to IdleFileState
if (state != this) {
state.partComplete(finalPart);
}
else {
this.completed = true;
this.finalPart = finalPart;
}
} }
public void writeBuffer(DataBuffer dataBuffer) { public void writeBuffer(DataBuffer dataBuffer) {
@ -711,14 +721,16 @@ final class PartGenerator extends BaseSubscriber<MultipartParser.Token> {
private void writeComplete() { private void writeComplete() {
IdleFileState newState = new IdleFileState(this); IdleFileState newState = new IdleFileState(this);
if (this.completed) { if (this.disposed) {
newState.onComplete();
}
else if (this.disposed) {
newState.dispose(); newState.dispose();
} }
else if (changeState(this, newState)) { else if (changeState(this, newState)) {
requestToken(); if (this.completed) {
newState.partComplete(this.finalPart);
}
else {
requestToken();
}
} }
else { else {
MultipartUtils.closeChannel(this.channel); MultipartUtils.closeChannel(this.channel);