@Slf4j
@Component
public class BidirectionReques
tHeaderProcessor extends
AuthRequiredHeaderProcessor<BidirectionRequestChannel, BidirectionRequestChannelInform
ation> {
priv
ate ChannelManager channelManager; priv
ate ContainerManager containerManager; priv
ate CloudAccessChannelHelper cloudAccessChannelHelper; @Autowired public BidirectionReques
tHeaderProcessor(AuthClient authClient, BidirectionReques
tHeaderProcessorProp prop, ChannelManager channelManager, ContainerManager containerManager, CloudAccessChannelHelper cloudAccessChannelHelper) { super(authClient, prop); th
is.channelManager = channelManager; th
is.containerManager = containerManager; th
is.cloudAccessChannelHelper = cloudAccessChannelHelper; } @Override protected BidirectionRequestChannel doBeforeAddExecutor(BidirectionRequestChannel bidirectionChannel) { if (log.
isDebugEnabled()) { log.debug("[sid:{}] D
isable auto read of Bidirection channel:", bidirectionChannel.getSid()); } bidirectionChannel.d
isableAutoRead(); return bidirectionChannel; } @Override public void doHeaderProcess(BidirectionRequestChannel bidirectionChannel) throws Exception { // Add into channel manager String channelId = bidirectionChannel.getChannelId(); channelManager.addRequestChannel(channelId, bidirectionChannel); // Add life time event if necessary. As
the bidirection has sink channel character
istic,
the life time event // should be cre
ated before channel adding into container. if (bidirectionChannel instanceof ILifeTimeControl) { ((ILifeTimeControl) bidirectionChannel).cre
ateLifeTimeEvent(ILifeTimeControl.LifeTimeType.CON_IDLE); } // Obtain
the container. BidirectionContainer container = containerManager.obtainBidirectionContainer(bidirectionChannel); if (container ==
null) { log.info("[sid:{}] Bidirection channel
is kicked off by o
thers when obtaining container.", bidirectionChannel.getSid()); bidirectionChannel.sendHttpResponseAndClose(503, "Kicked off", RelayConsts.CloseReason.BIDIRECTION_KICKED_OFF); return; } if (bidirectionChannel instanceof CloudAccessBidirectionRequestChannel) { doCloudAccessHeaderProcess(bidirectionChannel, container); } // Initialize
the container. if (log.
isDebugEnabled()) { log.debug("[sid:{}] Bidirection channel obtain container successfully: container={}", bidirectionChannel.getSid(), container.toString()); } boolean hasPeerChannel = bidirectionChannel.initContainer(container); if (hasPeerChannel) { if (log.
isDebugEnabled()) { log.debug("[sid:{}] Initialize Bidirection channel after container initializ
ation", bidirectionChannel.getSid()); } bidirectionChannel.startD
ataAcceptanceAfterSinkChannelAccess(); } } public void doCloudAccessHeaderProcess(final BidirectionRequestChannel bidirectionChannel, BidirectionContainer container) throws Exception { RequestChannel cloudAccessChannel = cloudAccessChannelHelper .cre
ateCloudAccessChannel(bidirectionChannel.get
BindingId(), bidirectionChannel.getInform
ation().getOrignalRequest(), bidirectionChannel.getNettyChannel()); if (cloudAccessChannel instanceof CloudAccessBidirectionRequestChannel) { CloudAccessBidirectionRequestChannel cloudRelayChannel = (CloudAccessBidirectionRequestChannel) cloudAccessChannel; channelManager.addRequestChannel(cloudRelayChannel.getChannelId(), cloudRelayChannel); cloudRelayChannel.cre
ateLifeTimeEvent(ILifeTimeControl.LifeTimeType.CON_IDLE); containerManager.obtainBidirectionContainer(cloudRelayChannel); cloudRelayChannel.initContainer(container); } else { log.error("failed to cre
ate cloud access channel for {}", bidirectionChannel.get
BindingId()); } }
}
@Slf4j
@Component
public class PassthroughGetRequestD
ataProcessor extends AbstractProcessor {
priv
ate ChannelManager channelManager; @Autowired public PassthroughGetRequestD
ataProcessor(ChannelManager channelManager) { th
is.channelManager = channelManager; } @Override protected PassthroughGetRequestChannel doProcess(PassthroughGetRequestChannel passthroughGetChannel) throws Exception { String
bindingId = passthroughGetChannel.get
BindingId(); PostRequestChannel postChannel = (PostRequestChannel) channelManager.getRequestChannel(
bindingId); if (postChannel ==
null) { log.info("[sid:{}] POST channel
is m
issing when passthrough GET d
ata arrives.", passthroughGetChannel.getSid()); return passthroughGetChannel; } // Start Get channel transm
ission: only for
1.3 multiple mapping request passthroughGetChannel.startTransm
ission(postChannel); L
ist<IHttpD
ata> d
ataL
ist = passthroughGetChannel.getD
ataL
ist(); IHttpResponseGener
ator gener
ator = postChannel.getSuccessResponseGener
ator(); while (!d
ataL
ist.
isEmpty()) { IHttpD
ata d
ata = d
ataL
ist.remove(0); if (gener
ator !=
null && !d
ata.
isLastD
ata()) { postChannel.sendD
ata(gener
ator.getHttpD
ata(d
ata)); } else { // Maybe POST channel
is Version
1.2, th
is should not happen. d
ata.release(); } } return passthroughGetChannel; } @Override public void exceptionCaught(PassthroughGetRequestChannel passthroughGetChannel, Throwable cause) { if (cause instanceof IllegalArgumentException || cause instanceof IllegalRequestException) { log.error("[sid:{}] Illegal request d
ata:", passthroughGetChannel.getSid(), cause); passthroughGetChannel.close(RelayConsts.CloseReason.ILLEGAL_REQUEST_D
ATA_FORM
AT); } else { log.error("[sid:{}] PassthroughGetRequestD
ataProcessor failed:", passthroughGetChannel.getSid(), cause); passthroughGetChannel.close(RelayConsts.CloseReason.SERVER_INTERNAL_ERROR); } }
}
@Slf4j
@Component
public class BufferedRequestD
ataProcessor extends RelayPostRequestD
ataProcessor {
@Override protected D
ataProcessSt
ate processD
ata(BufferedPostRequestChannel bufferedPostChannel, D
ataProcessSt
ate d
ataProcessSt
ate, IHttpD
ata d
ata) { if (d
ataProcessSt
ate == D
ataProcessSt
ate.TRANSMIT_DIRECT) { return super.transmitD
ataDirectly(bufferedPostChannel, D
ataProcessSt
ate.TRANSMIT_DIRECT, d
ata); } else if (d
ataProcessSt
ate == D
ataProcessSt
ate.WANT_BUFFERED_D
ATA) { if (log.
isDebugEnabled()) { log.debug("[sid:{}] Process buffered d
ata:
index={}", bufferedPostChannel.getSid(), d
ata.get
Index()); } // store buffered d
ata int left = bufferedPostChannel.addBufferedHttpD
ata(d
ata); if (left > 0) { // transmit buffered d
ata to bound channels bufferedPostChannel.transmitD
ata(d
ata); // keep st
ate return D
ataProcessSt
ate.WANT_BUFFERED_D
ATA; } else if (left == 0) { // transmit buffered d
ata to bound channels bufferedPostChannel.transmitD
ata(d
ata); // change d
ata process st
at to D
ataProcessSt
at. return bufferedPostChannel.changeToNextSt
ate(); } else { // Th
is will not happen super.transmitD
ataDirectly(bufferedPostChannel, D
ataProcessSt
ate.TRANSMIT_DIRECT, d
ata); return bufferedPostChannel.changeToNextSt
ate(); } } else if (d
ataProcessSt
ate == D
ataProcessSt
ate.WANT_RESPONSE_MOULD) { return super.processHttpResponseMould(bufferedPostChannel, d
ata); } else { // Use direct transm
ission as default action. Th
is case should not happen. return super.transmitD
ataDirectly(bufferedPostChannel, D
ataProcessSt
ate.TRANSMIT_DIRECT, d
ata); } }
}
@Slf4j
@Component(“relayPostRequestD
ataProcessor”)
public class RelayPostRequestD
ataProcessor extends PostRequestD
ataProcessor {
@Override protected D
ataProcessSt
ate processD
ata(T postChannel, D
ataProcessSt
ate d
ataProcessSt
ate, IHttpD
ata d
ata) { if (d
ataProcessSt
ate == D
ataProcessSt
ate.TRANSMIT_DIRECT) { return super.transmitD
ataDirectly(postChannel, D
ataProcessSt
ate.TRANSMIT_DIRECT, d
ata); } else if (d
ataProcessSt
ate == D
ataProcessSt
ate.WANT_RESPONSE_MOULD) { return processHttpResponseMould(postChannel, d
ata); } else { // Use direct transm
ission as default action. Th
is case should not happen. return super.transmitD
ataDirectly(postChannel, D
ataProcessSt
ate.TRANSMIT_DIRECT, d
ata); } } /** * Parse
the response mould into response header, set it into POST request channel and transmit it to all GET * request channels rel
ated with
the POST request channel. * * @param postChannel
The POST request channel which received
the response mould d
ata. * @param d
ata
The response mould d
ata */ protected D
ataProcessSt
ate processHttpResponseMould(T postChannel, IHttpD
ata d
ata) throws IllegalRequestException { log.debug("[sid:{}] Process first d
ata:
index={}", postChannel.getSid(), d
ata.get
Index()); try { String content =
null; if (d
ata instanceof MultipartHttpD
ata) { content = ((MultipartHttpD
ata) d
ata).addedContents().toString(d
ata.getContentCharset()); } else { content = d
ata.toString(); } HttpResponse httpResponse = parseHttpResponse(content); // If
the Transfer-Encoding: chunked
is set, do not modified as Chunked
is used to measure
the message // length of
the HTTP response even if
the Content-Length
is set
at the same time. if (!HttpUtil.
isTransferEncodingChunked(httpResponse)) { String type = postChannel.getInform
ation().getParams().get(RelayConsts.ParamKey.TYPE); if (RelayConsts.Type.FILE.equals(type)) { // If
the type
is file,
the Content-Length field in response should be
the file size and multipart // should not be used. if (postChannel.getInform
ation().getContentType().
isMultiPart() && httpResponse.headers().contains(HttpHeaderNames.CONTENT_LENGTH)) { throw new IllegalRequestException( "file type with multipart package should use Transfer-Encoding: chunked."); } } else { // If
the type
is o
ther
value, only when
the Content-Length field in response does not ex
ist or
is // negtive, new Content-Length will be set. if (HttpUtil.getContentLength(httpResponse, -
1L) < 0) { long contentLength = HttpUtil.getContentLength(postChannel.getInform
ation().getOrignalRequest(), -
1L); httpResponse.headers().set(RelayConsts.HttpNames.CONTENT_LENGTH, contentLength > 0 ? contentLength : Long.MAX_
VALUE); } } } // if st
atus
is 429, it shows device has too many relay connections
at the same time // if (postChannel.hasSegmenter() && // httpResponse.st
atus().code() == HttpResponseSt
atus.TOO_MANY_REQUESTS.code()) { // postChannel.stopSegmenterWhenCanNotGetVideoStream(RelayConsts.SegmenterStopReason.RELAY_CONNECTION_EXCEEDED); // } postChannel.setResponseHeader(httpResponse); postChannel.transmitHttpResponse(httpResponse); return postChannel.changeToNextSt
ate(); } finally { d
ata.release(); } } /** * Parse response header from {@link String} form
at to {@link HttpResponse} form
at. * * @param string
Value {@link String} form
at of response header. * * @return {@link HttpResponse} form
at of response header. */ public HttpResponse parseHttpResponse(String string
Value) throws IllegalRequestException { String[] parts = string
Value.split("\r\n", 2); if (parts.length != 2 || parts[
1].
isEmpty()) { throw new IllegalRequestException("Invalid first d
ata as a response header: " + string
Value); } String[] st
atusLine = parts[0].split(" ", 3); if (st
atusLine.length != 3 || st
atusLine[
1].
isEmpty() || st
atusLine[2].
isEmpty()) { throw new IllegalRequestException("Invalid first d
ata with an invalid response line: " + parts[0]); } HttpResponse httpResponse =
null; try { HttpResponseSt
atus st
atus = new SpecificCodeHttpResponseSt
atus(st
atusLine[
1], st
atusLine[2]); httpResponse = new DefaultHttpResponse(HttpVersion.HTTP_
1_
1, st
atus); } c
atch (NumberForm
atException e) { throw new IllegalRequestException("Invalid first d
ata with an unresovled st
atus code: " + parts[0]); } String[] entries = parts[
1].split("\r\n"); for (String entry : entries) { String[] content = entry.split(":", 2); if (content.length != 2 || content[
1].
isEmpty()) { throw new IllegalRequestException("Invalid first d
ata with an unresovled header: " + entry); } String name = content[0].trim(); String
value = content[
1].trim(); if (name.
isEmpty()) { throw new IllegalRequestException("Invalid first d
ata with an empty key of header:" + entry); } httpResponse.headers().add(name,
value); } // Netty channel of GET could not be reused. httpResponse.headers().set(HttpHeaderNames.CONNECTION, HttpHeader
Values.CLOSE); httpResponse.headers().set(HttpHeaderNames.PRAGMA, HttpHeader
Values.NO_CACHE); httpResponse.headers().set(HttpHeaderNames.CACHE_CONTROL, HttpHeader
Values.NO_CACHE); return httpResponse; }
}
@Slf4j
@Component(“postRequestD
ataProcessor”)
public class PostRequestD
ataProcessor extends AbstractProcessor {
@Override protected T doProcess(T sourceChannel) throws IllegalRequestException { D
ataProcessSt
ate d
ataProcessSt
at = sourceChannel.getD
ataProcessSt
ate(); L
ist<IHttpD
ata> d
ataL
ist = sourceChannel.getD
ataL
ist(); while (!d
ataL
ist.
isEmpty()) { IHttpD
ata d
ata = d
ataL
ist.remove(0); d
ataProcessSt
at = processD
ata(sourceChannel, d
ataProcessSt
at, d
ata); } return sourceChannel; } protected D
ataProcessSt
ate processD
ata(T sourceChannel, D
ataProcessSt
ate d
ataProcessSt
ate, IHttpD
ata d
ata) { return transmitD
ataDirectly(sourceChannel, d
ataProcessSt
ate, d
ata); } protected D
ataProcessSt
ate transmitD
ataDirectly(T sourceChannel, D
ataProcessSt
ate d
ataProcessSt
at, IHttpD
ata d
ata) { assert d
ataProcessSt
at == D
ataProcessSt
ate.TRANSMIT_DIRECT; // Transmit
the d
ata sourceChannel.transmitD
ata(d
ata); // Release d
ata original reference d
ata.release(); return D
ataProcessSt
ate.TRANSMIT_DIRECT; } @Override public void exceptionCaught(T sourceChannel, Throwable cause) { if (cause instanceof IllegalArgumentException || cause instanceof IllegalRequestException) { log.error("[sid:{}] Illegal request d
ata form
at:", sourceChannel.getSid(), cause); sourceChannel.close(RelayConsts.CloseReason.ILLEGAL_REQUEST_D
ATA_FORM
AT); } else { log.error("[sid:{}] PostRequestD
ataProcessor failed:", sourceChannel.getSid(), cause); sourceChannel.close(RelayConsts.CloseReason.SERVER_INTERNAL_ERROR); } }
}
@Slf4j
public class DefaultGetRequestChannel extends GetRequestChannel implements ILifeTimeControl, IHeartBe
atControl {
priv
ate st
atic final RuntimeException HEART_BE
AT_TIMEOUT_EXCEPTION = new RuntimeException("Heart be
at timeout"); protected ILifeTimeControl lifeTimeController; protected
AtomicLong closedTimeStamp; public DefaultGetRequestChannel(String channelId, Channel nettyChannel, String
bindingId, int basicLifeTime, GetRequestChannelInform
ation inform
ation) { super(channelId, nettyChannel,
bindingId, inform
ation); th
is.lifeTimeController = buildLifeTimeController(nettyChannel, basicLifeTime); th
is.closedTimeStamp = new
AtomicLong(System.currentTimeMill
is() + IHeartBe
atControl.PROLONGED_TIME_MS); } protected ILifeTimeControl buildLifeTimeController(Channel nettyChannel, int basicLifeTime) { IEventCallBack callback = new IEventCallBack() { @Override public void handleEvent() { if (log.
isDebugEnabled()) { log.debug("[sid:{}] Channel
is closing as timeout: channelType={}", getSid(), DefaultGetRequestChannel.th
is.getClass().getSimpleName()); } sendHttpResponseAndClose(HttpResponseSt
atus.NOT_FOUND, RelayConsts.CloseReason.GET_LIFETIME_EXPIRE); } }; return new LifeTimeController(getSid(), ILifeTimeControl.GET_IDLE_TIME_S, basicLifeTime, EventTimer.getLifeTimeEventKey(nettyChannel), getClass().getSimpleName(), callback); } @Override protected
ISinkHandler buildBasicSinkHandler(GetRequestChannelInform
ation inform
ation) { assert !inform
ation.
isFFmpegChannel(); if (
isAudio(inform
ation)) { return new AudioSinkHandler(th
is); } else { return new DefaultSinkHandler<>(th
is); } } priv
ate boolean
isAudio(GetRequestChannelInform
ation inform
ation) { String type = inform
ation.getParams().get(RelayConsts.ParamKey.TYPE); switch (type) { case RelayConsts.Type.AUDIO: return true; case RelayConsts.Type.NVR: case RelayConsts.Type.SMART_NVR: String resolution = inform
ation.getParams().get(RelayConsts.ParamKey.RESOLUTION); if (resolution ==
null) { return false; } switch (resolution) { case RelayConsts.AudioResolution.AAC: case RelayConsts.AudioResolution.MP2: case RelayConsts.AudioResolution.PCM: return true; case RelayConsts.VideoResolution.HD: case RelayConsts.VideoResolution.QVGA: case RelayConsts.VideoResolution.VGA: default: return false; } default: return false; } } @Override public long getClosedTimeStamp() { return closedTimeStamp.get(); } @Override public void setClosedTimeStamp(long timeStamp) { closedTimeStamp.lazySet(timeStamp); inform
ation.addHeartBe
atNum(); } @Override public String getLifeTimeEventKey() { return lifeTimeController.getLifeTimeEventKey(); } @Override public int getBasicLifeTime() { return lifeTimeController.getBasicLifeTime(); } @Override public int cre
ateLifeTimeEvent(LifeTimeType type) throws TimerEventException { return lifeTimeController.cre
ateLifeTimeEvent(type); } @Override public void prolongLeftTimeTo(int newLifeTime) throws TimerEventException { lifeTimeController.prolongLeftTimeTo(newLifeTime); } @Override public void shortenLeftTimeTo(int newLifeTime) { lifeTimeController.shortenLeftTimeTo(newLifeTime); } @Override public void removeLifeTimeEvent() { lifeTimeController.removeLifeTimeEvent(); } /** * {@inheritDoc} */ @Override public boolean upd
ateAfterContainerInitializ
ation(
ISourceRequestChannel postRequestChannel) { try { boolean
isSuccess = super.upd
ateAfterContainerInitializ
ation(postRequestChannel); if (
isSuccess) { lifeTimeController.prolongLeftTimeTo(lifeTimeController.getBasicLifeTime()); } return
isSuccess; } c
atch (Exception e) { log.error("[sid:{}] Exception caught when upd
ating Get channel after container initializ
ation", getSid(), e); return false; } } /** * {@inheritDoc} */ @Override public boolean upd
ateWhen
Binding(PostRequestChannel postRequestChannel) { try { boolean
isSuccess = super.upd
ateWhen
Binding(postRequestChannel); if (
isSuccess) { cre
ateLifeTimeEvent(LifeTimeType.BASIC); } return
isSuccess; } c
atch (Exception e) { log.error("[sid:{}] Exception caught when
binding Get channel", getSid(), e); return false; } } /** * If
the heart be
at is expired,
the Get channel will be closed instead of sending templ
ate d
ata.
At th
is time,
the * templ
ate d
ata reference will not be increased. * * @param templ
ateD
ata
The templ
ate d
ata */ @Override public ChannelFuture sendTempl
ateD
ata(IHttpD
ata templ
ateD
ata) { if (System.currentTimeMill
is() > getClosedTimeStamp()) { if (log.
isDebugEnabled()) { log.debug("[sid:{}] Close GET channel as losing heartbe
at:", getSid()); } close(RelayConsts.CloseReason.GET_LACK_HEART_BE
AT); DefaultChannelProm
ise prom
ise = new DefaultChannelProm
ise(getNettyChannel()); prom
ise.setFailure(HEART_BE
AT_TIMEOUT_EXCEPTION); return prom
ise; } else { return super.sendTempl
ateD
ata(templ
ateD
ata); } } @Override protected void releaseResourceAfterNettyChannelClosure() { super.releaseResourceAfterNettyChannelClosure(); removeLifeTimeEvent(); }
}
public class PassthroughGetRequestChannel extends DefaultGetRequestChannel implements ID
ataFollowedRelayRequestChannel {
priv
ate RelayServiceD
ataController<PassthroughGetRequestChannel> d
ataController; priv
ate vol
atile boolean
isStartD
ataProcess; public PassthroughGetRequestChannel(String channelId, Channel nettyChannel, String
bindingId, // ClientInfoController clientInfoController, int basicLifeTime, GetRequestChannelInform
ation inform
ation) { super(channelId, nettyChannel,
bindingId, // clientInfoController, basicLifeTime, inform
ation); th
is.d
ataController = new RelayServiceD
ataController<>(th
is, false); } @Override public boolean
isStartD
ataProcess() { return
isStartD
ataProcess; } @Override public void setStartD
ataProcess() { th
is.
isStartD
ataProcess = true; } @Override public boolean upd
ateAfterContainerInitializ
ation(
ISourceRequestChannel postRequestChannel) { if (super.upd
ateAfterContainerInitializ
ation(postRequestChannel)) { setStartD
ataProcess(); enableAutoRead(); addPushD
ataTask(); return true; } else { return false; } } /** * {@inheritDoc} */ @Override public boolean upd
ateWhen
Binding(PostRequestChannel postRequestChannel) { if (super.upd
ateWhen
Binding(postRequestChannel)) { setStartD
ataProcess(); enableAutoRead(); addPushD
ataTask(); return true; } else { return false; } } @Override public void d
isableAutoRead() { d
ataController.d
isableAutoRead(); } @Override public void enableAutoRead() { d
ataController.enableAutoRead(); } @Override public L
ist<IHttpD
ata> getD
ataL
ist() { return d
ataController.getD
ataL
ist(); } @Override public void addPushD
ataTask() { d
ataController.addPushD
ataTask(); } @Override protected void releaseResourceAfterNettyChannelClosure() { super.releaseResourceAfterNettyChannelClosure(); d
ataController.releaseD
ataL
ist(); }
}
@Slf4j
public class BidirectionRequestChannel
extends SourceRequestChannel<BidirectionContainer, BidirectionRequestChannelInform
ation>
implements
ISinkRequestChannel {
priv
ate DefaultSinkHandler<BidirectionRequestChannel> sinkHandler; public BidirectionRequestChannel(String channelId, Channel nettyChannel, String
bindingId, BidirectionRequestChannelInform
ation inform
ation) { super(channelId, nettyChannel,
bindingId, inform
ation); th
is.sinkHandler = new DefaultSinkHandler<>(th
is); } @Override public String getRequestType() { if (
isDeviceChannel()) { return "relayservice-deviceStream"; } else { return "relayservice-appStream"; } } @Override public boolean initContainer(BidirectionContainer container) { assert th
is.container ==
null : "initContainer() should only be called once."; th
is.container = container; BidirectionRequestChannel peerChannel = getPeerDirectionChannel(container); if (peerChannel ==
null) { return false; } if (peerChannel.upd
ateAfterContainerInitializ
ation(th
is)) { return true; } else { // As peer channel
is closing, th
is bidirection channel will be closing soon. close(RelayConsts.CloseReason.BIDIRECTION_CLOSED_BY_PEER); return false; } } public BidirectionRequestChannel getPeerDirectionChannel() { BidirectionContainer container = th
is.container; if (container !=
null) { return getPeerDirectionChannel(container); } else { return
null; } } priv
ate BidirectionRequestChannel getPeerDirectionChannel(BidirectionContainer container) { if (
isDeviceChannel()) { return container.getAppChannel(); } else { return container.getDeviceChannel(); } } /** * Only contains peer channel if it ex
ists. */ @Override protected L
ist<? extends
ISinkRequestChannel> getTransmittedSinkRequestChannels() { BidirectionContainer container = th
is.container; if (container ==
null) { return new ArrayL
ist<>(); } L
ist<BidirectionRequestChannel> l
ist = new ArrayL
ist<>(); BidirectionRequestChannel peerChannel = getPeerDirectionChannel(container); if (peerChannel !=
null) { l
ist.add(peerChannel); } return l
ist; } /** * Do nothing, waiting terminal to close
the request channel. */ @Override protected void doProcessBeforeTransmitD
ata(IHttpD
ata templ
ateD
ata) { // do nothing } @Override protected void transmitD
ataToSinkRequestChannel(
ISinkRequestChannel sinkChannel, IHttpD
ata templ
ateD
ata) { // 当channel来自回放请求时，channel为长连接，此时为了适配web无法发起正常http长连接，手动将lastD
ata的空包丢掉 String type = th
is.getInform
ation().getParam
Value(RelayConsts.ParamKey.TYPE); if ((Objects.equals(type, RelayConsts.Type.SDVOD) || Objects.equals(type, RelayConsts.Type.DOWNLOAD)) && templ
ateD
ata.
isLastD
ata()) { return; } sinkChannel.sendTempl
ateD
ata(templ
ateD
ata); } @Override public void startD
ataAcceptanceAfterSinkChannelAccess() { assert container !=
null : "initContainer should be called first"; BidirectionRequestChannel peerChannel = getPeerDirectionChannel(container); if (peerChannel ==
null) { log.warn("[sid:{}] Peer channel
is m
issing, may be closed?", getSid()); return; } if (inform
ation.getSt
atisticInform
ation().get
IsWs()) { // Start d
ata process setStartD
ataProcess(); // Enable auto read enableAutoRead(); return; } do
BindInitializ
ation(peerChannel); } @Override public D
ataProcessSt
ate getD
ataProcessSt
ate() { return D
ataProcessSt
ate.TRANSMIT_DIRECT; } @Override public D
ataProcessSt
ate changeToNextSt
ate() { return D
ataProcessSt
ate.TRANSMIT_DIRECT; } @Override public boolean upd
ateAfterContainerInitializ
ation(
ISourceRequestChannel sourceRequestChannel) { if (log.
isDebugEnabled()) { log.debug("[sid:{}] Do
bind initializ
ation after container initializ
ation", getSid()); } do
BindInitializ
ation(sourceRequestChannel); return true; } protected void do
BindInitializ
ation(
ISourceRequestChannel sourceRequestChannel) { // Upd
ate sid String sourceSid = sourceRequestChannel.getSid(); if (sourceSid.
indexOf('-') == -
1) { String oldSid = getSid(); if (oldSid.
indexOf('-') == -
1) { upd
ateSid(sourceSid + '-' + oldSid); } } // Try to send response header HttpResponse responseHeader = sourceRequestChannel.getResponseHeader(); if (responseHeader !=
null) { sendTempl
ateHttpResponse(responseHeader); } // Start d
ata process setStartD
ataProcess(); // Enable auto read enableAutoRead(); // Add push d
ata task addPushD
ataTask(); } @Override public ChannelFuture sendTempl
ateHttpResponse(HttpResponse templ
ateHttpResponse) { return sinkHandler.sendTempl
ateHttpResponse(templ
ateHttpResponse); } @Override public ChannelFuture sendTempl
ateHttpResponseAndClose(HttpResponse templ
ateHttpResponse, RelayConsts.CloseReason closeReason) { return sinkHandler.sendTempl
ateHttpResponseAndClose(templ
ateHttpResponse, closeReason); } @Override public ChannelFuture sendTempl
ateD
ata(IHttpD
ata templ
ateD
ata) { return sinkHandler.sendTempl
ateD
ata(templ
ateD
ata); } @Override public ChannelFuture sendTempl
ateD
ataAndClose(IHttpD
ata templ
ateD
ata, RelayConsts.CloseReason closeReason) { return sinkHandler.sendTempl
ateD
ataAndClose(templ
ateD
ata, closeReason); } @Override protected void releaseResourceBeforeNettyChannelClosure() { if (inform
ation.getCloseReason() != RelayConsts.CloseReason.BIDIRECTION_CLOSED_BY_PEER) { ContainerManager.getInstance().removeBidirectionContainer(th
is); } super.releaseResourceBeforeNettyChannelClosure(); }
}
@Slf4j
public abstract class SourceRequestChannel<T extends IRequestChannelContainer<? extends
ISinkRequestChannel>, I extends AbstractTransactionRequestChannelInform
ation>
extends TransactionRequestChannel implements
ISourceRequestChannel {
priv
ate String
bindingId; priv
ate ChannelTrafficShapingHandler trafficHandler; priv
ate RelayServiceD
ataController<SourceRequestChannel<T, I>> d
ataController; priv
ate vol
atile boolean
isStartD
ataProcess; priv
ate HttpResponse responseHeader; protected T container; public SourceRequestChannel(String channelId, Channel nettyChannel, String
bindingId, I inform
ation) { super(channelId, nettyChannel, inform
ation); th
is.
bindingId =
bindingId; th
is.d
ataController = new RelayServiceD
ataController<>(th
is, false); } @Override public String get
BindingId() { return
bindingId; } public ChannelTrafficShapingHandler getTrafficHandler() { return trafficHandler; } public void setTrafficHandler(ChannelTrafficShapingHandler trafficHandler) { th
is.trafficHandler = trafficHandler; } @Override public void d
isableAutoRead() { d
ataController.d
isableAutoRead(); } @Override public void enableAutoRead() { d
ataController.enableAutoRead(); } @Override public L
ist<IHttpD
ata> getD
ataL
ist() { return d
ataController.getD
ataL
ist(); } @Override public void addPushD
ataTask() { d
ataController.addPushD
ataTask(); } @Override public HttpResponse getResponseHeader() { return responseHeader; } @Override public void setResponseHeader(HttpResponse responseHeader) { th
is.responseHeader = responseHeader; } @Override public boolean
isStartD
ataProcess() { return
isStartD
ataProcess; } @Override public void setStartD
ataProcess() { th
is.
isStartD
ataProcess = true; } /** * Initialize
the source channel with a container in which
the sink channel
is put into. * * @param container
The container * * @return True if
the container has
at least one sink channel. */ public abstract boolean initContainer(T container); public boolean
isInitContainer() { return container !=
null; } public T getContainer() { return container; } /** * Only used for Junit test. */ void setContainer(T container) { th
is.container = container; } /** * Start source channel d
ata acceptance after sink channel access. */ public abstract void startD
ataAcceptanceAfterSinkChannelAccess(); /** * {@inheritDoc} */ @Override public void transmitHttpResponse(HttpResponse templ
ateHttpResponse) { L
ist<? extends
ISinkRequestChannel> currentL
ist = getTransmittedSinkRequestChannels(); for (
ISinkRequestChannel sinkChannel : currentL
ist) { try { sinkChannel.sendTempl
ateHttpResponse(templ
ateHttpResponse); } c
atch (Exception e) { log.error("[sid:{}] Failed to transmit response to GET channel: channelId={}:", getSid(), sinkChannel.getChannelId(), e); sinkChannel.close(RelayConsts.CloseReason.SERVER_INTERNAL_ERROR); } } } /** * {@inheritDoc} */ @Override public void transmitD
ata(IHttpD
ata templ
ateD
ata) { L
ist<? extends
ISinkRequestChannel> currentL
ist = templ
ateD
ata.
isLastD
ata() ? getTransmittedSinkRequestChannels() :
null; doProcessBeforeTransmitD
ata(templ
ateD
ata); currentL
ist = currentL
ist ==
null ? getTransmittedSinkRequestChannels() : currentL
ist; for (
ISinkRequestChannel sinkChannel : currentL
ist) { try { transmitD
ataToSinkRequestChannel(sinkChannel, templ
ateD
ata); } c
atch (Exception e) { log.error("[sid:{}] Failed to transmit d
ata to GET channel: channelId={},
index={}", getSid(), sinkChannel.getChannelId(), templ
ateD
ata.get
Index(), e); sinkChannel.close(RelayConsts.CloseReason.SERVER_INTERNAL_ERROR); } } } /** * Obtain all sink channel which can be transmitted response and d
ata to. * * @return
The sink channel which can be transmitted response and d
ata to. */ protected abstract L
ist<? extends
ISinkRequestChannel> getTransmittedSinkRequestChannels(); /** * Process before transmit templ
ate d
ata. * * @param templ
ateD
ata
The templ
ate d
ata */ protected abstract void doProcessBeforeTransmitD
ata(IHttpD
ata templ
ateD
ata); /** * Transmit templ
ate d
ata to sink channel. * * @param sinkChannel
The sink channel * @param templ
ateD
ata
The templ
ate d
ata. */ protected abstract void transmitD
ataToSinkRequestChannel(
ISinkRequestChannel sinkChannel, IHttpD
ata templ
ateD
ata); @Override protected void releaseResourceAfterNettyChannelClosure() { super.releaseResourceAfterNettyChannelClosure(); d
ataController.releaseD
ataL
ist(); }
}
@Slf4j
public abstract class TransactionRequestChannel extends
RequestChannel {
protected String channelId; @Getter @Setter priv
ate String content; public TransactionRequestChannel(String channelId, Channel nettyChannel, I inform
ation) { super(nettyChannel, inform
ation); th
is.channelId = channelId; } public String getChannelId() { return channelId; } public ChannelVersion getChannelVersion() { return inform
ation.getChannelVersion(); } public boolean
isDeviceChannel() { return inform
ation.
isFromDevice(); } public AbstractAuthContext getAuthContext() { return inform
ation.getAuthContext(); } public RequestLevel getRequestLevel() { return getAuthContext().getLevel(); } @Override protected void releaseResourceBeforeNettyChannelClosure() { // Call super method super.releaseResourceBeforeNettyChannelClosure(); // Reomve itself from ChannelManager immedi
ately if (
isRequestChannelRemovedImmedi
ately()) { removeFromChannelManager(); } } @Override protected void releaseResourceAfterNettyChannelClosure() { // Call super method super.releaseResourceAfterNettyChannelClosure(); // Reomve itself from ChannelManager after
1s. if (!
isRequestChannelRemovedImmedi
ately()) { getNettyChannel().eventLoop().schedule(new Runnable() { @Override public void run() { removeFromChannelManager(); } },
1000, TimeUnit.MILL
ISECONDS); } } protected boolean
isRequestChannelRemovedImmedi
ately() { return true; } ; protected void removeFromChannelManager() { if (log.
isDebugEnabled()) { log.debug("[sid:{}] Remove request channel {} channelId={} from channel manager", getSid(), th
is.getClass().getSimpleName(), getChannelId()); } ChannelManager.getInstance().removeRequestChannel(channelId, th
is); }
}
@Slf4j
public class GetRequestChannel extends TransactionRequestChannel implements
ISinkRequestChannel {
protected String
bindingId; protected
AtomicBoolean
isAllowTransm
ission; protected
ISinkHandler sinkHandler; public GetRequestChannel(String channelId, Channel nettyChannel, String
bindingId, GetRequestChannelInform
ation inform
ation) { super(channelId, nettyChannel, inform
ation); th
is.
bindingId =
bindingId; th
is.
isAllowTransm
ission = new
AtomicBoolean(false); th
is.sinkHandler = buildSinkHandler(inform
ation); } /** * Build
the sink hanlder according to
the sink channel. When any templ
ate response and d
ata are sent to th
is Get * channel,
the sending action will be processed by using sink hanlder. * * @param inform
ation
The Get channel inform
ation * * @return
The sink hanlder. */ protected
ISinkHandler buildSinkHandler(GetRequestChannelInform
ation inform
ation) { if (inform
ation.
isNeedMultipleMappingSource()) { // Flow control
is only used in Preview type(such as video or nvr) return new FlowControlHandler(buildBasicSinkHandler(inform
ation)); } else { return buildBasicSinkHandler(inform
ation); } } /** *
The basic sink handler which
is used to do actual sending action. * * @param inform
ation
The Get channel inform
ation * * @return
The basic sink hanlder. */ protected
ISinkHandler buildBasicSinkHandler(GetRequestChannelInform
ation inform
ation) { return new DefaultSinkHandler<>(th
is); } @Override public String getRequestType() { if (getChannelVersion() == ChannelVersion.VERSION_
1_2) { return "relayservice-getStream"; } else if (
isDeviceChannel()) { return "relayservice-deviceStream"; } else { return "relayservice-appStream"; } } @Override public String get
BindingId() { return
bindingId; } public void upd
ate
BindingId(String new
BindingId) { th
is.
bindingId = new
BindingId; } public boolean
isAllowTransm
ission() { return
isAllowTransm
ission.get(); } public boolean setAllowTransm
ission() { return th
is.
isAllowTransm
ission.compareAndSet(false, true); } /** * Whe
ther
the type
is a multiple mapping type, such as video, audio, mixed and nvr. * * @return True if
the type
is a multiple mapping type. */ public boolean
isNeedMultipleMappingSource() { return inform
ation.
isNeedMultipleMappingSource(); } /** * If
the Get channel arrives before
the corresponding Post channel,
the Get channel will be put into a container. * When
the Post channel arrives,
the Post channel will find and initialize
the container to check whe
ther any Get * channel
is bound with it.
At th
is time, using th
is method
the notify
the Get channel of
binding action after * container initializ
ation. * <p> * <p> * Th
is method will be called in following cases: * <ul> * <li>After
the Post channel initializes
the container, any Get channel in
the container will be notified of
the * initializ
ation by calling th
is.</li> * <li>If
the container has been initialized by Post channel when
the Get channel
is added into container directly. *
The Get channel should be notified of
the initializ
ation by calling th
is.</li> * </ul> * </p> * * @param postRequestChannel
The source channel which th
is sink channel
is bound with. * * @return Whe
ther
the sink channel
is successfully upd
ated. If false,
the sink channel may be closed and
the source * channel should remove it. */ @Override public boolean upd
ateAfterContainerInitializ
ation(final
ISourceRequestChannel postRequestChannel) { do
BindInitializ
ation(postRequestChannel); return true; } /** * Upd
ate Get channel when
the Get channel
is bound with Post channel. * * @param postRequestChannel
The corresponding Post channel. * * @return True if
the upd
ating oper
ation
is successful. */ public boolean upd
ateWhen
Binding(final PostRequestChannel postRequestChannel) { do
BindInitializ
ation(postRequestChannel); return true; } priv
ate void do
BindInitializ
ation(final
ISourceRequestChannel requestChannel) { EventLoop eventLoop = getNettyChannel().eventLoop(); if (eventLoop.inEventLoop()) { do
BindInitializ
ationInEventLoop(requestChannel); } else { eventLoop.execute(() -> do
BindInitializ
ationInEventLoop(requestChannel)); } } /** *
The initializ
ation of Get channel when bound with Post channel * * @param postRequestChannel
The corresponding Post channel. */ protected void do
BindInitializ
ationInEventLoop(
ISourceRequestChannel postRequestChannel) { // Upd
ate sid upd
ateSidByPrefix(postRequestChannel.getSid()); // Try to send response header HttpResponse responseHeader = postRequestChannel.getResponseHeader(); if (responseHeader !=
null) { sendTempl
ateHttpResponse(responseHeader); } } /** * To initialize
the Get request channel transm
ission. * * @param postRequestChannel
The corresponding Post request channel. */ public void startTransm
ission(final PostRequestChannel postRequestChannel) { EventLoop eventLoop = getNettyChannel().eventLoop(); if (eventLoop.inEventLoop()) { doStartTransm
ission(postRequestChannel); } else { eventLoop.execute(() -> doStartTransm
ission(postRequestChannel)); } } priv
ate void doStartTransm
ission(PostRequestChannel postRequestChannel) { assert getNettyChannel().eventLoop().inEventLoop(); // Th
is is helpful when
the 2nd App for smart codec stream. initSmartCodec(postRequestChannel); if (setAllowTransm
ission()) { int num = postRequestChannel.upd
ateTransmittedGetRequestChannels(); if (log.
isDebugEnabled()) { log.debug("[sid:{}] Start transm
ission of GET channel: transmittedGetChannelNum={}", getSid(), num); } // Th
is is helpful when
the 1st App and smart codec notific
ation may be comming soon. initSmartCodec(postRequestChannel); /* L
ist<IHttpD
ata> templ
ateD
ataL
ist = getD
ataL
istToStartTransm
ission(postRequestChannel); for (IHttpD
ata templ
ateD
ata : templ
ateD
ataL
ist) { sendTempl
ateD
ata(templ
ateD
ata); }*/ } } priv
ate void initSmartCodec(PostRequestChannel postRequestChannel) { if (postRequestChannel.
isSmartCodec()) { startSmartCodec(); } else { stopSmartCodec(); } } /** * If
the corresponding channel
is {@link BufferedPostRequestChannel},
the buffered d
ata will be returned. * O
therw
ise, empty array will be returned. * * @param postRequestChannel
The corresponding Post request channel. * * @return
The d
ata l
ist for initializ
ation of transm
ission */ protected L
ist<IHttpD
ata> getD
ataL
istToStartTransm
ission(PostRequestChannel postRequestChannel) { L
ist<IHttpD
ata> d
ataL
ist; if (postRequestChannel instanceof BufferedPostRequestChannel) { d
ataL
ist = ((BufferedPostRequestChannel) postRequestChannel).getAllBufferedD
ata(); if (log.
isDebugEnabled()) { log.debug("[sid:{}] Add buffered HTTP d
ata: size={}", getSid(), d
ataL
ist.size()); } } else { d
ataL
ist = new ArrayL
ist<>(); } IHttpD
ata iFrame = postRequestChannel.getIFrame(); if (iFrame !=
null) { d
ataL
ist.add(iFrame); } return d
ataL
ist; } @Override public ChannelFuture sendTempl
ateHttpResponse(HttpResponse templ
ateHttpResponse) { return sinkHandler.sendTempl
ateHttpResponse(templ
ateHttpResponse); } @Override public ChannelFuture sendTempl
ateHttpResponseAndClose(HttpResponse templ
ateHttpResponse, RelayConsts.CloseReason closeReason) { return sinkHandler.sendTempl
ateHttpResponseAndClose(templ
ateHttpResponse, closeReason); } @Override public ChannelFuture sendTempl
ateD
ata(IHttpD
ata templ
ateD
ata) { return sinkHandler.sendTempl
ateD
ata(templ
ateD
ata); } @Override public ChannelFuture sendTempl
ateD
ataAndClose(IHttpD
ata templ
ateD
ata, RelayConsts.CloseReason closeReason) { return sinkHandler.sendTempl
ateD
ataAndClose(templ
ateD
ata, closeReason); } public void startSmartCodec() {
ISinkHandler handler = sinkHandler; FlowControlHandler flowControlHandler = findFlowControlHandler(handler); if (flowControlHandler !=
null) { flowControlHandler.setSmartCodec(true); } if (!(handler instanceof SmartCodecSinkHandler)) { sinkHandler = new SmartCodecSinkHandler(handler); } } public void stopSmartCodec() {
ISinkHandler handler = sinkHandler; FlowControlHandler flowControlHandler = findFlowControlHandler(handler); if (flowControlHandler !=
null) { flowControlHandler.setSmartCodec(false); } if (handler instanceof SmartCodecSinkHandler) { sinkHandler = ((SmartCodecSinkHandler) handler).unwrap(); } } public void upd
ateKeyFrameUTCTime(long utcTime) { FlowControlHandler h = findFlowControlHandler(sinkHandler); if (h !=
null) { h.upd
ateUTCTime(utcTime); } } public void upd
ateWritabilityEvent(boolean
isWritable) { FlowControlHandler h = findFlowControlHandler(sinkHandler); if (h !=
null) { h.setSinkChannelWriterable(
isWritable); } } priv
ate FlowControlHandler findFlowControlHandler(
ISinkHandler h) { while (h instanceof FilterSinkHandler) { if (h instanceof FlowControlHandler) { return (FlowControlHandler) h; } else { h = ((FilterSinkHandler) h).unwrap(); } } return
null; } @Override protected void doClose() { if (RelayConsts.Type.FILE.equals(inform
ation.getParams().get(RelayConsts.ParamKey.TYPE))) { // When
the last d
ata
is sent, netty channel may be closed immedi
ately and
the elb will not d
iscard
the last // d
ata.
The delay added here
is like SO_LINGER configur
ation. if (log.
isDebugEnabled()) { log.debug("[sid={}] Delay close netty channel when file type", getSid()); } getNettyChannel().eventLoop().schedule(GetRequestChannel.super::doClose,
100, TimeUnit.MILL
ISECONDS); } else { super.doClose(); } } @Override protected void releaseResourceBeforeNettyChannelClosure() { // Call super method super.releaseResourceBeforeNettyChannelClosure(); // As th
is channel has unbounded, un
bindGetRequestChannel() should be avoided. RelayConsts.CloseReason closeReason = inform
ation.getCloseReason(); if (closeReason == RelayConsts.CloseReason.GET_CLOSED_BY_POST || closeReason == RelayConsts.CloseReason.GET_CLOSED_AFTER_LAST_D
ATA_SENT || closeReason == RelayConsts.CloseReason.GET_CLOSED_BY_KILL_SHARER_CMD) { return; } ChannelManager manager = ChannelManager.getInstance(); PostRequestChannel postRequestChannel = (PostRequestChannel) manager.getRequestChannel(
bindingId); if (postRequestChannel !=
null) { postRequestChannel.un
bindGetRequestChannel(th
is); } else { ContainerManager.getInstance().removeFromSingleDirectionContainer(th
is); } }
}
@Slf4j
public class BufferedPostRequestChannel extends DefaultPostRequestChannel {
priv
ate vol
atile IHttpD
ata[] buffedD
ataL
ist; public BufferedPostRequestChannel(String channelId, Channel nettyChannel, String short
BindingId, D
ataProcessSt
ate initD
ataProcessSt
ate, int basicLifeTime, int bufferedNum, PostRequestChannelInform
ation inform
ation) { super(channelId, nettyChannel, short
BindingId, initD
ataProcessSt
ate, basicLifeTime, inform
ation); th
is.buffedD
ataL
ist = new IHttpD
ata[bufferedNum]; } /** * Try to add d
ata into buffer.
The return
value represents
the left number in
the buffer after adding th
is d
ata. If *
the d
ata
is added successfully,
the return
value is non-negtive, especially 0 indic
ates
the buffer
is full after * adding th
is d
ata. If
the d
ata
is failed to be added into buffer when
the buffer
is already full, -
1 will be * return. No m
atter
the d
ata
is added successfully or not,
the d
ata reference will not be modified. * * @return left number in
the buffer after adding th
is d
ata. * <ul> * <li>If
the value is positive,
the buffer still have capcacity after adding th
is d
ata</li> * <li>If
the value is 0,
the buffer
is full after adding th
is d
ata</li> * <li>If
the value is -
1,
the d
ata
is not added into th
is d
ata as
the buffer
is already full</li> * </ul> */ public int addBufferedHttpD
ata(IHttpD
ata bufferedHttpD
ata) { IHttpD
ata[] oldL
ist = th
is.buffedD
ataL
ist; int bufferedNum = oldL
ist.length; int
index = 0; for (;
index < bufferedNum;
index++) { if (oldL
ist[
index] ==
null) { break; } } if (
index < bufferedNum) { IHttpD
ata[] newL
ist = Arrays.copyOf(oldL
ist, bufferedNum); newL
ist[
index] = bufferedHttpD
ata; th
is.buffedD
ataL
ist = newL
ist; return bufferedNum - (
index +
1); } else { return -
1; } } /** * Obtain all buffered d
ata
at th
is time.
The return l
ist will be instanti
ated each time th
is method
is called. * * @return
The buffered d
ata l
ist. */ public L
ist<IHttpD
ata> getAllBufferedD
ata() { IHttpD
ata[] current = th
is.buffedD
ataL
ist; ArrayL
ist<IHttpD
ata> ret = new ArrayL
ist<>(current.length); for (int
index = 0;
index < current.length;
index++) { if (current[
index] !=
null) { ret.add(current[
index]); } else { break; } } return ret; } @Override protected D
ataProcessSt
ate getNextSt
ate(D
ataProcessSt
ate oldSt
ate) { if (oldSt
ate == D
ataProcessSt
ate.WANT_RESPONSE_MOULD) { return D
ataProcessSt
ate.WANT_BUFFERED_D
ATA; } else if (oldSt
ate == D
ataProcessSt
ate.WANT_BUFFERED_D
ATA) { return D
ataProcessSt
ate.TRANSMIT_DIRECT; } else if (oldSt
ate == D
ataProcessSt
ate.TRANSMIT_DIRECT) { return D
ataProcessSt
ate.TRANSMIT_DIRECT; } else { throw new RuntimeException("Bugs in d
ata process st
ate change:" + oldSt
ate); } } @Override protected void releaseResourceAfterNettyChannelClosure() { // release wave header if (log.
isDebugEnabled()) { log.debug("[sid:{}] Release buffered HTTP d
ata", getSid()); } IHttpD
ata[] current = th
is.buffedD
ataL
ist; th
is.buffedD
ataL
ist = new IHttpD
ata[0]; for (IHttpD
ata d
ata : current) { ReferenceCountUtil.safeRelease(d
ata); } if (inform
ation.
isFromDevice()) { log.debug("cloud access channel for {} from device
is closing", get
BindingId()); CloudStreamInfoContainer.getInstance().removeInfo(get
BindingId()); } super.releaseResourceAfterNettyChannelClosure(); }
}
@Slf4j
public class PostRequestChannel extends SourceRequestChannel<SingleDirectionContainer, PostRequestChannelInform
ation> {
/* It
is used for multiple mapping type when social share starts and
the preview of single stream output IPC */ priv
ate String short
BindingId; priv
ate D
ataProcessSt
ate d
ataProcessSt
ate; priv
ate boolean
isNeedResponseAfterGetAccess; priv
ate IHttpResponseGener
ator successResponseGener
ator; priv
ate IFrameHolder iFrameHolder; public PostRequestChannel(String channelId, Channel nettyChannel, String short
BindingId, D
ataProcessSt
ate initD
ataProcessSt
ate, PostRequestChannelInform
ation inform
ation) { super(channelId, nettyChannel, channelId, inform
ation); th
is.short
BindingId = short
BindingId; th
is.d
ataProcessSt
ate = initD
ataProcessSt
ate; if (th
is.getChannelVersion() == ChannelVersion.VERSION_
1_2) { Boolean
isNeedSuccessResponse = inform
ation.
isNeedSuccessResponse(); if (
isNeedSuccessResponse ==
null) { th
is.
isNeedResponseAfterGetAccess = false; th
is.successResponseGener
ator = buildSuccessResponseGener
ator(nettyChannel.alloc()); } else { th
is.
isNeedResponseAfterGetAccess =
isNeedSuccessResponse; th
is.successResponseGener
ator =
isNeedSuccessResponse ? buildSuccessResponseGener
ator(nettyChannel.alloc()) :
null; } } else { th
is.
isNeedResponseAfterGetAccess = true; th
is.successResponseGener
ator = buildSuccessResponseGener
ator(nettyChannel.alloc()); } } priv
ate IHttpResponseGener
ator buildSuccessResponseGener
ator(ByteBufAlloc
ator alloc) { String boundary = ContentType.gener
ateBoundary(); ContentType contentType = new ContentType("multipart/mixed", boundary, ContentType.DEFAULT_D
ATA_CHARSET); return new MultiPartResponseGener
ator(alloc, contentType, HttpUtil.
isTransferEncodingChunked(inform
ation .getOrignalRequest())); } @Override public String getRequestType() { if (getChannelVersion() == ChannelVersion.VERSION_
1_2) { return "relayservice-postStream"; } else if (
isDeviceChannel()) { return "relayservice-deviceStream"; } else { return "relayservice-appStream"; } } public boolean
isMultipleMapping() { return inform
ation.
isMultipleMapping(); } public boolean
isNeedResponseAfterGetAccess() { return
isNeedResponseAfterGetAccess; } public boolean
isSingleStream() { return inform
ation.
isSingleStream(); } public IHttpResponseGener
ator getSuccessResponseGener
ator() { return successResponseGener
ator; } public String getShort
BindingId() { return short
BindingId; } @Override public D
ataProcessSt
ate getD
ataProcessSt
ate() { return d
ataProcessSt
ate; } /** * Change to next st
ate and return
the new st
ate. As th
is method
is called in d
ata processor only,
the st
ate field * only uses <code>volitale</code> instead of <code>
AtomicReference</code> * * @return
The next d
ata process st
ate. */ @Override public D
ataProcessSt
ate changeToNextSt
ate() { assert getNettyChannel().eventLoop().inEventLoop(); D
ataProcessSt
ate st
ate = getNextSt
ate(th
is.d
ataProcessSt
ate); th
is.d
ataProcessSt
ate = st
ate; return st
ate; } protected D
ataProcessSt
ate getNextSt
ate(D
ataProcessSt
ate oldSt
ate) { if (oldSt
ate == D
ataProcessSt
ate.WANT_RESPONSE_MOULD) { return D
ataProcessSt
ate.TRANSMIT_DIRECT; } else if (oldSt
ate == D
ataProcessSt
ate.TRANSMIT_DIRECT) { return D
ataProcessSt
ate.TRANSMIT_DIRECT; } else { throw new RuntimeException("Bugs in d
ata process st
ate change:" + oldSt
ate); } } /** * Initialize
the Post channel with a container.
The Post channel only transmit d
ata to
the Get channel which has *
the same version(except ffmpeg Get channel). If
the Get channel version does not m
atch with
the Post version,
the * Get channel will be response 404/400 and closed. Especially, when
the Get channel
is 1.2 version and Post channel *
is 1.3 version, Get channel will be shown hint video whe next access. * * @param container
The alloc
ated container * * @return Whe
ther
the container has contained any Get channels, if <code>true</code>,
the container contains GET * channels. */ @Override public boolean initContainer(SingleDirectionContainer container) { assert th
is.container ==
null : "initContainer() should only be called once."; synchronized (PostRequestChannel.class) { th
is.container = container; return doInitContainer(container); } } protected boolean doInitContainer(SingleDirectionContainer container) { boolean hasGetChannels = false; Iter
ator<GetRequestChannel> iter = container.boundIter
ator(); while (iter.hasNext()) { GetRequestChannel getChannel = iter.next(); if (getChannel.getChannelVersion() != getChannelVersion() // && !getChannel.
isFFmpegChannel() ) { iter.remove(); inform
ation.addVersionUnm
atchNum(); getChannel.getInform
ation().setVersionUnm
atch(true); if (getChannel.getChannelVersion() == ChannelVersion.VERSION_
1_2) { getChannel.sendHttpResponseAndClose(HttpResponseSt
atus.NOT_FOUND, CloseReason.GET_CLOSED_WITH_LOW_VERSION); } else { getChannel.sendHttpResponseAndClose(HttpResponseSt
atus.BAD_REQUEST, CloseReason.GET_CLOSED_WITH_HIGHT_VERSION); } continue; } if (getChannel.getChannelVersion() == ChannelVersion.VERSION_
1_2) { getChannel.setAllowTransm
ission(); } boolean
isSuccess = getChannel.upd
ateAfterContainerInitializ
ation(th
is); if (
isSuccess) {
isSuccess = doUpd
ate
BindingInform
ation(getChannel); } if (
isSuccess) { hasGetChannels = true; } else { iter.remove(); getChannel.sendHttpResponseAndClose(HttpResponseSt
atus.NOT_FOUND, CloseReason.GET_CLOSED_WHEN_
BIND_FAILURE); } } if (hasGetChannels) { // As Post channel d
ata doesnot arrive or
is buffered in decoder, startTransm
ission of Get channel does // not need to be called. container.synchronizeTransmittedSinkRequestChannelL
ist(); } return hasGetChannels; } /** * Upd
ate Get and Post channel after container initializ
ation. Th
is is called when Get channel
is added into * container which has been initialized by th
is Post channel. * <ul> * <li>For Get channel, using {@link GetRequestChannel#upd
ateAfterContainerInitializ
ation(
ISourceRequestChannel)} to * upd
ate Get channel.</li> * <li>For Post channel, upd
ating
binding inform
ation</li> * </ul> * * @param getRequestChannel
The Get channel which
is added into an initialized container. * * @return True if
the upd
ating oper
ation
is successfully executed. */ public boolean upd
ateAfterContainerInitializ
ation(GetRequestChannel getRequestChannel) { // As th
is method
is called after
isInitContainer() checking,
the container has been initialized. assert container !=
null : "initContainer() or
isInitContainer() should be called first"; SingleDirectionContainer container = th
is.container; synchronized (container) { if (getRequestChannel.getChannelVersion() == ChannelVersion.VERSION_
1_2) { getRequestChannel.startTransm
ission(th
is); } boolean
isSuccess = getRequestChannel.upd
ateAfterContainerInitializ
ation(th
is); if (
isSuccess) {
isSuccess = doUpd
ate
BindingInform
ation(getRequestChannel); } if (!
isSuccess) { container.removeBoundSinkRequestChannel(getRequestChannel); container.synchronizeTransmittedSinkRequestChannelL
ist(); } return
isSuccess; } } /** * Call it after Post channel
is added into channel manager. Th
is method
is used to obtain and process d
ata which
is * stored in decoder, which
is useful to sticky package for version
1.3. */ public void callAfterAddIntoChannelManager() { for (GetRequestChannel getChannel : getBoundGetRequestChannels()) { if (getChannel instanceof ID
ataFollowedRelayRequestChannel) { ((ID
ataFollowedRelayRequestChannel) getChannel).setStartD
ataProcess(); ((ID
ataFollowedRelayRequestChannel) getChannel).enableAutoRead(); ((ID
ataFollowedRelayRequestChannel) getChannel).addPushD
ataTask(); } } } /** *
Bind the given Get channel with th
is Post channel and return
the binding result. If
the Post channel
is closed, * false will be returned.
The Post channel only transmit d
ata to
the Get channel which has
the same version(except * ffmpeg Get channel). If
the Get channel version does not m
atch with
the Post version,
the binding result will be * false. * * @param getRequestChannel
The Get channel needs to be bound with th
is Post channel * * @return
The binding action
is sucessful or not. True if
the Get channel
is successfully bound with th
is Post * channel. False if
the Post channel
is closed and fails to
bind the Get channel. */ public
BindingResult
bindGetRequestChannel(GetRequestChannel getRequestChannel) { // As th
is method
is called after PostRequestChannel being put into channel manager.
the container has been // initialized. assert container !=
null : "initContainer() or
isInitContainer() should be called first"; if (
isClosed()) { return
BindingResult.SOURCE_CHANNEL_CLOSURE; } SingleDirectionContainer container = th
is.container; synchronized (container) { return do
BindGetRequestChannel(container, getRequestChannel); } } protected
BindingResult do
BindGetRequestChannel(SingleDirectionContainer container, GetRequestChannel getRequestChannel) {
BindingResult result = container.addBoundSinkRequestChannel(getRequestChannel); if (result !=
BindingResult.SUCCESS) { return result; } if (getRequestChannel.getChannelVersion() == ChannelVersion.VERSION_
1_2) { getRequestChannel.startTransm
ission(th
is); } if (BooleanUtils.
isTrue(getRequestChannel.getWs())) { if (getRequestChannel instanceof ID
ataFollowedRelayRequestChannel) { ((ID
ataFollowedRelayRequestChannel) getRequestChannel).setStartD
ataProcess(); ((ID
ataFollowedRelayRequestChannel) getRequestChannel).enableAutoRead(); } return
BindingResult.SUCCESS; } boolean
isSuccess = getRequestChannel.upd
ateWhen
Binding(th
is); if (
isSuccess) {
isSuccess = doUpd
ate
BindingInform
ation(getRequestChannel); } if (!
isSuccess) { container.removeBoundSinkRequestChannel(getRequestChannel); container.synchronizeTransmittedSinkRequestChannelL
ist(); } return
isSuccess ?
BindingResult.SUCCESS :
BindingResult.UPD
ATE_INFORM
ATION_FAILURE; } priv
ate boolean doUpd
ate
BindingInform
ation(GetRequestChannel getRequestChannel) { switch (getRequestChannel.getRequestLevel()) { case OWNER: inform
ation.addOwnerNum(); break; case SLAVE: inform
ation.addSlaveNum(); break; case SHARED: inform
ation.addSharedNum(); break; case SOCIAL: inform
ation.addSocialNum(); break; default: break; } return true; } /** * Send pull-stream command when ffmpeg channel arrives and
the Post channel version
is 1.3 */ public void sendPullStreamCommandIfNecessary() { if (getChannelVersion() == ChannelVersion.VERSION_
1_3) { assert successResponseGener
ator !=
null; HttpResponse response = successResponseGener
ator.getHttpResponse(HttpResponseSt
atus.OK); if (response !=
null) { sendHttpResponse(response); } JSONObject command = buildPullStreamCommand(getInform
ation().getParams()); sendD
ata(successResponseGener
ator.getHttpD
ata(command)); } } /** * <p> * Build pull-stream command. * </p> * <p> * In
1.3 version, channel of video type
is 0, and channel of nvr type starts from 0 and has following mapping. * * <pre> * 'channel' in relay request url ----> 'channel' in JSON command * 0 ----> 0 *
1 ---->
1 * ... ----> ... * </pre> * * </p> * * @param params request url parameters. * * @return
The pull-stream command */ priv
ate JSONObject buildPullStreamCommand(Map<String, String> params) { // 实际上设备使用此信息控制推流RESOLUTION int channel = 0; if (Type.NVR.equals(params.get(ParamKey.TYPE)) || Type.SMART_NVR.equals(params.get(ParamKey.TYPE))) { channel = Integer.parseInt(params.get(ParamKey.CHANNEL)); } JSONObject preview = new JSONObject(); preview.put("channels", new JSONArray().put(channel)); String resolution = params.get(ParamKey.RESOLUTION); if (resolution !=
null) { preview.put("resolutions", new JSONArray().put(resolution)); } JSONObject command = new JSONObject(); command.put("type", "request"); command.put("seq", "0"); command.put("params", new JSONObject().put("method", "get").put("preview", preview)); return command; } /** * {@inheritDoc} * * <ul> * <li>For
1.3 version Post channel, as
the terminal will not send d
ata before receiving pull-d
ata command,
the * response and auto read COULD be sent and enabled after Get access.</li> * <li>For
1.2 version Post channel,
the response and auto read SHOULD be sent and enabled after Get access.</li> * </ul> */ @Override public void startD
ataAcceptanceAfterSinkChannelAccess() { // Send success response header if necessary if (
isNeedResponseAfterGetAccess()) { HttpResponse httpResponse = successResponseGener
ator.getHttpResponse(HttpResponseSt
atus.OK); if (httpResponse !=
null) { sendHttpResponse(httpResponse); } } // Start d
ata process setStartD
ataProcess(); // Try to enable auto read of POST channel enableAutoRead(); // Add pull d
ata task addPushD
ataTask(); } /** * Un
bind the given GET channel. If
the GET channel ex
ists in
the binding l
ist, true will be returned. * * @param getRequestChannel
The GET channel needs to be unbound. * * @return True if
the GET channel ex
ists in
the binding l
ist. */ public boolean un
bindGetRequestChannel(GetRequestChannel getRequestChannel) { // As th
is method
is called after PostRequestChannel being put into channel manager.
the container has been // initialized. assert container !=
null : "Should call initContainer() first."; SingleDirectionContainer container = th
is.container; synchronized (container) { boolean
isEx
isted = container.removeBoundSinkRequestChannel(getRequestChannel); if (
isEx
isted) { // If
the ffmpeg channel
is closed explicitly,
the channel will not ex
ist in l
ist. container.synchronizeTransmittedSinkRequestChannelL
ist(); // 取消FFmpeg重试 // if (getRequestChannel.
isFFmpegChannel()) { // doRestartSegmenter(container); // } else { doTryToClose(container, CloseReason.POST_CLOSED_AFTER_GET_EXIT); // } } return
isEx
isted; } } /** * Try to close
the Post channel.
The closure str
ategy
is shown in {@link #doTryToClose(SingleDirectionContainer, * CloseReason)} * * @param closeReason
The specific close reason. */ public void tryToClose(CloseReason closeReason) { SingleDirectionContainer container = th
is.container; if (container ==
null) { return; } synchronized (container) { doTryToClose(container, closeReason); } } /** * Try to close current POST channel after GET channel
is unbound. For single mapping channel,
the POST channel will * be closed immedi
ately if no GET channel
is bound any more. For multiple mapping channel,
the POST channel will be * closed in following cases: * <ul> * <li>When
the bound channel number
is 0,
the segmenter doesn't ex
ist or
is in Segmenter.MAX_RESTART_RETRIES st
ate, * or doesn't have consumer.</li> * <li>When
the bound channel number
is 1,
the segmenter
is in Segmenter.SEGMENTING st
ate and doesn't have consumer</li> * </ul> * * @param closeReason
The reason of closing action */ protected void doTryToClose(SingleDirectionContainer container, CloseReason closeReason) { // Before doTryToClose, container has been checked and could not be
null. if (
isMultipleMapping()) { // int
bindedChannelNum = container.getBoundSinkRequestChannelNum(); // if (
bindedChannelNum == 0) { // Segmenter s = segmenter; // if (s ==
null || s.getSt
ate() == Segmenter.MAX_RESTART_RETRIES || !s.hasConsumer()) { // close(closeReason); // } // } else if (
bindedChannelNum ==
1) { // Segmenter s = segmenter; // if (s !=
null && s.
isSegmenting() && !s.hasConsumer()) { // close(closeReason); // } // } //log.error("
Is multiple mapping"); close(closeReason); } else { if (container.hasNoBoundSinkRequestChannel()) { close(closeReason); } } } /** * Close all GET channels bound with th
is POST channel.
The method
is only used in {@link * #releaseResourceAfterNettyChannelClosure()} */ priv
ate void closeAllGetRequestChannels() { SingleDirectionContainer container = th
is.container; if (container ==
null) { return; } synchronized (container) { Iter
ator<GetRequestChannel> iter = container.boundIter
ator(); while (iter.hasNext()) { GetRequestChannel getChannel = iter.next(); if (log.
isDebugEnabled()) { log.debug("[sid:{}] Close GET Channel: channelId={}", getSid(), getChannel.getChannelId()); } getChannel.close(CloseReason.GET_CLOSED_BY_POST); } container.removeAllBoundSinkeRequestChannels(); container.synchronizeTransmittedSinkRequestChannelL
ist(); } } public L
ist<GetRequestChannel> getBoundGetRequestChannels() { SingleDirectionContainer container = th
is.container; if (container ==
null) { return new ArrayL
ist<>(); } synchronized (container) { return container.getBoundSinkRequestChannels(); } } public int getBoundGetRequestChannelNum() { SingleDirectionContainer container = th
is.container; if (container ==
null) { return 0; } synchronized (container) { return container.getBoundSinkRequestChannelNum(); } } @SuppressWarnings("unchecked") public L
ist<GetRequestChannel> getTransmittedGetRequestChannels() { return (L
ist<GetRequestChannel>) getTransmittedSinkRequestChannels(); } public int getTransmittedGetRequestChannelNum() { SingleDirectionContainer container = th
is.container; if (container ==
null) { return 0; } return container.getTransmittedSinkRequestChannelNum(); } public int upd
ateTransmittedGetRequestChannels() { SingleDirectionContainer container = th
is.container; if (container ==
null) { return 0; } synchronized (container) { return container.synchronizeTransmittedSinkRequestChannelL
ist(); } }
public class ControlRequestChannel extends BaseHttpRequestChannel {
public ControlRequestChannel(String channelId, Channel nettyChannel, ControlRequestChannelInform
ation inform
ation) { super(channelId, nettyChannel, ILifeTimeControl.IDLE_LIFE_TIME_S, inform
ation); } @Override public String getRequestType() { String cmdMethod = inform
ation.getCmdMethod(); return cmdMethod !=
null ? RelayConsts.Service.CONTROL + '-' + cmdMethod : RelayConsts.Service.CONTROL; }
}
博客指出数据库查询报错原因，在数据库中查询name为name1的数据时，name1未赋值为空，导致查询出现问题。
扫一扫
3610
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
