From: Joaquim Rocha <joaquim.rocha@cern.ch>
Date: Thu, 4 Jun 2015 14:23:55 +0000 (+0200)
Subject: cls_numops: Add cls_numops client
X-Git-Tag: v9.1.0~255^2~2
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=d17f158ef312d066f02587c10243cde6373288f2;p=ceph.git

cls_numops: Add cls_numops client

Signed-off-by: Joaquim Rocha <joaquim.rocha@cern.ch>
---

diff --git a/src/cls/numops/cls_numops.cc b/src/cls/numops/cls_numops.cc
index 37464592bbd..a632e8ecffa 100644
--- a/src/cls/numops/cls_numops.cc
+++ b/src/cls/numops/cls_numops.cc
@@ -27,6 +27,8 @@
 #include <sstream>
 #include <cstdio>
 
+#define DECIMAL_PRECISION 10
+
 CLS_VER(1,0)
 CLS_NAME(numops)
 
@@ -81,9 +83,7 @@ static int add(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
   value += difference;
 
   std::stringstream stream;
-
-  stream.str("");
-  stream << std::setprecision(10) << value;
+  stream << std::setprecision(DECIMAL_PRECISION) << value;
 
   bufferlist new_value;
   new_value.append(stream.str());
@@ -138,9 +138,7 @@ static int mul(cls_method_context_t hctx, bufferlist *in, bufferlist *out)
   value *= difference;
 
   std::stringstream stream;
-
-  stream.str("");
-  stream << std::setprecision(10) << value;
+  stream << std::setprecision(DECIMAL_PRECISION) << value;
 
   bufferlist new_value;
   new_value.append(stream.str());
diff --git a/src/cls/numops/cls_numops_client.cc b/src/cls/numops/cls_numops_client.cc
new file mode 100644
index 00000000000..68928206335
--- /dev/null
+++ b/src/cls/numops/cls_numops_client.cc
@@ -0,0 +1,80 @@
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2015 CERN
+ *
+ * Author: Joaquim Rocha <joaquim.rocha@cern.ch>
+ *
+ *  This library is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public
+ *  License as published by the Free Software Foundation; either
+ *  version 2.1 of the License, or (at your option) any later version.
+ *
+ */
+
+#include "objclass/objclass.h"
+#include "cls/numops/cls_numops_client.h"
+#include "include/encoding.h"
+
+#include <cstdlib>
+#include <errno.h>
+#include <sstream>
+
+namespace rados {
+  namespace cls {
+    namespace numops {
+
+      int add(librados::IoCtx *ioctx,
+              const std::string& oid,
+              const std::string& key,
+              double value_to_add)
+      {
+        bufferlist in, out;
+        ::encode(key, in);
+
+        std::stringstream stream;
+        stream << value_to_add;
+
+        ::encode(stream.str(), in);
+
+        return ioctx->exec(oid, "numops", "add", in, out);
+      }
+
+      int sub(librados::IoCtx *ioctx,
+              const std::string& oid,
+              const std::string& key,
+              double value_to_subtract)
+      {
+        return add(ioctx, oid, key, -value_to_subtract);
+      }
+
+      int mul(librados::IoCtx *ioctx,
+              const std::string& oid,
+              const std::string& key,
+              double value_to_multiply)
+      {
+        bufferlist in, out;
+        ::encode(key, in);
+
+        std::stringstream stream;
+        stream << value_to_multiply;
+
+        ::encode(stream.str(), in);
+
+        return ioctx->exec(oid, "numops", "mul", in, out);
+      }
+
+      int div(librados::IoCtx *ioctx,
+              const std::string& oid,
+              const std::string& key,
+              double value_to_divide)
+      {
+        if (value_to_divide == 0)
+          return -EINVAL;
+
+        return mul(ioctx, oid, key, 1 / value_to_divide);
+      }
+
+    } // namespace numops
+  } // namespace cls
+} // namespace rados
diff --git a/src/cls/numops/cls_numops_client.h b/src/cls/numops/cls_numops_client.h
new file mode 100644
index 00000000000..8d776bffcaf
--- /dev/null
+++ b/src/cls/numops/cls_numops_client.h
@@ -0,0 +1,49 @@
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2015 CERN
+ *
+ * Author: Joaquim Rocha <joaquim.rocha@cern.ch>
+ *
+ *  This library is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public
+ *  License as published by the Free Software Foundation; either
+ *  version 2.1 of the License, or (at your option) any later version.
+ *
+ */
+
+#ifndef CEPH_LIBRBD_CLS_NUMOPS_CLIENT_H
+#define CEPH_LIBRBD_CLS_NUMOPS_CLIENT_H
+
+#include "include/rados/librados.hpp"
+
+namespace rados {
+  namespace cls {
+    namespace numops {
+
+      extern int add(librados::IoCtx *ioctx,
+                     const std::string& oid,
+                     const std::string& key,
+                     double value_to_add);
+
+      extern int sub(librados::IoCtx *ioctx,
+                     const std::string& oid,
+                     const std::string& key,
+                     double value_to_subtract);
+
+      extern int mul(librados::IoCtx *ioctx,
+                     const std::string& oid,
+                     const std::string& key,
+                     double value_to_multiply);
+
+      extern int div(librados::IoCtx *ioctx,
+                     const std::string& oid,
+                     const std::string& key,
+                     double value_to_divide);
+
+    } // namespace numops
+  } // namespace cls
+} // namespace rados
+
+#endif // CEPH_LIBRBD_CLS_NUMOPS_CLIENT_H
+