Handle write sessions

For write sessions we throw a SessionExpiredException on connection failures

Author: pontusmelke

driver/src/main/java/org/neo4j/driver/internal/ClusterDriver.java

@@ -40,7 +40,7 @@
 
 public class ClusterDriver extends BaseDriver
 {
-    private static final String DISCOVER_MEMBERS = "dbms.cluster.discoverMembers";
+    private static final String DISCOVER_MEMBERS = "dbms.cluster.discoverEndpointAcquisitionServers";
     private static final String ACQUIRE_ENDPOINTS = "dbms.cluster.acquireEndpoints";
 
     private final Endpoints endpoints = new Endpoints();
@@ -210,7 +210,7 @@ public void accept( Connection connection )
                 }
             }, clusterSettings, log );
         case WRITE:
-            throw new UnsupportedOperationException();
+            return new WriteNetworkSession( acquireConnection( mode ), clusterSettings, log );
         default:
             throw new UnsupportedOperationException();
         }
@@ -237,7 +237,7 @@ private synchronized Connection acquireConnection( SessionMode mode )
                 @Override
                 public void accept( Record record )
                 {
-                    String serverMode = record.get( "mode" ).asString();
+                    String serverMode = record.get( "role" ).asString();
                     if ( serverMode.equals( "READ" ) )
                     {
                         endpoints.readServer = new BoltServerAddress( record.get( "address" ).asString() );

driver/src/main/java/org/neo4j/driver/internal/WriteNetworkSession.java

@@ -0,0 +1,50 @@
+/**
+ * Copyright (c) 2002-2016 "Neo Technology,"
+ * Network Engine for Objects in Lund AB [http://neotechnology.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.neo4j.driver.internal;
+
+
+import org.neo4j.driver.internal.spi.Connection;
+import org.neo4j.driver.v1.Logger;
+import org.neo4j.driver.v1.Statement;
+import org.neo4j.driver.v1.StatementResult;
+import org.neo4j.driver.v1.exceptions.ConnectionFailureException;
+import org.neo4j.driver.v1.exceptions.SessionExpiredException;
+
+public class WriteNetworkSession extends NetworkSession
+{
+
+    WriteNetworkSession(Connection connection, ClusterSettings clusterSettings, Logger logger )
+    {
+        super(connection, logger);
+    }
+
+    @Override
+    public StatementResult run( Statement statement )
+    {
+        try
+        {
+            return super.run( statement );
+        }//TODO we need to catch exceptions due to leader switches etc here
+        catch ( ConnectionFailureException e )
+        {
+            throw new SessionExpiredException( "Failed to perform write load to server", e );
+        }
+
+    }
+}

driver/src/main/java/org/neo4j/driver/v1/exceptions/SessionExpiredException.java

@@ -0,0 +1,33 @@
+/**
+ * Copyright (c) 2002-2016 "Neo Technology,"
+ * Network Engine for Objects in Lund AB [http://neotechnology.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.neo4j.driver.v1.exceptions;
+
+/**
+ * A <em>SessionExpiredException</em> indicates that the session can no longer satisfy the criteria under which it
+ * was acquired, e.g. a server no longer accepts write requests. A new session needs to be acquired from the driver
+ * and all actions taken on the expired session must be replayed.
+ * @since 1.1
+ */
+public class SessionExpiredException extends Neo4jException
+{
+    public SessionExpiredException( String message, Throwable throwable )
+    {
+        super( message, throwable );
+    }
+}