78 lines
1.2 KiB
Go
78 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/go-xorm/xorm"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
type Status struct {
|
|
Name string
|
|
Color string
|
|
}
|
|
|
|
var (
|
|
Registed Status = Status{"Registed", "white"}
|
|
Approved Status = Status{"Approved", "green"}
|
|
Removed Status = Status{"Removed", "red"}
|
|
Statuses map[string]Status = map[string]Status{
|
|
Registed.Name: Registed,
|
|
Approved.Name: Approved,
|
|
Removed.Name: Removed,
|
|
}
|
|
)
|
|
|
|
func (s *Status) FromDB(bytes []byte) error {
|
|
if r, ok := Statuses[string(bytes)]; ok {
|
|
*s = r
|
|
return nil
|
|
} else {
|
|
return errors.New("no this data")
|
|
}
|
|
}
|
|
|
|
func (s *Status) ToDB() ([]byte, error) {
|
|
return []byte(s.Name), nil
|
|
}
|
|
|
|
type User struct {
|
|
Id int64
|
|
Name string
|
|
Status Status `xorm:"varchar(40)"`
|
|
}
|
|
|
|
func main() {
|
|
f := "conversion.db"
|
|
os.Remove(f)
|
|
|
|
Orm, err := xorm.NewEngine("sqlite3", f)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
Orm.ShowSQL(true)
|
|
err = Orm.CreateTables(&User{})
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
_, err = Orm.Insert(&User{1, "xlw", Registed})
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
users := make([]User, 0)
|
|
err = Orm.Find(&users)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
fmt.Println(users)
|
|
}
|