Code style cleanup. (No functional changes)

[moira.git] / backup / db2rest.awk

#	$Source$
#	$Header$
#
#	This converts the file used to originally create the database
#	into a program to restore it from a backup.
#	This is not guaranteed to work for all data types; it may
#	need to be extended.

BEGIN {
	print "/* This file automatically generated */";
	print "/* Do not edit */\n";
	print "#include <stdio.h>";
	print "EXEC SQL INCLUDE sqlca;";
	print "EXEC SQL WHENEVER SQLERROR DO dbmserr();";
	print "void parse_nl(FILE *), parse_str(FILE *, char *, int), parse_sep(FILE *);\n";

	print "/* This file automatically generated */" > "rest1.pc";
	print "/* Do not edit */\n" >> "rest1.pc";
	print "#include <stdio.h>" >> "rest1.pc";
	print "FILE *open_file(char *prefix, char *suffix);\n" >> "rest1.pc";
	print "int do_restores(char *prefix)\n{" >> "rest1.pc";
}

$1=="#" { next; }

/^create/ {
	printf "int restore_%s(FILE *f)\n", $3;
	print "{\n  EXEC SQL BEGIN DECLARE SECTION;";
	printf "  restore_%s(open_file(prefix, \"%s\"));\n", $3, $3 >> "rest1.pc";

	tablename = $3;
	rangename = substr(tablename, 1, 1);
	count = 0;
	next;
}

NF >= 2 {
	vname[count] = $1;
	printf "  /* %s */\n", $0;
	if ($2 ~ /INTEGER/ || $2 ~ /SMALLINT/ || $2 ~ /INTEGER1/) {
		printf "  int\tt_%s;\n", vname[count];
		vtype[count]="int";
	} else if ($2 ~ /CHAR\([0-9]*\)/) {
		t = split($2, temp, "(");
		if (t != 2) printf "Can't parse %s\n", $2;
		t = split(temp[2], temp2, ")");
		if (t != 2) printf "Can't parse %s\n", temp[2];
		printf "  char\tt_%s[%d];\n", vname[count], temp2[1]+1;
		if ($1 == "signature") {
			vtype[count]="bin";
			printf "  EXEC SQL VAR t_signature IS STRING(%d);\n", temp2[1]+1;
		} else vtype[count]="str";
		vsize[count] = temp2[1]+1;
	} else if ($2 ~ /DATE/) {
		printf "  char\tt_%s[26];\n", vname[count];
		vtype[count]="date";
	} else printf "Unknown data type %s\n", $2;
	count++;
}

/^\);$/ {
	printf "  EXEC SQL END DECLARE SECTION;\n\n  int count = 0;\n";

	print "  while (!feof(f))\n    {";
	print "      if (!(++count % 100))\n        EXEC SQL COMMIT;\n";

	for (i = 0; i < count; i++) {
		if (i != 0) print "      parse_sep(f);";
		if (vtype[i] ~ /int/) {
			printf("      t_%s = parse_int(f);\n", vname[i]);
		} else if (vtype[i] ~ /date/) {
			printf "      parse_str(f, t_%s, 26);\n", vname[i];
		} else {
			printf "      parse_str(f, t_%s, %d);\n", vname[i], vsize[i];
		}
		if (i == 0) print "      if (feof(f))\n        break;";
	}
	printf "      parse_nl(f);\n"

	printf "      EXEC SQL INSERT INTO %s (\n", tablename;
	for (i = 0; i < count; i++) {
		if (i != 0) printf ",\n";
		printf "          %s", vname[i];
	}
	printf ")\n        VALUES (\n";
	for (i = 0; i < count; i++) {
		if (i != 0) printf ",\n";
		if (vtype[i] ~ /date/) {
			printf "          TO_DATE(NVL(:t_%s,TO_CHAR(SYSDATE, 'DD_mon-YYYY HH24:MI:SS')), 'DD-mon-YYYY HH24:MI:SS')", vname[i];
		} else if(vtype[i] ~ /int/) {
			printf "          :t_%s", vname[i];
		} else {
			printf "          NVL(:t_%s,CHR(0))", vname[i];
		}
	}
	printf ");\n      if (sqlca.sqlcode != 0)\n        {\n";
	printf "          sqlca.sqlerrm.sqlerrmc[sqlca.sqlerrm.sqlerrml] = 0;\n";
	printf "          printf(\"%%s\\n\", sqlca.sqlerrm.sqlerrmc);\n";
	printf "          com_err(\"restore\", 0, \"insert failed\");\n";
	printf "          exit(2);\n        }\n    }\n";
	printf "  fclose(f);\n";
	printf "  EXEC SQL COMMIT;\n";
	printf "}\n\n";
}


END {
	print "/* All done */";
	print "}" >> "rest1.pc";
}