client: support per-client job files
Currently the fio network client sends all job files to all clients.
This isn't very flexible.
Add support for specifying a specific job file for each client. Example:
$ fio --client hostname1 jobfile1 --client hostname2 jobfile2
would before add jobfile1 and jobfile2 as job files for both clients.
With this change, only jobfile1 is sent to hostname1, and only
jobfile2 is sent to hostname2.
Signed-off-by: Jens Axboe <axboe@kernel.dk>
diff --git a/client.c b/client.c
index c1e516d..a9b63e2 100644
--- a/client.c
+++ b/client.c
@@ -111,6 +111,10 @@
free(client->argv);
if (client->name)
free(client->name);
+ while (client->nr_ini_file)
+ free(client->ini_file[--client->nr_ini_file]);
+ if (client->ini_file)
+ free(client->ini_file);
free(client);
}
@@ -227,6 +231,19 @@
return NULL;
}
+void fio_client_add_ini_file(void *cookie, const char *ini_file)
+{
+ struct fio_client *client = cookie;
+ size_t new_size;
+
+ dprint(FD_NET, "client <%s>: add ini %s\n", client->hostname, ini_file);
+
+ new_size = (client->nr_ini_file + 1) * sizeof(char *);
+ client->ini_file = realloc(client->ini_file, new_size);
+ client->ini_file[client->nr_ini_file] = strdup(ini_file);
+ client->nr_ini_file++;
+}
+
int fio_client_add(struct client_ops *ops, const char *hostname, void **cookie)
{
struct fio_client *existing = *cookie;
@@ -620,7 +637,18 @@
flist_for_each_safe(entry, tmp, &client_list) {
client = flist_entry(entry, struct fio_client, list);
- if (fio_client_send_ini(client, filename))
+ if (client->nr_ini_file) {
+ int i;
+
+ for (i = 0; i < client->nr_ini_file; i++) {
+ const char *ini = client->ini_file[i];
+
+ if (fio_client_send_ini(client, ini)) {
+ remove_client(client);
+ break;
+ }
+ }
+ } else if (!filename || fio_client_send_ini(client, filename))
remove_client(client);
}
diff --git a/client.h b/client.h
index acb5a88..341d260 100644
--- a/client.h
+++ b/client.h
@@ -61,6 +61,9 @@
struct client_ops *ops;
void *client_data;
+
+ char **ini_file;
+ unsigned int nr_ini_file;
};
struct cmd_iolog_pdu;
@@ -120,6 +123,7 @@
extern int fio_client_add(struct client_ops *, const char *, void **);
extern struct fio_client *fio_client_add_explicit(struct client_ops *, const char *, int, int);
extern void fio_client_add_cmd_option(void *, const char *);
+extern void fio_client_add_ini_file(void *, const char *);
extern int fio_client_terminate(struct fio_client *);
extern void fio_clients_terminate(void);
extern struct fio_client *fio_get_client(struct fio_client *);
diff --git a/init.c b/init.c
index 2b581fb..ae96c6c 100644
--- a/init.c
+++ b/init.c
@@ -1577,6 +1577,18 @@
exit_val = 1;
break;
}
+ /*
+ * If the next argument exists and isn't an option,
+ * assume it's a job file for this client only.
+ */
+ while (optind < argc) {
+ if (!strncmp(argv[optind], "--", 2) ||
+ !strncmp(argv[optind], "-", 1))
+ break;
+
+ fio_client_add_ini_file(cur_client, argv[optind]);
+ optind++;
+ }
break;
default:
do_exit++;
@@ -1660,6 +1672,11 @@
free(ini_file[i]);
}
}
+ } else if (nr_clients) {
+ if (fill_def_thread())
+ return 1;
+ if (fio_clients_send_ini(NULL))
+ return 1;
}
free(ini_file);