Военную подготовку проходил в составе 2 учебного стрелкового полка и  66 учебного полка отличных стрелков снайперской подготовки 11 учебной стрелковой бригады (станция Инза). 

В начале 1944г. младший сержант Козьяков  в составе маршевой роты 12835 направлен в действующую армию.

К июню 1944г. номер расчета противотанкового ружья (ПТР) роты ПТР 1 мотострелкового батальона 8 мотострелковой бригады 9 танкового корпуса. Командир 1 мотострелкового батальона- старший лейтенант Сергеев Александр Степанович, командир взвода ПТР младший лейтенант Бельский Николай Васильевич.

10 июня получен приказ- передислоцироваться на восточный берег р.Друть севернее г.Жлобин.

До 23 июня продолжалось изучение переднего края противника, велась подготовка по вводу бригады в  прорыв с задачей- отрезать пути отхода Жлобинской группировки врага на Бобруйск.

В ночь на 26 июня бригада с 218 минометным полком, 1508 самоходно-артиллерийским полком, 286 стрелковым гвардейским минометным дивизионом вошла в прорыв за 95 танковой бригадой и 216 зенитно-артиллерийским полком в направлении Осовик, Борчица, Капустино, Старим, Титовка с задачей- не допустить отхода противника к Бобруйску и захватить переправу через р.Березина в районе Титовки. К полуночи части бригады заняли круговую оборону.

27 и в ночь на 28 июня бригада отражала контратаки разбитых частей 6, 57, 134, 283 пехотных дивизий и 236 кавалерийской дивизии.

В боях младший сержант Козьяков из ружья ПТР уничтожил станковый пулемет и 2 ручных пулеметов.

Был представлен к медали «За отвагу», приказ о награждении от 30 июля 1944г.

28 июня бригада переправилась на западный берег р.Березина и вела наступление в направлении Минска.

В ночь на 29 июня колонна бригады была атакована отрядом, прорвавшихся их окружения гитлеровцев, численностью до полка. В бою было уничтожено до 800 солдат и офицеров противника. С 12-00 бригада занимала круговую оборону в лесу 2км северо-восточнее Горожа. 

30 июня двумя батальонами вела наступление вдоль железной дороги на станцию Свислочь.

01 июля части бригады выступили в направлении Шищицы.

02 июля были освобождены Валерьяны.

03 июля бригада овладела Даниловичи, вела бои за Добринёво.

Во время отражение контратаки противника в 4км. западнее д.Озеро без вести пропал младший сержант Козьяков.