mirror of https://github.com/mitchell/selfpass.git
72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"math/rand"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/mediocregopher/radix/v3"
|
||
|
|
"github.com/spf13/pflag"
|
||
|
|
|
||
|
|
"github.com/mitchell/selfpass/services/credentials/types"
|
||
|
|
"github.com/mitchell/selfpass/services/migrations/migration"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
redisHost := pflag.StringP("redis-host", "r", "127.0.0.1:6379", "specify the redis host to target")
|
||
|
|
help := pflag.BoolP("help", "h", false, "see help")
|
||
|
|
pflag.Parse()
|
||
|
|
|
||
|
|
if *help {
|
||
|
|
pflag.PrintDefaults()
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
pool, err := radix.NewPool("tcp", *redisHost, 10)
|
||
|
|
migration.Check(err)
|
||
|
|
|
||
|
|
fmt.Println("Beginning migration...")
|
||
|
|
|
||
|
|
var pipeCmds []radix.CmdAction
|
||
|
|
var creds []*types.Credential
|
||
|
|
scanner := radix.NewScanner(pool, radix.ScanAllKeys)
|
||
|
|
|
||
|
|
for key := ""; scanner.Next(&key); {
|
||
|
|
var cred types.Credential
|
||
|
|
pipeCmds = append(pipeCmds, radix.Cmd(&cred, "HGETALL", key))
|
||
|
|
pipeCmds = append(pipeCmds, radix.Cmd(nil, "DEL", key))
|
||
|
|
creds = append(creds, &cred)
|
||
|
|
}
|
||
|
|
|
||
|
|
migration.Check(pool.Do(radix.Pipeline(pipeCmds...)))
|
||
|
|
pipeCmds = nil
|
||
|
|
|
||
|
|
for _, cred := range creds {
|
||
|
|
tcred := *cred
|
||
|
|
tcred.ID = generateID()
|
||
|
|
|
||
|
|
fmt.Printf("Migrating %s to %s.\n", cred.ID, tcred.ID)
|
||
|
|
|
||
|
|
pipeCmds = append(pipeCmds, radix.FlatCmd(nil, "HMSET", tcred.ID, tcred))
|
||
|
|
}
|
||
|
|
|
||
|
|
migration.Check(pool.Do(radix.Pipeline(pipeCmds...)))
|
||
|
|
|
||
|
|
fmt.Println("Done migrating.")
|
||
|
|
}
|
||
|
|
|
||
|
|
func generateID() string {
|
||
|
|
const idLen = 8
|
||
|
|
const alphanumerics = "abcdefghijkmnopqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ0123456789"
|
||
|
|
const alphaLen = len(alphanumerics)
|
||
|
|
|
||
|
|
rand.Seed(time.Now().UnixNano())
|
||
|
|
id := make([]byte, idLen)
|
||
|
|
|
||
|
|
for index := range id {
|
||
|
|
id[index] = alphanumerics[rand.Int63()%int64(alphaLen)]
|
||
|
|
}
|
||
|
|
|
||
|
|
return fmt.Sprintf("%s-%s", types.KeyCredential, string(id))
|
||
|
|
}
|