Add some logging.

kujtimiihoxha · Apr 10, 2017 · be49181 · be49181
1 parent c520590
commit be49181
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 17 deletions.
diff --git a/generator/generator.go b/generator/generator.go
@@ -205,12 +205,14 @@ func (sg *ServiceInitGenerator) Generate(name string) error {
 func (sg *ServiceInitGenerator) generateTransport(name string, iface *parser.Interface, transport string) error {
 	switch transport {
 	case "http":
+		logrus.Info("Selected http transport.")
 		return sg.generateHttpTransport(name, iface)
 	default:
 		return errors.New(fmt.Sprintf("Transport `%s` not supported", transport))
 	}
 }
 func (sg *ServiceInitGenerator) generateHttpTransport(name string, iface *parser.Interface) error {
+	logrus.Info("Generating http transport...")
 	te := template.NewEngine()
 	defaultFs := fs.Get()
 	handlerFile := parser.NewFile()
@@ -329,6 +331,7 @@ func (sg *ServiceInitGenerator) generateHttpTransport(name string, iface *parser
 	return defaultFs.WriteFile(tfile, handlerFile.String(), false)
 }
 func (sg *ServiceInitGenerator) generateEndpoints(name string, iface *parser.Interface) error {
+	logrus.Info("Generating endpoints...")
 	te := template.NewEngine()
 	defaultFs := fs.Get()
 	enpointsPath, err := te.ExecuteString(viper.GetString("endpoints.path"), map[string]string{

diff --git a/main.go b/main.go
@@ -1,25 +1,11 @@
-// Copyright © 2017 NAME HERE <EMAIL ADDRESS>
-//
-// 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.
-
 package main
 
 import (
+	"github.com/Sirupsen/logrus"
 	"github.com/kujtimiihoxha/gk/cmd"
 	"github.com/spf13/viper"
-	"strings"
 	"os"
-	"github.com/Sirupsen/logrus"
+	"strings"
 )
 
 func main() {
@@ -30,7 +16,7 @@ func main() {
 		logrus.Error(err)
 		return
 	}
-	if !strings.HasPrefix(pwd,gosrc) {
+	if !strings.HasPrefix(pwd, gosrc) {
 		logrus.Error("The project must be in the $GOPATH/src folder for the generator to work.")
 		return
 	}