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