Add those lines in vs code settings.json:

"editor.codeActionsOnSave": {
    "source.organizeImports": true
}