Не случайно, что из всех боевых групп, созданных в России, 1-я оставалась первой до конца войны и по заслугам, и по количеству побед.
В апреле – мае 1917-го ротмистр Козаков ведёт весьма активную боевую работу. Свои очередные победы он одерживает, как правило, в паре или в группе. Замечательные успехи не остаются не замеченными командованием. Ему предлагают возглавить школу воздушного боя в Евпатории. Но Козаков от заманчивого предложения отказывается:
«На должность начальника школы воздушного боя очень прошу не назначать. Хочу быть только в группе. Благодарю за предложение. Ротмистр Козаков».
14 (27) июня 1917 г. в двух вылетах в паре Александр Александрович одерживает ещё две победы. Однако впервые получает ранение четырьмя пулями в руку, с повреждением локтевого сустава. Командира группы срочно эвакуируют в тыл.
В том же 1917-м Козакова награждают ещё двумя боевыми орденами. 21 апреля – орденом Св. Анны 2-й степени с мечами «за воздушные разведки под г. Двинском в январе-феврале 1916 г.» и орденом Св. Георгия 4-й степени от 31 июля «за то, что, будучи в чине штабс-ротмистра, 8-го декабря 1916 г., вылетев на самолёте “ньюпор”, типа истребителей, для преследования немецких самолётов, появляющихся над нашим расположением, отогнал в районе Горохова 2 немцев, атаковал третий самолёт в районе д. Забороль, в 5 верстах к западу от Луцка, и двумя пулями в голову убил пилота. Аппарат, кое-как управляемый легко раненым наблюдателем австрийским обер-лейтенантом, упал в нашем расположении и был захвачен нашими войсками».
8 сентября следует производство в чин подполковника. Но ни чины, ни награды, ни слава не меняют этого человека. Он продолжает летать, как делал это всегда, до самых последних дней Российской империи.
Как указано в послужном списке Козакова, Александр Александрович с 24 ноября 1917 г. временно исполняет должность командира 7-го авиационного дивизиона, с оставлением в должности командира 1-й боевой авиационной группы Юго-Западного фронта. Лишь 9 декабря 1917 г. его отстраняют от должности командира 1-й боевой авиационной группы Юго-Западного фронта и избирают на отрядном совете командиром 19-го корпусного авиационного отряда.
Известно, что после Октябрьской революции новая власть пыталась привлечь подполковника Козакова к работе по созданию Рабочее-Крестьянского воздушного флота. Александр Александрович даже встречался с наркомом по военным делам Л. Д. Троцким в его кремлёвском кабинете. В беседе он вместе со своими коллегами «предложил советскому правительству своё содействие в деле возрождения российского воздушного флота».
Однако Лев Давидович предпочёл опереться на тех лётчиков, кто выразил активную поддержку новой власти сразу, а не потом. Таким образом, хотя кандидатура Козакова и рассматривалась на должность начальника Главного управления Военно-воздушного флота, всё же его прикомандировали лишь к комиссии ГУВВФ по выработке штатов и положений о службе истребительных авиаотрядов.
«Пребывание в Петрограде, встречи с новыми руководителями авиации, предлагавшими Козакову достойный его незаурядного таланта командный пост, не вразумили лётчика, – пишет Ю. М. Гальперин. – Верх взяли старые, привычные связи…
В Петрограде, где затаились, ожидая краха большевиков, немало офицеров, Козакова всячески остерегали от “перехода” в стан погубителей отечества». <…>
И его же славой, его героизмом объясняли Козакову невозможность перехода на службу “предателям России”. Разобраться, кто заблуждается или просто лжёт, он оказался не в состоянии. Оставалось положиться на авторитет уважаемых им людей своего круга, в том числе нескольких бывших начальников. А тут ещё появился школьный друг, гатчинец Модрах, тоже командовавший на фронте авиационной группой. Он-то и свёл Козакова с эмиссарами англичан, начавшими на севере свою интервенцию.