[Sheepdog] [PATCH] support specifying output log file
MORITA Kazutaka
morita.kazutaka at lab.ntt.co.jp
Tue Apr 6 07:12:11 CEST 2010
When running more than one collies in the same machine, it is
confusing to send all the output to the syslog.
You can specify the output file with -o option.
$ collie -o /tmp/sdog /store
(log messages are sent to /tmp/sdog)
$ script/start-sheepdog -n=3 -o=/tmp/sdog- -d=/store
(log messages are sent to /tmp/sdog-0, /tmp/sdog-1, and /tmp/sdog-2)
Signed-off-by: MORITA Kazutaka <morita.kazutaka at lab.ntt.co.jp>
---
collie/collie.c | 14 ++++++++++++--
include/logger.h | 3 ++-
lib/logger.c | 21 +++++++++++++++++----
script/start-sheepdog | 15 +++++++++++----
4 files changed, 42 insertions(+), 11 deletions(-)
diff --git a/collie/collie.c b/collie/collie.c
index 8367487..0acc2d2 100644
--- a/collie/collie.c
+++ b/collie/collie.c
@@ -25,12 +25,13 @@ static struct option const long_options[] = {
{"port", required_argument, 0, 'p'},
{"foreground", no_argument, 0, 'f'},
{"loglevel", required_argument, 0, 'l'},
+ {"outfile", required_argument, 0, 'o'},
{"debug", no_argument, 0, 'd'},
{"help", no_argument, 0, 'h'},
{0, 0, 0, 0},
};
-static char *short_options = "p:fl:dh";
+static char *short_options = "p:fl:o:dh";
static void usage(int status)
{
@@ -44,6 +45,7 @@ Sheepdog Daemon, version %s\n\
-p, --port specify the listen port number\n\
-f, --foreground make the program run in the foreground\n\
-l, --loglevel specify the message level printed by default\n\
+ -o, --outfile location for output (defaults to the system log)\n\
-d, --debug print debug messages\n\
-h, --help display this help and exit\n\
", SD_VERSION);
@@ -60,6 +62,7 @@ int main(int argc, char **argv)
char *dir = DEFAULT_OBJECT_DIR;
int is_daemon = 1;
int log_level = LOG_INFO;
+ char *out_file = NULL;
while ((ch = getopt_long(argc, argv, short_options, long_options,
&longindex)) >= 0) {
@@ -72,6 +75,12 @@ int main(int argc, char **argv)
break;
case 'l':
log_level = atoi(optarg);
+ break;
+ case 'o':
+ out_file = optarg;
+ if (strcmp(out_file, "syslog") == 0)
+ out_file = NULL;
+ break;
case 'd':
/* removed soon. use loglevel instead */
log_level = LOG_DEBUG;
@@ -88,7 +97,8 @@ int main(int argc, char **argv)
if (optind != argc)
dir = argv[optind];
- ret = log_init(program_name, LOG_SPACE_SIZE, is_daemon, log_level);
+ ret = log_init(program_name, LOG_SPACE_SIZE, is_daemon, log_level,
+ out_file);
if (ret)
exit(1);
diff --git a/include/logger.h b/include/logger.h
index 7fa4cf1..6b4a399 100644
--- a/include/logger.h
+++ b/include/logger.h
@@ -55,9 +55,10 @@ struct logarea {
char *buff;
int semid;
union semun semarg;
+ int fd;
};
-extern int log_init(char * progname, int size, int daemon, int level);
+extern int log_init(char * progname, int size, int daemon, int level, char *outfile);
extern void log_close (void);
extern void dump_logmsg (void *);
extern void log_write(int prio, const char *func, int line, const char *fmt, ...)
diff --git a/lib/logger.c b/lib/logger.c
index 41dc0af..bd0e8e2 100644
--- a/lib/logger.c
+++ b/lib/logger.c
@@ -132,6 +132,8 @@ static int logarea_init (int size)
static void free_logarea (void)
{
+ if (la->fd >= 0)
+ close(la->fd);
semctl(la->semid, 0, IPC_RMID, la->semarg);
shmdt(la->buff);
shmdt(la->start);
@@ -260,7 +262,10 @@ static void log_syslog (void * buff)
{
struct logmsg * msg = (struct logmsg *)buff;
- syslog(msg->prio, "%s", (char *)&msg->str);
+ if (la->fd >= 0)
+ write(la->fd, (char *)&msg->str, strlen((char *)&msg->str));
+ else
+ syslog(msg->prio, "%s", (char *)&msg->str);
}
static void dolog(int prio, const char *func, int line, const char *fmt, va_list ap)
@@ -342,7 +347,7 @@ static void log_flush(void)
}
}
-int log_init(char *program_name, int size, int daemon, int level)
+int log_init(char *program_name, int size, int daemon, int level, char *outfile)
{
log_level = level;
@@ -356,8 +361,15 @@ int log_init(char *program_name, int size, int daemon, int level)
struct sigaction sa_new;
int fd;
- openlog(log_name, 0, LOG_DAEMON);
- setlogmask (LOG_UPTO (LOG_DEBUG));
+ if (outfile) {
+ fd = open(outfile, O_CREAT | O_RDWR, 0644);
+ if (fd < 0)
+ syslog(LOG_ERR, "failed to open %s\n", outfile);
+ } else {
+ fd = -1;
+ openlog(log_name, 0, LOG_DAEMON);
+ setlogmask (LOG_UPTO (LOG_DEBUG));
+ }
if (logarea_init(size)) {
syslog(LOG_ERR, "failed to initialize the logger\n");
@@ -365,6 +377,7 @@ int log_init(char *program_name, int size, int daemon, int level)
}
la->active = 1;
+ la->fd = fd;
pid = fork();
if (pid < 0) {
syslog(LOG_ERR, "fail to fork the logger\n");
diff --git a/script/start-sheepdog b/script/start-sheepdog
index 8f4b09a..c527288 100755
--- a/script/start-sheepdog
+++ b/script/start-sheepdog
@@ -8,6 +8,8 @@ $port = 7000;
$nr = 1;
@hosts = ();
$home = ".";
+$outfile_prefix = "";
+$outfile = "syslog";
$dir = "/tmp/".rand(100);
@@ -18,6 +20,8 @@ while (@ARGV && $ARGV[0] =~ m/^-/) {
$nr = $1;
} elsif (m/^-d=(.*)$/) {
$dir = $1;
+ } elsif (m/^-o=(.*)$/) {
+ $outfile_prefix = $1;
} elsif (m/^-h=(.*)$/) {
@hosts = &expand(split ':', $1);
} elsif (m/^-H=(.*)$/) {
@@ -36,12 +40,15 @@ foreach $host (grep {!$count{$_}++} @hosts) {
for ($i = 0; $i < $nr; $i++, $port++) {
$_dir = $dir.$i;
+ if ($outfile_prefix) {
+ $outfile = $outfile_prefix.$i;
+ }
if ($hosts[$i]) {
- print("ssh $hosts[$i] $home/collie/collie --port $port $_dir -d\n");
- system("ssh $hosts[$i] $home/collie/collie --port $port $_dir -d");
+ print("ssh $hosts[$i] $home/collie/collie --port $port $_dir -d -o $outfile\n");
+ system("ssh $hosts[$i] $home/collie/collie --port $port $_dir -d -o $outfile");
} else {
- print("$home/collie/collie --port $port $_dir -d\n");
- system("$home/collie/collie --port $port $_dir -d");
+ print("$home/collie/collie --port $port $_dir -d -o $outfile\n");
+ system("$home/collie/collie --port $port $_dir -d -o $outfile");
}
}
--
1.5.6.5
More information about the sheepdog
mailing list