Merge pull request #4 from cockroachdb/nvanbenschoten/gorm

gorm: Setup initial project structure, schema, and db connection

Author: nvb

Makefile

@@ -0,0 +1,30 @@
+# Copyright 2016 The Cockroach Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+# implied. See the License for the specific language governing
+# permissions and limitations under the License. See the AUTHORS file
+# for names of contributors.
+#
+# Author: Nathan VanBenschoten ([email protected])
+
+GO ?= go
+
+.PHONY: all
+all: test
+
+.PHONY: test
+test:
+	$(GO) test -v -i ./testing
+	$(GO) test -v ./testing
+
+.PHONY: deps
+deps:
+	$(GO) get -d -t ./...

go/gorm/.gitignore

@@ -0,0 +1 @@
+gorm

go/gorm/Makefile

@@ -0,0 +1,25 @@
+# Copyright 2016 The Cockroach Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+# implied. See the License for the specific language governing
+# permissions and limitations under the License. See the AUTHORS file
+# for names of contributors.
+#
+# Author: Nathan VanBenschoten ([email protected])
+
+ifneq ($(ADDR),)
+ADDRFLAG = -addr=$(ADDR)
+endif
+
+.PHONY: start
+start:
+	@go build
+	@./gorm $(ADDRFLAG)

go/gorm/main.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"flag"
+	"log"
+	"net/http"
+
+	"github.com/cockroachdb/examples-orms/go/gorm/model"
+	"github.com/jinzhu/gorm"
+	_ "github.com/jinzhu/gorm/dialects/postgres"
+	"github.com/julienschmidt/httprouter"
+)
+
+var (
+	addr = flag.String("addr", "postgresql://root@localhost:26257/company_gorm?sslmode=disable", "the address of the database")
+)
+
+func main() {
+	flag.Parse()
+
+	db := setupDB(*addr)
+	defer db.Close()
+
+	router := httprouter.New()
+
+	server := NewServer(db)
+	server.RegisterRouter(router)
+
+	log.Fatal(http.ListenAndServe(":6543", router))
+}
+
+func setupDB(addr string) *gorm.DB {
+	db, err := gorm.Open("postgres", addr)
+	if err != nil {
+		panic("failed to connect database")
+	}
+
+	// Migrate the schema
+	db.AutoMigrate(&model.Customer{}, &model.Order{}, &model.Product{})
+
+	return db
+}

go/gorm/model/models.go

@@ -0,0 +1,25 @@
+package model
+
+// Customer is a model in the "customers" table.
+type Customer struct {
+	ID   int
+	Name *string `gorm:"not null"`
+}
+
+// Order is a model in the "orders" table.
+type Order struct {
+	ID       int
+	Subtotal float64 `gorm:"type:decimal(18,2)"`
+
+	Customer   Customer `gorm:"ForeignKey:CustomerID"`
+	CustomerID int      `json:"-"`
+
+	Products []Product `gorm:"many2many:order_products"`
+}
+
+// Product is a model in the "products" table.
+type Product struct {
+	ID    int
+	Name  *string `gorm:"not null;unique"`
+	Price float64 `gorm:"type:decimal(18,2)"`
+}