added forgotten files

diff --git a/org.glite.lb.server/src/server_stats.c b/org.glite.lb.server/src/server_stats.c
new file mode 100644 (file)
index 0000000..d194b6e
--- /dev/null
+++ b/org.glite.lb.server/src/server_stats.c
@@ -0,0 +1,126 @@
+#include <sys/file.h>
+#include <sys/mman.h>
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <time.h>
+
+#include "server_stats.h"
+
+#include "glite/lb/context-int.h"
+
+#include "glite/lbu/log.h"
+
+static edg_wll_server_statistics *serverStatisticsMap = NULL;
+static int serverStatisticsFD;
+
+int edg_wll_InitServerStatistics(edg_wll_Context ctx, char *file)
+{
+       //TODO get file name from command line
+        char *fname;
+       if (file)
+               asprintf(&fname, "%s", file);
+       else{
+               asprintf(&fname, "/tmp/lb_server_stats");
+               glite_common_log(LOG_CATEGORY_LB_SERVER, LOG_PRIORITY_INFO,
+                        "server stats file not found, using default %s", fname);
+       }
+       serverStatisticsFD = open(fname, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR);
+       if (serverStatisticsFD < 0) 
+               return edg_wll_SetError(ctx,errno,fname);
+
+       off_t size = lseek(serverStatisticsFD, 0, SEEK_END) - lseek(serverStatisticsFD, 0, SEEK_SET);
+
+       glite_common_log(LOG_CATEGORY_LB_SERVER, LOG_PRIORITY_DEBUG,
+                        "server stats: using %s, size %i", fname, (int)size);
+
+       // modify file size if does not exist or format changed
+       if (size == 0){
+               glite_common_log(LOG_CATEGORY_LB_SERVER, LOG_PRIORITY_INFO,
+                        "Server statistics does not exist, creating them...");
+               edg_wll_server_statistics *zero = calloc(SERVER_STATISTICS_COUNT, sizeof(*serverStatisticsMap));
+               time_t now = time(NULL);
+               int i;
+               for (i = 0; i < SERVER_STATISTICS_COUNT; i++)
+                       zero[i].start = now;
+               
+               write(serverStatisticsFD, zero, SERVER_STATISTICS_COUNT*sizeof(*serverStatisticsMap));
+               free(zero);
+       }
+       else if (size < SERVER_STATISTICS_COUNT*sizeof(*serverStatisticsMap)){
+               int new_stats = SERVER_STATISTICS_COUNT-size/sizeof(*serverStatisticsMap);
+               glite_common_log(LOG_CATEGORY_LB_SERVER, LOG_PRIORITY_INFO,
+                        "Server statistics format extended, adding %i new items.", new_stats);
+               lseek(serverStatisticsFD, 0, SEEK_END);
+               edg_wll_server_statistics *zero = calloc(new_stats, sizeof(*serverStatisticsMap));
+               int i;
+               time_t now = time(NULL);
+               for (i = 0; i < new_stats; i++)
+                       zero[i].start = now;
+               write(serverStatisticsFD, zero, SERVER_STATISTICS_COUNT*sizeof(*serverStatisticsMap) - size);
+               lseek(serverStatisticsFD, 0, SEEK_SET);
+               free(zero);
+       }
+
+       // read and mmap statistics
+       read(serverStatisticsFD, serverStatisticsMap, SERVER_STATISTICS_COUNT*sizeof(*serverStatisticsMap));
+        serverStatisticsMap = mmap(NULL, SERVER_STATISTICS_COUNT*sizeof(*serverStatisticsMap), PROT_READ|PROT_WRITE, MAP_SHARED, serverStatisticsFD, 0);
+        if (serverStatisticsMap == MAP_FAILED) return edg_wll_SetError(ctx,errno,"mmap()");
+
+       msync(serverStatisticsMap, SERVER_STATISTICS_COUNT*sizeof(*serverStatisticsMap), MS_ASYNC);
+
+        return 0;
+}
+
+int edg_wll_ServerStatisticsIncrement(edg_wll_Context ctx, enum edg_wll_server_statistics_type type)
+{
+       if (! serverStatisticsMap)
+               return -1;
+
+       if (type >= SERVER_STATISTICS_COUNT){
+               glite_common_log(LOG_CATEGORY_LB_SERVER, LOG_PRIORITY_WARN,
+                       "edg_wll_ServerStatisticsIncrement(): unknown statistic item.");
+               return edg_wll_SetError(ctx,NULL,"edg_wll_ServerStatisticsIncrement()");
+       }
+
+       if (flock(serverStatisticsFD, LOCK_EX))
+                return edg_wll_SetError(ctx,errno,"flock()");
+       serverStatisticsMap[type].value++;
+       flock(serverStatisticsFD, LOCK_UN);
+
+       return 0;
+}
+
+int edg_wll_ServerStatisticsGetValue(edg_wll_Context ctx, enum edg_wll_server_statistics_type type)
+{
+       if (! serverStatisticsMap)
+               return -1;
+
+        if (type >= SERVER_STATISTICS_COUNT){
+                glite_common_log(LOG_CATEGORY_LB_SERVER, LOG_PRIORITY_WARN,
+                        "edg_wll_ServerStatisticsGetValue(): unknown statistic item.");
+                return -1;
+        }
+
+        if (flock(serverStatisticsFD, LOCK_EX))
+                return edg_wll_SetError(ctx,errno,"flock()");
+        int ret = serverStatisticsMap[type].value;
+        flock(serverStatisticsFD, LOCK_UN);
+
+        return ret;
+}
+
+time_t* edg_wll_ServerStatisticsGetStart(edg_wll_Context ctx, enum edg_wll_server_statistics_type type)
+{
+       if (! serverStatisticsMap)
+               return NULL;
+
+        if (type >= SERVER_STATISTICS_COUNT){
+                glite_common_log(LOG_CATEGORY_LB_SERVER, LOG_PRIORITY_WARN,
+                        "edg_wll_ServerStatisticsGetStart(): unknown statistic item.");
+                return NULL;
+        }
+
+        return &(serverStatisticsMap[type].start);
+}
+

