Add set-count subcommand

goatherd.go

@@ -13,6 +13,7 @@ import (
 	"log"
 	"net"
 	"os"
+	"strconv"
 	"strings"
 	"sync"
 	"time"
@@ -537,6 +538,16 @@ func main() {
 			user := args[0]
 			secret := args[1]
 			create_user(user, secret)
+		case "set-counter":
+			if len(args) != 2 {
+				log.Fatalf("set-counter: Invalid number of arguments: %v (expecting <username> <counter>)\n", len(args))
+			}
+			user := args[0]
+			counter, err := strconv.ParseUint(args[1], 0, 64)
+			err_fatal(err)
+			if err = set_count(db, user, counter); err != nil {
+				err_fatal(err)
+			}
 		case "serve":
 			fs := flag.NewFlagSet("serve", flag.ExitOnError)
 			flag_addr := fs.String("addr", "", "Address to listen on. (default 127.0.0.1:9999)")