He was elected Polk County attorney in 1976, serving from 1977 to 1985, when he decided to resign and move to New York City.
Johnston's highest profile case as a lawyer was Tinker v. Des Moines, a case about the free speech rights of two Des Moines public school students who wore black armbands to protest the Vietnam War and were subsequently suspended by school administrators.