wormhole/event_database/database.go

40 lines
1.4 KiB
Go

package main
import (
"flag"
"log"
)
// tableName is a const rather than an arg because using different BigTable instances
// will be more common than having multiple tables in a single instance.
// Table name is also passed to devnet guardians.
const tableName = "v2Events"
// These column family names match the guardian code that does the inserting.
var columnFamilies = []string{"MessagePublication", "Signatures", "VAAState", "QuorumState"}
func main() {
project := flag.String("project", "", "The Google Cloud Platform project ID. Required.")
instance := flag.String("instance", "", "The Google Cloud Bigtable instance ID. Required.")
keyFilePath := flag.String("keyFilePath", "", "The Google Cloud Service Account json key file path.")
setupDB := flag.Bool("setupDB", false, "Run database setup - create table and column families.")
rowKey := flag.String("queryRowKey", "", "Query by row key, print the retrieved values.")
previousMinutes := flag.Int("queryPreviousMinutes", 0, "Query for rows with a Timestamp in the last X minutes.")
flag.Parse()
for _, f := range []string{"project", "instance", "keyFilePath"} {
if flag.Lookup(f).Value.String() == "" {
log.Fatalf("The %s flag is required.", f)
}
}
if *setupDB {
RunSetup(*project, *instance, *keyFilePath)
}
if *rowKey != "" || *previousMinutes != 0 {
Query(*project, *instance, *keyFilePath, *rowKey, *previousMinutes)
}
}