Kafka reprezentoval Česko na všech devíti Mistrovstvích světa mezi roky 2000 a 2016. Je tak s odstupem českým hráčem s nejvyšším počtem účastí. Je jediným hráčem, který získal všechny tři dosavadní české medaile. Na Mistrovství světa v roce 2010 byl zvolen do All Stars teamu. Zároveň byl i u největšího českého neúspěchu na Mistrovství světa v roce 2012.