完成基本功能

gigibox

2023-06-15 ff3cadba4a63cd1b63cd0e36358f49ccedb88bef

 models/db.go

@@ -1,39 +1,27 @@
package models

import (
   "fmt"
   "github.com/jinzhu/gorm"

   _ "github.com/jinzhu/gorm/dialects/mssql"
   _ "github.com/jinzhu/gorm/dialects/sqlite"
)

var db *gorm.DB
var err error

// Init .
func Init(username, password, addr, dbName string) error {
   var err error

   sqlServer := fmt.Sprintf("sqlserver://%s:%s@%s:1433?database=%s",
      username, password, addr, dbName)
   fmt.Println(sqlServer)
func Init() error {
   // 打开数据库连接
   //db, err = gorm.Open("mssql", "sqlserver://sa:LZdba@)@)@10.6.201.7:1433?database=LZGS")
   db, err = gorm.Open("mssql", sqlServer)
   db, err = gorm.Open("sqlite3", "kingdee-api.db")
   if err != nil {
      return err
   }

   return nil
}
   db.AutoMigrate(&Order{})

func GetDB() *gorm.DB {
   return db
   return nil
}

// CloseDB .
func CloseDB() {
   if db != nil {
      db.Close()
   }
   db.Close()
}