Cosmetic improvements.

[invirt/third/libt4.git] / lock_smain.cc

diff --git a/lock_smain.cc b/lock_smain.cc
index d62a25b..2c9828b 100644 (file)
--- a/lock_smain.cc
+++ b/lock_smain.cc
@@ -1,10 +1,6 @@
-#include "rpc/rpc.h"
+#include "lock_server.h"
 #include <arpa/inet.h>
-#include <stdlib.h>
-#include "threaded_log.h"
 #include <unistd.h>
-#include "lock_server.h"
-#include "rsm.h"
 
 // Main loop of lock_server
 
@@ -17,17 +13,14 @@ int main(int argc, char *argv[]) {
     srandom((uint32_t)getpid());
 
     if(argc != 3){
-        fprintf(stderr, "Usage: %s [master:]port [me:]port\n", argv[0]);
+        LOG_NONMEMBER("Usage: " << argv[0] << " [master:]port [me:]port");
         exit(1);
     }
 
     rsm rsm(argv[1], argv[2]);
-    lock_server ls(&rsm);
-    rsm.set_state_transfer(&ls);
+    lock_server ls(rsm);
 
-    rsm.reg(lock_protocol::acquire, &lock_server::acquire, &ls);
-    rsm.reg(lock_protocol::release, &lock_server::release, &ls);
-    rsm.reg(lock_protocol::stat, &lock_server::stat, &ls);
+    rsm.start();
 
     while(1)
         sleep(1000);