40 lines
799 B
Go
40 lines
799 B
Go
package cmds
|
|
|
|
import (
|
|
"database/sql"
|
|
"inp2025/handlers/database"
|
|
"inp2025/middlewares"
|
|
"inp2025/tcp"
|
|
"inp2025/utils"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/uptrace/bun"
|
|
"github.com/uptrace/bun/dialect/sqlitedialect"
|
|
"github.com/uptrace/bun/driver/sqliteshim"
|
|
)
|
|
|
|
var databaseCmd = &cobra.Command{
|
|
Use: "database",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
router := tcp.NewRouter().
|
|
Use(middlewares.ErrorHandler).
|
|
Use(middlewares.AccessLog)
|
|
|
|
sqldb, err := sql.Open(sqliteshim.ShimName, "file::memory:?cache=shared")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
db := bun.NewDB(sqldb, sqlitedialect.New())
|
|
utils.InitDB(db)
|
|
|
|
handlers := database.NewHandlers(db)
|
|
|
|
router.GET("/users", handlers.GetUsers)
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
databaseCmd.Flags().
|
|
String("port", "5432", "")
|
|
}
|