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
}