openaccounting-server/migrations/migrate1.go

package main

import (
	"encoding/json"
	"github.com/openaccounting/oa-server/core/model/db"
	"github.com/openaccounting/oa-server/core/model/types"
	"log"
	"os"
)

func main() {
	if len(os.Args) != 2 {
		log.Fatal("Usage: migrate1.go <upgrade/downgrade>")
	}

	command := os.Args[1]

	if command != "upgrade" && command != "downgrade" {
		log.Fatal("Usage: migrate1.go <upgrade/downgrade>")
	}

	//filename is the path to the json config file
	var config types.Config
	file, err := os.Open("./config.json")

	if err != nil {
		log.Fatal(err)
	}

	decoder := json.NewDecoder(file)
	err = decoder.Decode(&config)

	if err != nil {
		log.Fatal(err)
	}

	connectionString := config.User + ":" + config.Password + "@/" + config.Database
	db, err := db.NewDB(connectionString)

	if command == "upgrade" {
		err = upgrade(db)
	} else {
		err = downgrade(db)
	}

	if err != nil {
		log.Fatal(err)
	}

	log.Println("done")
}

func upgrade(db *db.DB) (err error) {
	tx, err := db.Begin()

	if err != nil {
		return
	}

	defer func() {
		if p := recover(); p != nil {
			tx.Rollback()
			panic(p) // re-throw panic after Rollback
		} else if err != nil {
			tx.Rollback()
		} else {
			err = tx.Commit()
		}
	}()

	query1 := "ALTER TABLE user ADD COLUMN signupSource VARCHAR(100) NOT NULL AFTER emailVerifyCode"

	if _, err = tx.Exec(query1); err != nil {
		return
	}

	return
}

func downgrade(db *db.DB) (err error) {
	tx, err := db.Begin()

	if err != nil {
		return
	}

	defer func() {
		if p := recover(); p != nil {
			tx.Rollback()
			panic(p) // re-throw panic after Rollback
		} else if err != nil {
			tx.Rollback()
		} else {
			err = tx.Commit()
		}
	}()

	query1 := "ALTER TABLE user DROP COLUMN signupSource"

	if _, err = tx.Exec(query1); err != nil {
		return
	}

	return
}