在某些業務場景中,需要判斷獲取到的InputStream是否為空。
InputStream沒有接口去判斷是否為空或者獲取其大小,本文會列出項目中見到的一些解決方案。
輸入流可用
有些同學直接把InputStream.available當成流的實際大小,比如下載的時候就直接把InputStream.available設置為Content-Length,這個是很大的錯誤,available方法注釋里明確寫到:
返回可從此輸入流讀取(或跳過)的字節數估計值estimate是什么意思?怎么讀,而不會被下次調用此輸入流的方法阻塞。
估計是一個大概的估計,并不代表流的實際大小,如果是FileInputStream的話,我的測試結果和文件大小是一致的,但是有可能不同的操作系統,不同品牌的JDK版本可能會產生不同的結果。
如果你的業務可以接受這個估算值,那么就可以用它來判斷流是否為空。
toByteArray 轉換字節數組
IOUtils.toByteArray(InputStream) 轉為字節數組,由于通過流無法獲取大小,所以我就繞了個彎子,把流轉為字節數組,這樣之后不就為所欲為了了嗎?
這樣確實能拿到值,而且能準確判斷是否為空。但是如果一次性把流讀成字節數組,你不覺得內存可能受不了?
InputStream其實就是連接自來水廠的水管,不管是一噸水還是十噸水,這個InputStream占用的內存基本是固定的。用專業的話來說,它的空間復雜度是O(1)。如果把它轉化成字節數組,就相當于把你家里的十噸水全部存起來了。數據量少的話還好,但是如果遇到大數據量或者高并發的話,內存就會立馬爆掉。
聽我的建議,除非你能清楚地評估沒有 OOM 風險,否則不要轉換為字節數組。
讀取第一個字節
既然只需要判斷是否為空,那我何必這么麻煩呢?InputStream不是有read方法嗎?難道不能先讀取第一個字節,然后判斷是否為空嗎?
前面我們說過,InputStream 就像是一根水管,每讀到一個字節,流中就會少一個字節。它就像一個送貨員網校頭條,你問他湯咸不咸estimate是什么意思?怎么讀,他喝了一口說:湯不錯,不咸。如果你喝到一半湯,你會是什么感覺?雖然 InputStream 提供了 reset 方法,但是默認會拋出異常。并不是所有的流都可以 reset,就像愛情有多少次可以重來一樣。
????public?synchronized?void?reset()?throws?IOException?{
????????throw?new?IOException("mark/reset?not?supported");
????}
PushbackInputStream 的終極解決方案
PushbackInputStream,顧名思義就是可以回滾的流,你可以用它來包裝原來的流,這樣就可以檢查流是否為空。
????/**
?????*?檢查輸入流是否為空,并返回包裝后的流
?????*?請注意,原始流已經被讀了一個字節,后續不能直接對原始流進行讀取
?????*
?????*?@param?inputStream?inputStream
?????*?@return?包裝之后的流,后續操作的都是這個流
?????*/
????public?InputStream?checkStreamIsNotEmpty(InputStream?inputStream)?throws?IOException,
????????????EmptyInputStreamException?{
????????AssertKit.isNull(inputStream,?"流不能為null");
????????PushbackInputStream?pushbackInputStream?=?new?PushbackInputStream(inputStream);
????????int?b?=?pushbackInputStream.read();
????????if?(b?==?-1)?{
????????????throw?new?EmptyInputStreamException("這個流是空的,啥也沒有。?"?+?inputStream);
????????}
????????pushbackInputStream.unread(b);
????????return?pushbackInputStream;
????}