diff --git a/org.glite.lb.server/src/server_stats.h b/org.glite.lb.server/src/server_stats.h
new file mode 100644 (file)
index 0000000..e5276d6
--- /dev/null
+++ b/org.glite.lb.server/src/server_stats.h
@@ -0,0 +1,42 @@
+#ifndef GLITE_LB_SERVER_STATS_H
+#define GLITE_LB_SERVER_STATS_H
+
+#ifndef _GNU_SOURCE
+#define _GNU_SOURCE
+#endif
+
+#include <stdint.h>
+#include "glite/lb/context.h"
+
+typedef struct _edg_wll_server_statistics{
+       uint64_t value;
+       time_t start;
+} edg_wll_server_statistics;
+
+#define SERVER_STATISTICS_COUNT 15
+
+enum edg_wll_server_statistics_type{
+       SERVER_STATS_GLITEJOB_REGS = 0,
+       SERVER_STATS_PBSJOB_REGS,
+       SERVER_STATS_CONDOR_REGS,
+       SERVER_STATS_CREAM_REGS,
+       SERVER_STATS_SANDBOX_REGS,
+       SERVER_STATS_JOB_EVENTS,
+       SERVER_STATS_HTML_VIEWS,
+       SERVER_STATS_TEXT_VIEWS,
+       SERVER_STATS_RSS_VIEWS,
+       SERVER_STATS_NOTIF_LEGACY_REGS,
+       SERVER_STATS_NOTIF_MSG_REGS,
+       SERVER_STATS_NOTIF_LEGACY_SENT,
+       SERVER_STATS_NOTIF_MSG_SENT,
+       SERVER_STATS_WS_QUERIES,
+       SERVER_STATS_LBPROTO
+};
+
+int edg_wll_InitServerStatistics(edg_wll_Context ctx, char *file);
+
+int edg_wll_ServerStatisticsIncrement(edg_wll_Context ctx, enum edg_wll_server_statistics_type type);
+int edg_wll_ServerStatisticsGetValue(edg_wll_Context ctx, enum edg_wll_server_statistics_type type);
+time_t* edg_wll_ServerStatisticsGetStart(edg_wll_Context ctx, enum edg_wll_server_statistics_type type);
+
+#endif /* GLITE_LB_SERVER_STATS_H */