Changeset 16573

Timestamp:

Feb 21, 2008, 2:15:54 PM (18 years ago)

Author:

eugene

Message:

mods by Bill on head

File:

• branches/eam_branch_20080219/ippTools/src/pstamptool.c (modified) (2 diffs)

branches/eam_branch_20080219/ippTools/src/pstamptool.c

@@ -108 +108 @@
 
     psString lastFileset = psMetadataLookupStr(&status, config->args, "-last_fileset");
+    psString state = psMetadataLookupStr(&status, config->args, "-state");
+    if (!state) {
+        state = "enabled";
+    }
 
     if (!pstampDataStoreInsert(config->dbh,
             0,
             uri,
-            lastFileset
+            lastFileset,
+            state
         )) {
         psError(PS_ERR_UNKNOWN, false, "database error");
@@ -182 +187 @@
         return false;
     }
-    if (!lastFileset) {
-        psError(PS_ERR_UNKNOWN, true, "-last_fileset is required");
-        return false;
-    }
-
-    char *query ="UPDATE pstampDataStore"
-                 " SET lastFileset = '%s'"
-                 " WHERE ds_id = '%s'";
-
-    if (!p_psDBRunQuery(config->dbh, query, lastFileset, ds_id)) {
+    psString state = psMetadataLookupStr(&status, config->args, "-state");
+    if (!status) {
+        psError(PS_ERR_UNKNOWN, false, "failed to lookup value for -state");
+        return false;
+    }
+
+    if (!state && !lastFileset) {
+        psError(PS_ERR_UNKNOWN, true, "one of -last_fileset or -state is required");
+        return false;
+    }
+
+    char *query = NULL;
+    bool needComma = false;
+    
+    psStringAppend(&query, "UPDATE pstampDataStore SET");
+
+    if (lastFileset) {
+        psStringAppend(&query, " lastFileset = '%s'", lastFileset);
+        needComma = true;
+    }
+
+    if (state) {
+        psStringAppend(&query, "%s state = '%s'", needComma ? "," : " ",
+            state);
+        needComma = true; // be ready in case we add another field
+    }
+                
+    psStringAppend(&query, " WHERE ds_id = '%s'", ds_id);
+
+    if (!p_psDBRunQuery(config->dbh, query)) {
         psError(PS_ERR_UNKNOWN, false, "database error");
         psFree(query);