Database runs a small CRUD example using gorm:

- Only Simulator model is used
- The postgres Array type is provided by lib/pq
- The default empty string is fixed
This commit is contained in:
smavros 2019-03-31 19:19:38 +02:00
parent b17d66dbf8
commit 64aeda36bd
2 changed files with 49 additions and 74 deletions

View file

@ -1,78 +1,52 @@
package main package main
import ( import (
"database/sql" "fmt"
"fmt" "git.rwth-aachen.de/acs/public/villas/villasweb-backend-go/simulator"
_ "github.com/lib/pq" "github.com/jinzhu/gorm"
//"time" _ "github.com/jinzhu/gorm/dialects/postgres"
) "log"
)
const ( type Simulator simulator.Simulator
//DB_USER = "postgres"
//DB_PASSWORD = "postgres"
//DB_NAME = "test"
DB_USER = "odiseas"
DB_PASSWORD = "toumpacity"
DB_NAME = "mytestdb"
)
func main() { const (
dbinfo := fmt.Sprintf("user=%s password=%s dbname=%s sslmode=disable", DB_NAME = "villasdb"
DB_USER, DB_PASSWORD, DB_NAME) )
db, err := sql.Open("postgres", dbinfo)
checkErr(err)
defer db.Close()
//fmt.Println("# Inserting values") func main() {
// Init connection's information
dbinfo := fmt.Sprintf("host=/tmp sslmode=disable dbname=%s", DB_NAME)
db, err := gorm.Open("postgres", dbinfo)
checkErr(err)
defer db.Close()
//var lastInsertId int // Check that db is reachable
//err = db.QueryRow("INSERT INTO userinfo(username,departname,created) VALUES($1,$2,$3) returning uid;", "astaxie", "研发部门", "2012-12-09").Scan(&lastInsertId) err = db.DB().Ping()
//checkErr(err) checkErr(err)
//fmt.Println("last inserted id =", lastInsertId)
//fmt.Println("# Updating") // Migrate one model
//stmt, err := db.Prepare("update userinfo set username=$1 where uid=$2") db.AutoMigrate(&Simulator{})
//checkErr(err)
//res, err := stmt.Exec("astaxieupdate", lastInsertId) // Create
//checkErr(err) db.Create(&Simulator{UUID: "12"})
//affect, err := res.RowsAffected() // Read
//checkErr(err) var dummy Simulator
db.First(&dummy, 1)
fmt.Printf("%s\n", dummy.UUID)
//fmt.Println(affect, "rows changed") // Update
db.Model(&dummy).Update("UUID", "100")
db.First(&dummy, 1)
fmt.Printf("%s\n", dummy.UUID)
//fmt.Println("# Querying") // Delete
//rows, err := db.Query("SELECT * FROM userinfo") db.Unscoped().Delete(&dummy)
//checkErr(err) }
//for rows.Next() { func checkErr(err error) {
//var uid int if err != nil {
//var username string log.Fatal(err)
//var department string }
//var created time.Time }
//err = rows.Scan(&uid, &username, &department, &created)
//checkErr(err)
//fmt.Println("uid | username | department | created ")
//fmt.Printf("%3v | %8v | %6v | %6v\n", uid, username, department, created)
//}
//fmt.Println("# Deleting")
//stmt, err = db.Prepare("delete from userinfo where uid=$1")
//checkErr(err)
//res, err = stmt.Exec(lastInsertId)
//checkErr(err)
//affect, err = res.RowsAffected()
//checkErr(err)
//fmt.Println(affect, "rows changed")
}
func checkErr(err error) {
if err != nil {
fmt.Println("Something went wrong!!")
panic(err)
}
}

View file

@ -2,17 +2,18 @@ package simulator
import ( import (
"github.com/jinzhu/gorm" "github.com/jinzhu/gorm"
"github.com/lib/pq"
"time" "time"
) )
type Simulator struct { type Simulator struct {
gorm.Model gorm.Model
UUID string `gorm:"unique;not null"` UUID string `gorm:"unique;not null"`
Host string `gorm:"default:"` Host string `gorm:"default:''"`
Model string `gorm:"default:"` Modeltype string `gorm:"default:''"`
Uptime int `gorm:"default:0"` Uptime int `gorm:"default:0"`
State string `gorm:"default:"` State string `gorm:"default:''"`
StateUpdateAt Time `gorm:"default:Time.Now"` StateUpdateAt time.Time
Properties []string // TODO: mixed type Properties pq.StringArray `gorm:"type:varchar(128)[]"` // TODO: mixed type
RawProperties []string // TODO: mixed type RawProperties pq.StringArray `gorm:"type:varchar(128)[]"` // TODO: mixed type
} }