/* Ċ Ġ* ĠCopyright Ġ 2 0 1 7 ĠThe ĠHy ve Ċ Ġ* Ċ Ġ* ĠLicensed Ġunder Ġthe ĠApache ĠLicense , ĠVersion Ġ 2 . 0 Ġ( the Ġ" License "); Ċ Ġ* Ġyou Ġmay Ġnot Ġuse Ġthis Ġfile Ġexcept Ġin Ġcompliance Ġwith Ġthe ĠLicense . Ċ Ġ* ĠYou Ġmay Ġobtain Ġa Ġcopy Ġof Ġthe ĠLicense Ġat Ċ Ġ* Ċ Ġ* Ġhttp :// www . apache . org / licenses / LICENSE - 2 . 0 Ċ Ġ* Ċ Ġ* ĠUnless Ġrequired Ġby Ġapplicable Ġlaw Ġor Ġagreed Ġto Ġin Ġwriting , Ġsoftware Ċ Ġ* Ġdistributed Ġunder Ġthe ĠLicense Ġis Ġdistributed Ġon Ġan Ġ" AS ĠIS " ĠBASIS , Ċ Ġ* ĠWITHOUT ĠWARRANTIES ĠOR ĠCONDITIONS ĠOF ĠANY ĠKIND , Ġeither Ġexpress Ġor Ġimplied . Ċ Ġ* ĠSee Ġthe ĠLicense Ġfor Ġthe Ġspecific Ġlanguage Ġgoverning Ġpermissions Ġand Ċ Ġ* Ġlimitations Ġunder Ġthe ĠLicense . Ċ Ġ*/ Ċ Ċ package Ġorg . radar base . android . kafka Ċ Ċ import Ġandroid . os . System Clock Ċ import Ġorg . radar base . android . util . Delayed Retry Ċ import Ġorg . radar base . android . util . Safe Handler Ċ import Ġorg . radar base . producer . Authentication Exception Ċ import Ġorg . radar base . producer . Kafka Sender Ċ import Ġorg . slf 4 j . LoggerFactory Ċ import Ġjava . util . concurrent . atomic . Atomic Boolean Ċ Ċ /** Ċ Ġ* ĠChecks Ġthe Ġconnection Ġof Ġa Ġsender . ĠIt Ġdoes Ġso Ġusing Ġtwo Ġmechanisms : Ġa Ġregular Ċ Ġ* Ġheartbeat Ġsignal Ġwhen Ġthe Ġconnection Ġis Ġassumed Ġto Ġbe Ċ Ġ* Ġpresent , Ġand Ġa Ġexponential Ġback - off Ġmechanism Ġif Ġthe Ġconnection Ġis Ġsever ed . ĠIf Ġthe Ġconnection Ġis Ċ Ġ* Ġas ses sed Ġto Ġbe Ġpresent Ġthrough Ġanother Ġmechanism , Ġ[ did Connect ] Ġshould Ġbe Ġcalled , Ċ Ġ* Ġcon verse ly , Ġif Ġit Ġis Ġas ses sed Ġto Ġbe Ġsever ed , Ġ[ did Disconnect ] Ġshould Ġbe Ċ Ġ* Ġcalled . Ċ Ġ*/ Ċ internal Ġclass ĠKafka Connection Checker ( private Ġval Ġsender : ĠKafka Sender , ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġprivate Ġval Ġm Handler : ĠSafe Handler , ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġprivate Ġval Ġlistener : ĠServer Status Listener , ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġheartbeat Seconds Interval : ĠLong ) Ġ{ ĊĠĠĠ Ġprivate Ġval Ġis Connected Backing : ĠAtomic Boolean Ġ= ĠAtomic Boolean ( false ) ĊĠĠĠ Ġprivate Ġvar Ġfuture : ĠSafe Handler . Handler Future ? Ġ= Ġnull ĊĠĠĠ Ġprivate Ġval Ġheartbeat Interval : ĠLong Ġ= Ġheartbeat Seconds Interval Ġ* Ġ 1 0 0 0 L ĊĠĠĠ Ġprivate Ġval Ġretry Delay Ġ= ĠDelay ed Retry ( INCREMENT AL _ BACK OFF _ MILLISECONDS , ĠMAX _ BACK OFF _ MILLISECONDS ) ĊĠĠĠ Ġprivate Ġvar Ġlast Connection : ĠLong Ġ= Ġ- 1 L ĊĊĠĠĠ Ġval Ġis Connected : ĠBoolean ĊĠĠĠĠĠĠĠ Ġget () Ġ= Ġis Connected Backing . get () ĊĊĠĠĠ Ġinit Ġ{ ĊĠĠĠĠĠĠĠ Ġm Handler . execute Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġif Ġ( sender . isConnected ) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġis Connected Backing . set ( false ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġdid Connect () ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ} Ġelse Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġis Connected Backing . set ( true ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġdid Disconnect ( null ) ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠ Ġ} ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġ/** ĊĠĠĠĠ Ġ* ĠCheck Ġwhether Ġthe Ġconnection Ġwas Ġclosed Ġand Ġtry Ġto Ġreconnect . ĊĠĠĠĠ Ġ*/ ĊĠĠĠ Ġprivate Ġfun Ġmake Check () Ġ{ ĊĠĠĠĠĠĠĠ Ġtry Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġif Ġ(! isConnected ) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġif Ġ( sender . reset Connection ()) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġdid Connect () ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġlistener . update Server Status ( Server Status Listener . Status . CONNECTED ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġlogger . info (" Sender Ġre connected ") ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ} Ġelse Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġretry () ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ} Ġelse Ġif Ġ( System Clock . uptime Millis () Ġ- Ġlast Connection Ġ> Ġ 1 5 _ 0 0 0 L ) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġif Ġ( sender . isConnected Ġ|| Ġsender . reset Connection ()) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġdid Connect () ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ} Ġelse Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġdid Disconnect ( null ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠ Ġ} Ġcatch Ġ( ex : ĠAuthentication Exception ) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġdid Disconnect ( ex ) ĊĠĠĠĠĠĠĠ Ġ} ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġ/** ĠCheck Ġthe Ġconnection Ġas Ġsoon Ġas Ġpossible . Ġ Ġ*/ ĊĠĠĠ Ġfun Ġcheck () Ġ{ ĊĠĠĠĠĠĠĠ Ġm Handler . execute (:: make Check ) ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġ/** ĠRetry Ġthe Ġconnection Ġwith Ġan Ġincremental Ġbackoff . Ġ Ġ*/ ĊĠĠĠ Ġprivate Ġfun Ġretry () Ġ{ ĊĠĠĠĠĠĠĠ Ġfuture Ġ= Ġm Handler . delay ( retry Delay . next Delay (), Ġ:: make Check ) ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġ/** ĠSignal Ġthat Ġthe Ġsender Ġsuccessfully Ġconnected . Ġ Ġ*/ ĊĠĠĠ Ġfun Ġdid Connect () Ġ{ ĊĠĠĠĠĠĠĠ Ġm Handler . execute Re entrant Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġlast Connection Ġ= ĠSystem Clock . uptime Millis () ĊĠĠĠĠĠĠĠĠĠĠĠ Ġif Ġ( isConnected Backing . compare AndSet ( false , Ġtrue )) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġfuture ?. cancel () ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġfuture Ġ= Ġm Handler . repeat ( heartbeat Interval , Ġ:: make Check ) ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠĠĠĠĠ Ġretry Delay . reset () ĊĠĠĠĠĠĠĠ Ġ} ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġ/** ĊĠĠĠĠ Ġ* ĠSignal Ġthat Ġthe ĠKafka ĠREST Ġsender Ġhas Ġdisconnected . ĊĠĠĠĠ Ġ* Ġ@ param Ġex Ġexception Ġthe Ġsender Ġdisconnected Ġwith , Ġmay Ġbe Ġnull ĊĠĠĠĠ Ġ*/ ĊĠĠĠ Ġfun Ġdid Disconnect ( ex : ĠException ?) Ġ{ ĊĠĠĠĠĠĠĠ Ġm Handler . execute Re entrant Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġlogger . warn (" Sender Ġis Ġdisconnected ", Ġex ) ĊĊĠĠĠĠĠĠĠĠĠĠĠ Ġif Ġ( isConnected Backing . compare AndSet ( true , Ġfalse )) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġfuture ?. cancel () ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġfuture Ġ= Ġm Handler . delay ( INCREMENT AL _ BACK OFF _ MILLISECONDS , Ġ:: make Check ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġif Ġ( ex Ġis ĠAuthentication Exception ) Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġlogger . warn (" Failed Ġto Ġauthenticate Ġto Ġserver : Ġ{}", Ġex . message ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġlistener . update Server Status ( Server Status Listener . Status . UN AUTHORIZ ED ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ} Ġelse Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġlistener . update Server Status ( Server Status Listener . Status . DIS CONNECTED ) ĊĠĠĠĠĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠĠĠĠĠ Ġ} ĊĠĠĠĠĠĠĠ Ġ} ĊĠĠĠ Ġ} ĊĊĠĠĠ Ġcompanion Ġobject Ġ{ ĊĠĠĠĠĠĠĠ Ġprivate Ġval Ġlogger Ġ= ĠLoggerFactory . getLogger ( Kafka Connection Checker :: class . java ) ĊĊĠĠĠĠĠĠĠ Ġprivate Ġconst Ġval ĠIN CRE MENT AL _ BACK OFF _ MILLISECONDS Ġ= Ġ 6 0 Ġ* Ġ 1 0 0 0 L Ġ// Ġ 1 Ġminute ĊĠĠĠĠĠĠĠ Ġprivate Ġconst Ġval ĠMAX _ BACK OFF _ MILLISECONDS Ġ= Ġ 4 Ġ* Ġ 6 0 Ġ* Ġ 6 0 Ġ* Ġ 1 0 0 0 L Ġ// Ġ 4 Ġhours ĊĠĠĠ Ġ} Ċ } Ċ