/
main.go
64 lines (54 loc) · 1.53 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package main
import (
"flag"
"fmt"
"github.com/bhameyie/cassymig/files"
"github.com/bhameyie/cassymig/migrate"
"github.com/bhameyie/cassymig/versioning"
"github.com/gocql/gocql"
)
func buildSession(host, keyspace, username, password string) (*gocql.Session, error) {
cluster := gocql.NewCluster(host)
cluster.Keyspace = keyspace
cluster.Authenticator = gocql.PasswordAuthenticator{
Username: username,
Password: password,
}
return cluster.CreateSession()
}
func main() {
directionPtr := flag.String("direction", "up", "up or down")
folderPtr := flag.String("source", ".", "scripts folder")
uriPtr := flag.String("uri", "localhost", "CASSANDRA")
kspacePtr := flag.String("kspace", "service", "CASSANDRA keyspace")
userPtr := flag.String("user", "cassandra", "CASSANDRA user")
passPtr := flag.String("pass", "cassandra", "CASSANDRA pass")
flag.Parse()
fmt.Println("Cassymig")
fmt.Println("")
session, err := buildSession(*uriPtr, *kspacePtr, *userPtr, *passPtr)
if err != nil {
panic(err)
}
if err := versioning.EnsureTableExists(session); err != nil {
panic(err)
}
defer session.Close()
service := &versioning.VersionService{Session: session}
repo := &files.CqlFileRepo{Path: *folderPtr}
switch *directionPtr {
case "up":
if err := migration.MigrateUp(repo, service); err != nil {
fmt.Println(err)
panic(err)
}
case "down":
if err := migration.MigrateDown(repo, service); err != nil {
fmt.Println(err)
panic(err)
}
default:
panic("unrecognized mode: " + *directionPtr)
}
fmt.Println("Success!!")
}