Add transactional vise db ability #11
| @ -1,86 +0,0 @@ | |||||||
| package main |  | ||||||
| 
 |  | ||||||
| import ( |  | ||||||
| 	"context" |  | ||||||
| 	"crypto/sha1" |  | ||||||
| 	"flag" |  | ||||||
| 	"fmt" |  | ||||||
| 	"os" |  | ||||||
| 	"path" |  | ||||||
| 
 |  | ||||||
| 	testdataloader "github.com/peteole/testdata-loader" |  | ||||||
| 	"git.defalsify.org/vise.git/logging" |  | ||||||
| 	"git.grassecon.net/grassrootseconomics/sarafu-vise/config" |  | ||||||
| 	"git.grassecon.net/grassrootseconomics/visedriver/storage" |  | ||||||
| 	"git.grassecon.net/grassrootseconomics/sarafu-vise/store" |  | ||||||
| 	storedb "git.grassecon.net/grassrootseconomics/sarafu-vise/store/db" |  | ||||||
| ) |  | ||||||
| 
 |  | ||||||
| var ( |  | ||||||
| 	logg      = logging.NewVanilla() |  | ||||||
| 	baseDir   = testdataloader.GetBasePath() |  | ||||||
| 	scriptDir = path.Join("services", "registration") |  | ||||||
| ) |  | ||||||
| 
 |  | ||||||
| func main() { |  | ||||||
| 	config.LoadConfig() |  | ||||||
| 
 |  | ||||||
| 	var connStr string |  | ||||||
| 	var sessionId string |  | ||||||
| 	var database string |  | ||||||
| 	var engineDebug bool |  | ||||||
| 	var err error |  | ||||||
| 
 |  | ||||||
| 	flag.StringVar(&sessionId, "session-id", "075xx2123", "session id") |  | ||||||
| 	flag.StringVar(&connStr, "c", "", "connection string") |  | ||||||
| 	flag.BoolVar(&engineDebug, "d", false, "use engine debug output") |  | ||||||
| 	flag.Parse() |  | ||||||
| 
 |  | ||||||
| 	if connStr != "" { |  | ||||||
| 		connStr = config.DbConn() |  | ||||||
| 	} |  | ||||||
| 	connData, err := storage.ToConnData(connStr) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Fprintf(os.Stderr, "connstr err: %v", err) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	logg.Infof("start command", "conn", connData) |  | ||||||
| 
 |  | ||||||
| 	ctx := context.Background() |  | ||||||
| 	ctx = context.WithValue(ctx, "SessionId", sessionId) |  | ||||||
| 	ctx = context.WithValue(ctx, "Database", database) |  | ||||||
| 
 |  | ||||||
| 	resourceDir := scriptDir |  | ||||||
| 	menuStorageService := storage.NewMenuStorageService(connData, resourceDir) |  | ||||||
| 	 |  | ||||||
| 	userDb, err := menuStorageService.GetUserdataDb(ctx) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Fprintf(os.Stderr, err.Error()) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| 	userStore := store.UserDataStore{userDb} |  | ||||||
| 
 |  | ||||||
| 	h := sha1.New() |  | ||||||
| 	h.Write([]byte(sessionId)) |  | ||||||
| 	address := h.Sum(nil) |  | ||||||
| 	addressString := fmt.Sprintf("%x", address) |  | ||||||
| 
 |  | ||||||
| 	err = userStore.WriteEntry(ctx, sessionId, storedb.DATA_PUBLIC_KEY, []byte(addressString)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Fprintf(os.Stderr, err.Error()) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	err = userStore.WriteEntry(ctx, addressString, storedb.DATA_PUBLIC_KEY_REVERSE, []byte(sessionId)) |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Fprintf(os.Stderr, err.Error()) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| 	err = userDb.Close() |  | ||||||
| 	if err != nil { |  | ||||||
| 		fmt.Fprintf(os.Stderr, err.Error()) |  | ||||||
| 		os.Exit(1) |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
		Loading…
	
		Reference in New Issue
	
	Block a user