Support setting state in StateServlet.

1 files changed, 17 insertions, 0 deletions

diff --git a/same/src/main/java/com/orbekk/same/Client.java b/same/src/main/java/com/orbekk/same/Client.java
index 2a5f11e..60faf1b 100644
--- a/same/src/main/java/com/orbekk/same/Client.java
+++ b/same/src/main/java/com/orbekk/same/Client.java
@@ -28,6 +28,23 @@ public class Client implements DiscoveryListener {
         public State getState() {
             return new State(state);
         }
+        
+        public void set(String name, String data) throws UpdateConflict {
+            String masterUrl = state.getDataOf(".masterUrl");
+            long revision = state.getRevision(name) + 1;
+            MasterService master = connections.getMaster(masterUrl);
+            try {
+                boolean success = master.updateStateRequest(name, data,
+                        revision);
+                if (!success) {
+                    throw new UpdateConflict("State update conflict when " +
+                            "updating " + name);
+                }
+            } catch (Exception e) {
+                logger.error("Unable to contact master. Update fails.", e);
+                throw new UpdateConflict("Unable to contact master. Update fails.");
+            }
+        }
     }
     
     private ClientInterface clientInterface = new ClientInterface();