[moira.git] / gen / stats.dc

/* $Header$
 *
 * This generates a list of how many lockers are on each NFS server
 * for statistics gathering purposes.
 *
 *  (c) Copyright 1991 by the Massachusetts Institute of Technology.
 *  For copying and distribution information, please see the file
 *  <mit-copyright.h>.
 */

#include <mit-copyright.h>
#include <stdio.h>
#include <moira.h>
#include <moira_site.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <strings.h>
EXEC SQL INCLUDE sqlca;

extern int errno;
char *whoami = "stats.gen";


main(argc, argv)
int argc;
char **argv;
{
    FILE *out = stdout;
    char *outf = NULL, outft[64];
    struct stat sb;
    int flag1, flag2, flag3;
    struct hash *machines;
    EXEC SQL BEGIN DECLARE SECTION;
    char name[257];
    int n, id;
    EXEC SQL END DECLARE SECTION;

#ifsql INGRES
    EXEC SQL CONNECT sms;
#endsql
#ifsql INFORMIX
    EXEC SQL DATABASE sms;
#endsql

    if (argc == 2) {
	if (stat(argv[1], &sb) == 0) {
	    if (ModDiff (&flag1, "machine", sb.st_mtime) == 0 &&
		ModDiff (&flag2, "filesys", sb.st_mtime) == 0 &&
		ModDiff (&flag3, "nfsphys", sb.st_mtime) == 0 &&
		flag1 < 0 && flag2 < 0 && flag3 < 0) {
		fprintf(stderr, "File %s does not need to be rebuilt.\n",
			argv[1]);
		exit(MR_NO_CHANGE);
	    }
	}
	outf = argv[1];
	sprintf(outft, "%s~", outf);
	if ((out = fopen(outft, "w")) == NULL) {
	    fprintf(stderr, "unable to open %s for output\n", outf);
	    exit(MR_OCONFIG);
	}
    } else if (argc != 1) {
	fprintf(stderr, "usage: %s [outfile]\n", argv[0]);
	exit(MR_ARGS);
    } else {
	outf = NULL;
    }

    EXEC SQL WHENEVER SQLERROR GOTO sqlerr;

    machines = create_hash(1000);

    EXEC SQL DECLARE m CURSOR FOR 
      SELECT mach_id, name 
	FROM machine;
    EXEC SQL OPEN m;
    while (1) {
      EXEC SQL FETCH m INTO :id, :name;
      if (sqlca.sqlcode != 0) break;
      hash_store(machines, id, strsave(name));
    }
    EXEC SQL CLOSE m;

    EXEC SQL DECLARE x CURSOR FOR
      SELECT DISTINCT f.mach_id, count(f.filsys_id)
      FROM filesys f
	WHERE f.type="NFS"
      GROUP BY f.mach_id;
    EXEC SQL OPEN x;
    while (1) {
	EXEC SQL FETCH x INTO :id, :n;
	if (sqlca.sqlcode != 0) break;
	fprintf(out, "%s %d\n", hash_lookup(machines, id), n);
    }
    EXEC SQL CLOSE x;

#ifsql INGRES
    EXEC SQL DISCONNECT;
#endsql
#ifsql INFORMIX
    EXEC SQL CLOSE DATABASE;
#endsql

    if (fclose(out)) {
	perror("close failed");
	exit(MR_CCONFIG);
    }
    if (outf)
      fix_file(outf);
    exit(MR_SUCCESS);

 sqlerr:
    com_err(whoami, MR_INGRES_ERR, " code %d\n", sqlca.sqlcode);
    critical_alert("DCM", "Central America build encountered INGRES ERROR %d",
		   sqlca.sqlcode);
    exit(MR_INGRES_ERR);
}
Commit	Line	Data
0966656e	1	/* $Header$
	2	*
	3	* This generates a list of how many lockers are on each NFS server
	4	* for statistics gathering purposes.
	5	*
	6	* (c) Copyright 1991 by the Massachusetts Institute of Technology.
	7	* For copying and distribution information, please see the file
	8	* <mit-copyright.h>.
	9	*/
	10
	11	#include <mit-copyright.h>
	12	#include <stdio.h>
	13	#include <moira.h>
	14	#include <moira_site.h>
	15	#include <sys/types.h>
	16	#include <sys/stat.h>
	17	#include <sys/time.h>
	18	#include <strings.h>
	19	EXEC SQL INCLUDE sqlca;
	20
	21	extern int errno;
	22	char *whoami = "stats.gen";
	23
	24
	25	main(argc, argv)
	26	int argc;
	27	char **argv;
	28	{
	29	FILE *out = stdout;
	30	char *outf = NULL, outft[64];
	31	struct stat sb;
	32	int flag1, flag2, flag3;
	33	struct hash *machines;
	34	EXEC SQL BEGIN DECLARE SECTION;
	35	char name[257];
	36	int n, id;
	37	EXEC SQL END DECLARE SECTION;
	38
	39	#ifsql INGRES
	40	EXEC SQL CONNECT sms;
	41	#endsql
	42	#ifsql INFORMIX
	43	EXEC SQL DATABASE sms;
	44	#endsql
	45
	46	if (argc == 2) {
	47	if (stat(argv[1], &sb) == 0) {
	48	if (ModDiff (&flag1, "machine", sb.st_mtime) == 0 &&
	49	ModDiff (&flag2, "filesys", sb.st_mtime) == 0 &&
	50	ModDiff (&flag3, "nfsphys", sb.st_mtime) == 0 &&
	51	flag1 < 0 && flag2 < 0 && flag3 < 0) {
	52	fprintf(stderr, "File %s does not need to be rebuilt.\n",
	53	argv[1]);
	54	exit(MR_NO_CHANGE);
	55	}
	56	}
	57	outf = argv[1];
	58	sprintf(outft, "%s~", outf);
	59	if ((out = fopen(outft, "w")) == NULL) {
	60	fprintf(stderr, "unable to open %s for output\n", outf);
	61	exit(MR_OCONFIG);
	62	}
	63	} else if (argc != 1) {
	64	fprintf(stderr, "usage: %s [outfile]\n", argv[0]);
65	exit(MR_ARGS);
66	} else {
67	outf = NULL;
68	}
69
70	EXEC SQL WHENEVER SQLERROR GOTO sqlerr;
71
72	machines = create_hash(1000);
73
74	EXEC SQL DECLARE m CURSOR FOR
75	SELECT mach_id, name
76	FROM machine;
77	EXEC SQL OPEN m;
78	while (1) {
79	EXEC SQL FETCH m INTO :id, :name;
80	if (sqlca.sqlcode != 0) break;
81	hash_store(machines, id, strsave(name));
82	}
83	EXEC SQL CLOSE m;
84
85	EXEC SQL DECLARE x CURSOR FOR
86	SELECT DISTINCT f.mach_id, count(f.filsys_id)
87	FROM filesys f
88	WHERE f.type="NFS"
89	GROUP BY f.mach_id;
90	EXEC SQL OPEN x;
91	while (1) {
92	EXEC SQL FETCH x INTO :id, :n;
93	if (sqlca.sqlcode != 0) break;
94	fprintf(out, "%s %d\n", hash_lookup(machines, id), n);
95	}
96	EXEC SQL CLOSE x;
97
98	#ifsql INGRES
99	EXEC SQL DISCONNECT;
100	#endsql
101	#ifsql INFORMIX
102	EXEC SQL CLOSE DATABASE;
103	#endsql
104
105	if (fclose(out)) {
106	perror("close failed");
107	exit(MR_CCONFIG);
108	}
109	if (outf)
110	fix_file(outf);
111	exit(MR_SUCCESS);
112
113	sqlerr:
114	com_err(whoami, MR_INGRES_ERR, " code %d\n", sqlca.sqlcode);
115	critical_alert("DCM", "Central America build encountered INGRES ERROR %d",
116	sqlca.sqlcode);
117	exit(MR_INGRES_ERR);
118	}