用查表代替 switch - struct, union 與 enum 的應用
很久以前，我已忘了是在哪本書看到的事(或許不只一本書)。書中說使用 switch..case.. 的場合，大部份都可以也應該改用查表方式代替。 這句話的意義也含括了一句程式設計領域的名言，即「資料結構 + 演算法 = 程式」。
大多數場合，連續的 switch..case.. 或 if..elseif.. 只是不斷複製類似的程式片段。 這種文章結構，沒有運用資料結構觀念，就連演算法的部份也很粗糙，是很糟糕的程式碼。 而在改用查表方式的重構過程中，規劃資料結構建「表」，並設計「查」的演算法， 就自然而然地實踐了「資料結構 + 演算法 = 程式」這句話，也提高了程式碼的品質。
我日前就專案需求，用 C 語言寫了一個程式。程式中有一段載入組態設置文件的動作。 這個動作的設計過程，符合查表法代替switch..case..的目標。 而且這段程式碼用了些少見的表述方式(但仍是 ANSI C 規範)，可以展現 C 語言語法的理解程度。 故本文以此例演示。
首先，來看看只用 if..elseif.. (switch..case..的變形) 的程式碼長什麼樣。
// gcc -o config-load1 config-load1.c parse_conf_line.c
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
int parse_conf_line(const char *line, char *buf, char **out_key, char **out_value);
// defs.h
#define CONFIG_PATH "/tmp/test.conf"
#define DEFAULT_SERVER "localhost"
#define LISTEN_PORT 5000
#define TIMEOUT 10
#define BUFLEN 1024
// end defs.h
// Global config variables.
const char *Server = DEFAULT_SERVER;
int Port = LISTEN_PORT;
time_t Timeout = TIMEOUT;
void load_config()
{
char buf[BUFLEN], buf2[BUFLEN];
char *key, *value;
FILE *fp = fopen(CONFIG_PATH, "r");
if ( fp ) {
while (fgets(buf, BUFLEN, fp)) {
if (parse_conf_line(buf, buf2, &key, &value) < 1)
continue;
if (!key || value[0] == 0)
continue;
if (strcmp("server", key) == 0) {
Server = strcpy(
(char*)malloc(strlen(value)+1), value);
printf("Set %s as %s\n", key, Server);
}
else if (strcmp("port", key) == 0) {
Port = atoi(value);
printf("Set %s as %d\n", key, Port);
}
else if (strcmp("timeout", key) == 0) {
Timeout = atoi(value);
printf("Set %s as %d\n", key, Timeout);
}
/*
more duplicate codes.
.
.
.
*/
}
fclose(fp);
}
else {
printf("%s not found. Use default settings.\n", CONFIG_PATH);
}
}
int main()
{
load_config();
printf("Server : %s\n", Server);
printf("Port : %d\n", Port);
printf("Timeout : %d\n", Timeout);
return 0;
}key = value 形式存在的文件即可。
config-load1.c 最大的問題就在 if..elseif.. 這一段。 每一段 if 只是決定組態名稱，並重複類似的變數值指派動作。 隨著可組態項目陸續增加，這一段程式碼就會被複製地愈來愈長，重複的程式片段也就愈來愈多。
接著再看用查表方式重構後的 config-load.c 。
// gcc -o config-load config-load.c parse_conf_line.c
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
int parse_conf_line(const char *line, char *buf, char **out_key, char **out_value);
// defs.h
#define CONFIG_PATH "/tmp/test.conf"
#define DEFAULT_SERVER "localhost"
#define DEFAULT_PATH "/local"
#define LISTEN_PORT 5000
#define TIMEOUT 10
#define PROCESS_UID 1000
#define PROCESS_GID 1000
#define BUFLEN 1024
// end defs.h
// Global config variables.
const char *Server = DEFAULT_SERVER;
const char *Path = DEFAULT_PATH;
int Port = LISTEN_PORT;
time_t Timeout = TIMEOUT;
uid_t ProcessUid = PROCESS_UID;
gid_t ProcessGid = PROCESS_GID;
int LogLevel = 5;
struct {
char *key;
enum {
STR,
INT
} value_type;
union {
const char **str_var;
int *int_var;
};
}
config_table[] = {
{"server", STR, {.str_var = &Server}},
{"path", STR, {.str_var = &Path}},
{"port", INT, {.int_var = &Port}},
{"timeout", INT, {.int_var = (int*) &Timeout}},
{"process_uid", INT, {.int_var = (int*) &ProcessUid}},
{"process_gid", INT, {.int_var = (int*) &ProcessGid}},
{"log_level", INT, {.int_var = &LogLevel}},
{NULL, STR, {NULL}} // end
},
*conf_item = config_table;
void load_config()
{
char buf[BUFLEN], buf2[BUFLEN];
char *key, *value;
FILE *fp = fopen(CONFIG_PATH, "r");
if ( fp ) {
while (fgets(buf, BUFLEN, fp)) {
if (parse_conf_line(buf, buf2, &key, &value) < 1)
continue;
if (!key || value[0] == 0)
continue;
for (conf_item = config_table;
conf_item->key;
++conf_item)
{
if (strcmp(conf_item->key, key) != 0)
continue;
if (conf_item->value_type == STR) {
*conf_item->str_var = strcpy(
(char*)malloc(strlen(value)+1), value);
printf("Set %s as %s\n", key, *conf_item->str_var);
}
else {
*conf_item->int_var = atoi(value);
printf("Set %s as %d\n", key, *conf_item->int_var);
}
}
}
fclose(fp);
}
else {
printf("%s not found. Use default settings.\n", CONFIG_PATH);
}
}
int main()
{
load_config();
printf("Server : %s\n", Server);
printf("Path : %s\n", Path);
printf("Port : %d\n", Port);
printf("Timeout : %d\n", Timeout);
printf("Uid : %d\n", ProcessUid);
printf("Gid : %d\n", ProcessGid);
printf("LogLevel: %d\n", LogLevel);
return 0;
}config-load.c 比較難理解的應該是下列這一段內容。
struct {
char *key;
enum {
STR,
INT
} value_type;
union {
const char **str_var;
int *int_var;
};
}簡單地說，我宣告了一個匿名結構(struct)。 這個結構包含了三個欄位，分別是字元指標的 key，enum 型別的 value_type，以及匿名union的 var 指標。var 指標指向一個實際儲存組態值的全域變數。雖然也可以用 void* 含混帶過，但編譯器會抱怨，且使用時也經常要帶上強迫轉型行為。故不如以 union 的形式明確表述。
緊接著這個匿名結構宣告語法後的符號 config_table 與 conf_item ，表示套用此匿名結構定義的變數。其中的 conf_item 還是指標型態，指向 config_table 。
不熟悉匿名表達形式的人，可以按照自己的理解，嘗試把名稱補上去，應該就會看懂了。
至於 {.str_var = &Server} 的 .str_var、.int_var 表述方式，則是在設定變數初值時，不採位置順序而是明確指定欄位的語法。
看懂這張表後，再去看演算法就不難理解了。 一個簡單的 for 迴圈找出符合的組態項目，然後判斷型別，調用適合的存值方法。
這個範例還有進一步的改善空間。 例如把分散的各個組態變數全部彙集成一張組態鍵值表，就更具使用彈性，只是會犧牲一些執行效率。
另一方面，在此也可以看出提供動態型別系統的程式語言如何提高程式開發生產力。 上述用 C 語言設計的表與演算法，在動態型別系統中，以 PHP 為例，能以下列形式簡單做到。
<?php
function load_config(&$config_table)
{
$fh = fopen(CONFIG_PATH, "r");
if ( $fh ) {
while (fgets($buf, $fh)) {
if ( !($kv = parse_conf_line($buf)))
continue;
list($key, $value) = $kv;
$config_table[$key] = $value;
}
fclose($fh);
}
}
?>多數動態型別程式語言皆內建鍵值表或關聯式陣列的資料型態，令程式人員可以很簡單地建表。
再者，像本文的 C 語言範例所建的表中，包含了一個 value_type 欄位，用以判斷該用什麼方法儲存組態值。 而動態型別語言則不必如此麻煩，一個泛用的指派行為(=)就可以實現。 這些都是動態型別語言的長處，程式人員應積極運用。
樂多舊回應