improve analysis server notifications

pkg/analysis_server/lib/src/domain_server.dart

 // Copyright (c) 2014, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 library domain.server;
 
 import 'package:analysis_server/src/analysis_server.dart';
 import 'package:analysis_server/src/protocol.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/java_io.dart';
@@ skipping 93 matching lines @@
       // TODO what error code should be returned here?
       return new Response(request.id, new RequestError(
           RequestError.CODE_SDK_ERROR, 'Failed to access sdk: $e'));
     }
     context.sourceFactory = new SourceFactory([
       new DartUriResolver(sdk),
       new FileUriResolver(),
       // new PackageUriResolver(),
     ]);
     server.contextMap[contextId] = context;
+    server.contextIdMap[context] = contextId;
 
     Response response = new Response(request.id);
     return response;
   }
 
   /**
    * Delete the context with the given id. Future attempts to use the context id
    * will result in an error being returned.
    */
   Response deleteContext(Request request) {
     String contextId = request.getRequiredParameter(AnalysisServer.CONTEXT_ID_PARAM).asString();
 
     AnalysisContext removedContext = server.contextMap.remove(contextId);
     if (removedContext == null) {
       return new Response.contextDoesNotExist(request);
     }
+    server.contextIdMap.remove(removedContext);
     Response response = new Response(request.id);
     return response;
   }
 
   /**
    * Cleanly shutdown the analysis server.
    */
   Response shutdown(Request request) {
     server.running = false;
     Response response = new Response(request.id);
     return response;
   }
 
   /**
    * Return the version number of the analysis server.
    */
   Response version(Request request) {
     Response response = new Response(request.id);
     response.setResult(VERSION_RESULT, '0.0.1');
     return response;
   }
 }