66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package dbs
|
|
|
|
import (
|
|
"database/sql"
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"errors"
|
|
"log"
|
|
|
|
// import for postgres
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
// PHONE struct pg table
|
|
type PHONE struct {
|
|
ID int64 `json:"id" db:"id"`
|
|
MAC string `json:"mac" db:"mac"`
|
|
IP string `json:"ip,omitempty" db:"ip"`
|
|
INVENTORY string `json:"inventory,omitempty" db:"inventory"`
|
|
MODEL string `json:"model,omitempty" db:"model"`
|
|
LANG string `json:"lang,omitempty" db:"lang"`
|
|
FIRMWARE string `json:"firmware,omitempty" db:"firmware"`
|
|
TYPECONN int64 `json:"typeconn" db:"typeconn"`
|
|
EXTS *ExtSlice `json:"exts" db:"exts"`
|
|
STATUS int `json:"busy" db:"busy"`
|
|
}
|
|
|
|
// ExtSlice slice of extension
|
|
type ExtSlice []*EXTS
|
|
|
|
// Value return jsonb to json
|
|
func (e ExtSlice) Value() (driver.Value, error) {
|
|
return json.Marshal(e)
|
|
}
|
|
|
|
// Scan return json to jsonb
|
|
func (e *ExtSlice) Scan(src interface{}) error {
|
|
switch v := src.(type) {
|
|
case []byte:
|
|
return json.Unmarshal(v, &e)
|
|
case string:
|
|
return json.Unmarshal([]byte(v), &e)
|
|
}
|
|
return errors.New("type assertion failed")
|
|
}
|
|
|
|
// EXTS struct extension
|
|
type EXTS struct {
|
|
NUM int64 `json:"num"`
|
|
PASSWORD string `json:"password,omitempty"`
|
|
NAME string `json:"name,omitempty"`
|
|
}
|
|
|
|
// GetPG return pg sql.db
|
|
func GetPG() *sql.DB {
|
|
pgconn, err := sql.Open("postgres", "postgres://<>?sslmode=disable")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err := pgconn.Ping(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
return pgconn
|
|
}
|