merge

app/lib/company/company.go

@@ -5,11 +5,17 @@ import (
 )
 
 type Company struct {
+	Id           int    `db:"id"`
 	Name         string `db:"name"`
 	MemberNumber string `db:"member_number"`
 	Username     string `db:"username"`
 	Password     string `db:"password"`
 }
+type CompanyInfo struct {
+	Company Company
+	Month   float64
+	Year    float64
+}
 
 func (c *Company) Create() error {
 	sq := `
@@ -19,3 +25,40 @@ VALUES (:name, :member_number, :username, :password)
 	_, err := run.DB.NamedExec(sq, &c)
 	return err
 }
+func (c *Company) Edit() error {
+	sq := `
+	UPDATE company SET 
+		name = :name, 
+    	member_number = :member_number, 
+    	username = :username, 
+    	password = :password
+	WHERE id = :id;`
+	_, err := run.DB.NamedExec(sq, &c)
+	return err
+}
+
+//func GetCompany(companyName string) (Company, error) {
+//	sq := `
+//		SELECT * FROM company WHERE company = ?
+//	`
+//	err := run.DB.Get(&Company, sq, companyName)
+//	if err != nil {
+//		return Company{}, err
+//	}
+//	return Company{}, nil
+//}
+//
+
+func GetCompany(companyID float64) (Company, error) {
+	var company Company
+
+	sq := `
+		SELECT * FROM company WHERE id = ?
+	`
+	err := run.DB.Get(&company, sq, companyID)
+	if err != nil {
+		return Company{}, err
+	}
+
+	return company, nil
+}

app/lib/company/company_test.go

@@ -1,27 +1,27 @@
-package company
-
-import (
-	"bordrobot/lib/run"
-	"testing"
-)
-
-func TestCompany_Create(t *testing.T) {
-	err := run.InitDB("../..")
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	c := &Company{
-		Name:         "Deneme Firması",
-		MemberNumber: "1234",
-		Username:     "u1",
-		Password:     "p1",
-	}
-
-	err = c.Create()
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	run.DB.Close()
-}
+package company
+
+import (
+	"bordrobot/lib/run"
+	"testing"
+)
+
+func TestCompany_Create(t *testing.T) {
+	err := run.InitDB("../..")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	c := &Company{
+		Name:         "Deneme Firması",
+		MemberNumber: "1234",
+		Username:     "u1",
+		Password:     "p1",
+	}
+
+	err = c.Create()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	run.DB.Close()
+}