Shaft heat treatment equipment