diff options
Diffstat (limited to 'xlators/mgmt/glusterd/src')
| -rw-r--r-- | xlators/mgmt/glusterd/src/glusterd-op-sm.c | 82 | ||||
| -rw-r--r-- | xlators/mgmt/glusterd/src/glusterd-rebalance.c | 82 | ||||
| -rw-r--r-- | xlators/mgmt/glusterd/src/glusterd-rpc-ops.c | 28 | 
3 files changed, 115 insertions, 77 deletions
diff --git a/xlators/mgmt/glusterd/src/glusterd-op-sm.c b/xlators/mgmt/glusterd/src/glusterd-op-sm.c index 3c1baa7e6..bda5e61e4 100644 --- a/xlators/mgmt/glusterd/src/glusterd-op-sm.c +++ b/xlators/mgmt/glusterd/src/glusterd-op-sm.c @@ -2709,6 +2709,83 @@ out:          return ret;  } + +int +glusterd_defrag_volume_node_rsp (dict_t *req_dict, dict_t *rsp_dict, +                                 dict_t *op_ctx) +{ +        int                             ret = 0; +        char                            *volname = NULL; +        glusterd_volinfo_t              *volinfo = NULL; +        uint64_t                        files = 0; +        uint64_t                        size = 0; +        uint64_t                        lookup = 0; +        gf_defrag_status_t              status = GF_DEFRAG_STATUS_NOT_STARTED; + +        GF_ASSERT (req_dict); + +        ret = dict_get_str (req_dict, "volname", &volname); +        if (ret) { +                gf_log ("", GF_LOG_ERROR, "Unable to get volume name"); +                goto out; +        } + +        ret  = glusterd_volinfo_find (volname, &volinfo); + +        if (ret) +                goto out; + +        ret = dict_get_uint64 (rsp_dict, "files", &files); +        if (ret) +                gf_log (THIS->name, GF_LOG_TRACE, +                        "failed to get file count"); + +        ret = dict_get_uint64 (rsp_dict, "size", &size); +        if (ret) +                gf_log (THIS->name, GF_LOG_TRACE, +                        "failed to get size of xfer"); + +        ret = dict_get_uint64 (rsp_dict, "lookups", &lookup); +        if (ret) +                gf_log (THIS->name, GF_LOG_TRACE, +                        "failed to get lookedup file count"); +        ret = dict_get_int32 (rsp_dict, "status", (int32_t *)&status); +        if (ret) +                gf_log (THIS->name, GF_LOG_TRACE, +                        "failed to get status"); + +        volinfo->rebalance_files += files; +        volinfo->rebalance_data += size; +        volinfo->lookedup_files += lookup; + +        if (!op_ctx) { +                dict_copy (rsp_dict, op_ctx); +                goto out; +        } + +        ret = dict_set_uint64 (op_ctx, "files", volinfo->rebalance_files); +        if (ret) +                gf_log (THIS->name, GF_LOG_ERROR, +                        "failed to set file count"); + +        ret = dict_set_uint64 (op_ctx, "size", volinfo->rebalance_data); +        if (ret) +                gf_log (THIS->name, GF_LOG_ERROR, +                        "failed to set size of xfer"); + +        ret = dict_set_uint64 (op_ctx, "lookups", volinfo->lookedup_files); +        if (ret) +                gf_log (THIS->name, GF_LOG_ERROR, +                        "failed to set lookedup file count"); +        ret = dict_set_int32 (op_ctx, "status", status); +        if (ret) +                gf_log (THIS->name, GF_LOG_ERROR, +                        "failed to set status"); + +out: +        return ret; +} +  int32_t  glusterd_handle_node_rsp (glusterd_req_ctx_t *req_ctx, void *pending_entry,                            glusterd_op_t op, dict_t *rsp_dict, dict_t *op_ctx, @@ -2732,8 +2809,9 @@ glusterd_handle_node_rsp (glusterd_req_ctx_t *req_ctx, void *pending_entry,                  break;          case GD_OP_DEFRAG_BRICK_VOLUME: -                dict_copy (rsp_dict, op_ctx); -        break; +                glusterd_defrag_volume_node_rsp (req_ctx->dict, +                                                 rsp_dict, op_ctx); +                break;          case GD_OP_HEAL_VOLUME:                  ret = glusterd_heal_volume_brick_rsp (req_ctx->dict, rsp_dict, diff --git a/xlators/mgmt/glusterd/src/glusterd-rebalance.c b/xlators/mgmt/glusterd/src/glusterd-rebalance.c index f7304b9c0..936a3b26e 100644 --- a/xlators/mgmt/glusterd/src/glusterd-rebalance.c +++ b/xlators/mgmt/glusterd/src/glusterd-rebalance.c @@ -82,60 +82,6 @@ out:          return ret;  } -int -glusterd_defrag_status_get (glusterd_volinfo_t *volinfo, -                            dict_t *dict) -{ -        int      ret    = 0; -        uint64_t files  = 0; -        uint64_t size   = 0; -        uint64_t lookup = 0; - -        if (!volinfo || !dict) -                goto out; - -        ret = 0; -        if (volinfo->defrag_status == GF_DEFRAG_STATUS_NOT_STARTED) -                goto out; - -        if (volinfo->defrag) { -                LOCK (&volinfo->defrag->lock); -                { -                        files  = volinfo->defrag->total_files; -                        size   = volinfo->defrag->total_data; -                        lookup = volinfo->defrag->num_files_lookedup; -                } -                UNLOCK (&volinfo->defrag->lock); -        } else { -                files  = volinfo->rebalance_files; -                size   = volinfo->rebalance_data; -                lookup = volinfo->lookedup_files; -        } - -        ret = dict_set_uint64 (dict, "files", files); -        if (ret) -                gf_log (THIS->name, GF_LOG_WARNING, -                        "failed to set file count"); - -        ret = dict_set_uint64 (dict, "size", size); -        if (ret) -                gf_log (THIS->name, GF_LOG_WARNING, -                        "failed to set size of xfer"); - -        ret = dict_set_uint64 (dict, "lookups", lookup); -        if (ret) -                gf_log (THIS->name, GF_LOG_WARNING, -                        "failed to set lookedup file count"); - -        ret = dict_set_int32 (dict, "status", volinfo->defrag_status); -        if (ret) -                gf_log (THIS->name, GF_LOG_WARNING, -                        "failed to set status"); - -out: -        return 0; -} -  void  glusterd_rebalance_cmd_attempted_log (int cmd, char *volname)  { @@ -338,6 +284,10 @@ glusterd_handle_defrag_start (glusterd_volinfo_t *volinfo, char *op_errstr,          volinfo->defrag_status = GF_DEFRAG_STATUS_STARTED; +        volinfo->rebalance_files = 0; +        volinfo->rebalance_data = 0; +        volinfo->lookedup_files = 0; +          volinfo->defrag_cmd = cmd;          glusterd_store_volinfo (volinfo, GLUSTERD_VOLINFO_VER_AC_INCREMENT); @@ -369,6 +319,8 @@ glusterd_handle_defrag_start (glusterd_volinfo_t *volinfo, char *op_errstr,                           NULL);          runner_add_arg (&runner, "--xlator-option");          runner_argprintf ( &runner, "*dht.rebalance-cmd=%d",cmd); +        runner_add_arg (&runner, "--xlator-option"); +        runner_argprintf (&runner, "*dht.node-uuid=%s", uuid_utoa(priv->uuid));          runner_add_arg (&runner, "--socket-file");          runner_argprintf (&runner, "%s",sockfile);          runner_add_arg (&runner, "--pid-file"); @@ -383,6 +335,7 @@ glusterd_handle_defrag_start (glusterd_volinfo_t *volinfo, char *op_errstr,                  goto out;          } +        sleep (5);          ret = rpc_clnt_transport_unix_options_build (&options, sockfile);          if (ret) {                  gf_log (THIS->name, GF_LOG_ERROR, "Unix options build failed"); @@ -609,7 +562,6 @@ glusterd_op_rebalance (dict_t *dict, char **op_errstr, dict_t *rsp_dict)          int32_t             cmd       = 0;          char                msg[2048] = {0};          glusterd_volinfo_t *volinfo   = NULL; -        void               *node_uuid = NULL;          glusterd_conf_t    *priv      = NULL;          priv = THIS->private; @@ -633,23 +585,6 @@ glusterd_op_rebalance (dict_t *dict, char **op_errstr, dict_t *rsp_dict)                  goto out;          } -        if ((cmd != GF_DEFRAG_CMD_STATUS) && -            (cmd != GF_DEFRAG_CMD_STOP)) { -                ret = dict_get_ptr (dict, "node-uuid", &node_uuid); -                if (ret) { -                        gf_log (THIS->name, GF_LOG_DEBUG, "node-uuid not found"); -                        goto out; -                } - -                /* perform this on only the node which has -                   issued the command */ -                if (uuid_compare (node_uuid, priv->uuid)) { -                        gf_log (THIS->name, GF_LOG_DEBUG, -                                "not the source node %s", uuid_utoa (priv->uuid)); -                        goto out; -                } -        } -          switch (cmd) {          case GF_DEFRAG_CMD_START:          case GF_DEFRAG_CMD_START_LAYOUT_FIX: @@ -659,6 +594,9 @@ glusterd_op_rebalance (dict_t *dict, char **op_errstr, dict_t *rsp_dict)                   break;          case GF_DEFRAG_CMD_STOP:          case GF_DEFRAG_CMD_STATUS: +                volinfo->rebalance_files = 0; +                volinfo->rebalance_data = 0; +                volinfo->lookedup_files = 0;                  break;          default:                  break; diff --git a/xlators/mgmt/glusterd/src/glusterd-rpc-ops.c b/xlators/mgmt/glusterd/src/glusterd-rpc-ops.c index 39a9c6161..4e55c383c 100644 --- a/xlators/mgmt/glusterd/src/glusterd-rpc-ops.c +++ b/xlators/mgmt/glusterd/src/glusterd-rpc-ops.c @@ -1059,6 +1059,8 @@ glusterd_volume_rebalance_use_rsp_dict (dict_t *rsp_dict)          glusterd_op_t  op       = GD_OP_NONE;          uint64_t       value    = 0;          int32_t        value32  = 0; +        char          *volname  = NULL; +        glusterd_volinfo_t *volinfo = NULL;          GF_ASSERT (rsp_dict); @@ -1071,9 +1073,22 @@ glusterd_volume_rebalance_use_rsp_dict (dict_t *rsp_dict)          if (!ctx_dict)                  goto out; +        ret = dict_get_str (ctx_dict, "volname", &volname); +        if (ret) { +                gf_log ("", GF_LOG_ERROR, "Unable to get volume name"); +                goto out; +        } + +        ret  = glusterd_volinfo_find (volname, &volinfo); + +        if (ret) +                goto out; +          ret = dict_get_uint64 (rsp_dict, "files", &value);          if (!ret) { -                ret = dict_set_uint64 (ctx_dict, "files", value); +                volinfo->rebalance_files += value; +                ret = dict_set_uint64 (ctx_dict, "files", +                                       volinfo->rebalance_files);                  if (ret) {                          gf_log (THIS->name, GF_LOG_DEBUG,                                  "failed to set the file count"); @@ -1082,7 +1097,9 @@ glusterd_volume_rebalance_use_rsp_dict (dict_t *rsp_dict)          ret = dict_get_uint64 (rsp_dict, "size", &value);          if (!ret) { -                ret = dict_set_uint64 (ctx_dict, "size", value); +                volinfo->rebalance_data += value; +                ret = dict_set_uint64 (ctx_dict, "size", +                                       volinfo->rebalance_data);                  if (ret) {                          gf_log (THIS->name, GF_LOG_DEBUG,                                  "failed to set the size of migration"); @@ -1091,7 +1108,9 @@ glusterd_volume_rebalance_use_rsp_dict (dict_t *rsp_dict)          ret = dict_get_uint64 (rsp_dict, "lookups", &value);          if (!ret) { -                ret = dict_set_uint64 (ctx_dict, "lookups", value); +                volinfo->lookedup_files += value; +                ret = dict_set_uint64 (ctx_dict, "lookups", +                                       volinfo->lookedup_files);                  if (ret) {                          gf_log (THIS->name, GF_LOG_DEBUG,                                  "failed to set lookuped file count"); @@ -1273,6 +1292,9 @@ glusterd3_1_commit_op_cbk (struct rpc_req *req, struct iovec *iov,                  case GD_OP_REBALANCE:                  case GD_OP_DEFRAG_BRICK_VOLUME: +                        ret = glusterd_volume_rebalance_use_rsp_dict (dict); +                        if (ret) +                                goto out;                  break;                  case GD_OP_HEAL_VOLUME:  | 
