github.com/gofunct/common@v0.0.0-20190131174352-fd058c7fbf22/pkg/db/service.go (about)

     1  package db
     2  
     3  import (
     4  	_ "github.com/go-sql-driver/mysql"
     5  	"github.com/jinzhu/gorm"
     6  	_ "github.com/jinzhu/gorm/dialects/sqlite"
     7  )
     8  
     9  type Model struct {
    10  	gorm.Model
    11  }
    12  
    13  func OpenSqlite3(args ...string) *gorm.DB {
    14  	db, err := gorm.Open("sqlite3", args)
    15  	if err != nil {
    16  		panic("failed to connect sqlite3 database")
    17  	}
    18  	return db
    19  }
    20  func OpenMySql(args ...string) *gorm.DB {
    21  	db, err := gorm.Open("mysql", args)
    22  	if err != nil {
    23  		panic("failed to connect mysql database")
    24  	}
    25  	return db
    26  }