diff --git a/api/main.go b/api/main.go index c006c50..3cf7a2d 100644 --- a/api/main.go +++ b/api/main.go @@ -10,6 +10,7 @@ import ( _ "github.com/go-sql-driver/mysql" "github.com/goapt/dotenv" "github.com/ilibs/gosql/v2" + "github.com/rs/cors" "github.com/fifsky/genstruct/generator" ) @@ -38,7 +39,11 @@ func main() { db := gosql.Use("default") gen := generator.NewGenerator(db) - http.HandleFunc("/api/struct/gen", func(w http.ResponseWriter, r *http.Request) { + c := cors.New(cors.Options{ + AllowedOrigins: []string{"http://api.fifsky.com","https://api.fifsky.com"}, + }) + + handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { body, err := ioutil.ReadAll(r.Body) if err != nil { w.Write([]byte(fmt.Sprintf("request body read error \n%s", err))) @@ -90,7 +95,9 @@ func main() { w.Write(st) }) - err = http.ListenAndServe(":8989", nil) + http.Handle("/api/struct/gen", c.Handler(handler)) + + err = http.ListenAndServe(":8989",nil) if err != nil { log.Fatal("ListenAndServe", err) diff --git a/go.mod b/go.mod index 0adcb27..2978bb5 100644 --- a/go.mod +++ b/go.mod @@ -8,4 +8,5 @@ require ( github.com/ilibs/gosql/v2 v2.0.2 github.com/mattn/go-runewidth v0.0.8 // indirect github.com/olekukonko/tablewriter v0.0.4 + github.com/rs/cors v1.7.0 // indirect ) diff --git a/go.sum b/go.sum index 1bffea0..ee49610 100644 --- a/go.sum +++ b/go.sum @@ -18,6 +18,8 @@ github.com/mattn/go-sqlite3 v1.9.0 h1:pDRiWfl+++eC2FEFRy6jXmQlvp4Yh3z1MJKg4UeYM/ github.com/mattn/go-sqlite3 v1.9.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc= github.com/olekukonko/tablewriter v0.0.4 h1:vHD/YYe1Wolo78koG299f7V/VAS08c6IpCLn+Ejf/w8= github.com/olekukonko/tablewriter v0.0.4/go.mod h1:zq6QwlOf5SlnkVbMSr5EoBv3636FWnp+qbPhuoO21uA= +github.com/rs/cors v1.7.0 h1:+88SsELBHx5r+hZ8TCkggzSstaWNbDvThkVK8H6f9ik= +github.com/rs/cors v1.7.0/go.mod h1:gFx+x8UowdsKA9AchylcLynDq+nNFfI8FkUZdN/jGCU= golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= golang.org/x/net v0.0.0-20190603091049-60506f45cf65/go.mod h1:HSz+uSET+XFnRR8LxR5pz3Of3rY3CfYBVs4xY44aLks= golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=