Update textanalytics.go (#259)

Get subscription key and endpoint from env vars.
Azure-Samples · Sep 30, 2019 · 8e7318b · 8e7318b
1 parent 79e3f3a
commit 8e7318b
Showing 1 changed file with 189 additions and 176 deletions.
diff --git a/cognitiveservices/textanalytics.go b/cognitiveservices/textanalytics.go
@@ -6,194 +6,207 @@
 package cognitiveservices
 
 import (
-	"context"
-	"encoding/json"
-	"fmt"
-
-	"github.com/Azure/azure-sdk-for-go/services/cognitiveservices/v2.1/textanalytics"
-	"github.com/Azure/go-autorest/autorest"
-	"github.com/Azure/go-autorest/autorest/to"
+    "context"
+    "encoding/json"
+    "fmt"
+    "log"
+    "os"
+
+    "github.com/Azure/azure-sdk-for-go/services/cognitiveservices/v2.1/textanalytics"
+    "github.com/Azure/go-autorest/autorest"
+    "github.com/Azure/go-autorest/autorest/to"
 )
 
-func GetTextAnalyticsClient(apiEndPoint string, apiKey string) textanalytics.BaseClient {
-	textAnalyticsClient := textanalytics.New(apiEndPoint)
-	subscriptionKey := apiKey
-	textAnalyticsClient.Authorizer = autorest.NewCognitiveServicesAuthorizer(subscriptionKey)
-	return textAnalyticsClient
+func GetTextAnalyticsClient() textanalytics.BaseClient {
+    var subscriptionKeyVar string = "TEXT_ANALYTICS_SUBSCRIPTION_KEY"
+    if "" == os.Getenv(subscriptionKeyVar) {
+        log.Fatal("Please set/export the environment variable " + subscriptionKeyVar + ".")
+    }
+    var subscriptionKey string = os.Getenv(subscriptionKeyVar)
+    var endpointVar string = "TEXT_ANALYTICS_ENDPOINT"
+    if "" == os.Getenv(endpointVar) {
+        log.Fatal("Please set/export the environment variable " + endpointVar + ".")
+    }
+    var endpoint string = os.Getenv(endpointVar)
+
+    textAnalyticsClient := textanalytics.New(endpoint)
+    textAnalyticsClient.Authorizer = autorest.NewCognitiveServicesAuthorizer(subscriptionKey)
+
+    return textAnalyticsClient
 }
 
 // detects the sentiment of a set of text records
-func SentimentAnalysis(apiEndPoint string, apiKey string) {
-	textAnalyticsclient := GetTextAnalyticsClient(apiEndPoint, apiKey)
-	ctx := context.Background()
-	inputDocuments := []textanalytics.MultiLanguageInput{
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("en"),
-			ID:       to.StringPtr("0"),
-			Text:     to.StringPtr("I had the best day of my life."),
-		},
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("en"),
-			ID:       to.StringPtr("1"),
-			Text:     to.StringPtr("This was a waste of my time. The speaker put me to sleep."),
-		},
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("es"),
-			ID:       to.StringPtr("2"),
-			Text:     to.StringPtr("No tengo dinero ni nada que dar..."),
-		},
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("it"),
-			ID:       to.StringPtr("3"),
-			Text:     to.StringPtr("L'hotel veneziano era meraviglioso. È un bellissimo pezzo di architettura."),
-		},
-	}
-
-	batchInput := textanalytics.MultiLanguageBatchInput{Documents: &inputDocuments}
-	result, _ := textAnalyticsclient.Sentiment(ctx, to.BoolPtr(false), &batchInput)
-	batchResult := textanalytics.SentimentBatchResult{}
-	jsonString, _ := json.Marshal(result.Value)
-	json.Unmarshal(jsonString, &batchResult)
-
-	// Printing sentiment results
-	for _, document := range *batchResult.Documents {
-		fmt.Printf("Document ID: %s ", *document.ID)
-		fmt.Printf("Sentiment Score: %f\n", *document.Score)
-	}
-
-	// Printing document errors
-	fmt.Println("Document Errors")
-	for _, error := range *batchResult.Errors {
-		fmt.Printf("Document ID: %s Message : %s\n", *error.ID, *error.Message)
-	}
+func SentimentAnalysis() {
+    textAnalyticsclient := GetTextAnalyticsClient()
+    ctx := context.Background()
+    inputDocuments := []textanalytics.MultiLanguageInput{
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("en"),
+            ID:       to.StringPtr("0"),
+            Text:     to.StringPtr("I had the best day of my life."),
+        },
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("en"),
+            ID:       to.StringPtr("1"),
+            Text:     to.StringPtr("This was a waste of my time. The speaker put me to sleep."),
+        },
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("es"),
+            ID:       to.StringPtr("2"),
+            Text:     to.StringPtr("No tengo dinero ni nada que dar..."),
+        },
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("it"),
+            ID:       to.StringPtr("3"),
+            Text:     to.StringPtr("L'hotel veneziano era meraviglioso. È un bellissimo pezzo di architettura."),
+        },
+    }
+
+    batchInput := textanalytics.MultiLanguageBatchInput{Documents: &inputDocuments}
+    result, _ := textAnalyticsclient.Sentiment(ctx, to.BoolPtr(false), &batchInput)
+    batchResult := textanalytics.SentimentBatchResult{}
+    jsonString, _ := json.Marshal(result.Value)
+    json.Unmarshal(jsonString, &batchResult)
+
+    // Printing sentiment results
+    for _, document := range *batchResult.Documents {
+        fmt.Printf("Document ID: %s ", *document.ID)
+        fmt.Printf("Sentiment Score: %f\n", *document.Score)
+    }
+
+    // Printing document errors
+    fmt.Println("Document Errors")
+    for _, error := range *batchResult.Errors {
+        fmt.Printf("Document ID: %s Message : %s\n", *error.ID, *error.Message)
+    }
 }
 
 //detects the language of a text document
-func DetectLanguage(apiEndPoint string, apiKey string) {
-	textAnalyticsclient := GetTextAnalyticsClient(apiEndPoint, apiKey)
-	ctx := context.Background()
-	inputDocuments := []textanalytics.LanguageInput{
-		textanalytics.LanguageInput{
-			ID:   to.StringPtr("0"),
-			Text: to.StringPtr("This is a document written in English."),
-		},
-		textanalytics.LanguageInput{
-			ID:   to.StringPtr("1"),
-			Text: to.StringPtr("Este es un document escrito en Español."),
-		},
-		textanalytics.LanguageInput{
-			ID:   to.StringPtr("2"),
-			Text: to.StringPtr("这是一个用中文写的文件"),
-		},
-	}
-
-	batchInput := textanalytics.LanguageBatchInput{Documents: &inputDocuments}
-	result, _ := textAnalyticsclient.DetectLanguage(ctx, to.BoolPtr(false), &batchInput)
-
-	// Printing language detection results
-	for _, document := range *result.Documents {
-		fmt.Printf("Document ID: %s ", *document.ID)
-		fmt.Printf("Detected Languages with Score: ")
-		for _, language := range *document.DetectedLanguages {
-			fmt.Printf("%s %f,", *language.Name, *language.Score)
-		}
-		fmt.Println()
-	}
-
-	// Printing document errors
-	fmt.Println("Document Errors")
-	for _, error := range *result.Errors {
-		fmt.Printf("Document ID: %s Message : %s\n", *error.ID, *error.Message)
-	}
+func DetectLanguage() {
+    textAnalyticsclient := GetTextAnalyticsClient()
+    ctx := context.Background()
+    inputDocuments := []textanalytics.LanguageInput{
+        textanalytics.LanguageInput{
+            ID:   to.StringPtr("0"),
+            Text: to.StringPtr("This is a document written in English."),
+        },
+        textanalytics.LanguageInput{
+            ID:   to.StringPtr("1"),
+            Text: to.StringPtr("Este es un document escrito en Español."),
+        },
+        textanalytics.LanguageInput{
+            ID:   to.StringPtr("2"),
+            Text: to.StringPtr("这是一个用中文写的文件"),
+        },
+    }
+
+    batchInput := textanalytics.LanguageBatchInput{Documents: &inputDocuments}
+    result, _ := textAnalyticsclient.DetectLanguage(ctx, to.BoolPtr(false), &batchInput)
+
+    // Printing language detection results
+    for _, document := range *result.Documents {
+        fmt.Printf("Document ID: %s ", *document.ID)
+        fmt.Printf("Detected Languages with Score: ")
+        for _, language := range *document.DetectedLanguages {
+            fmt.Printf("%s %f,", *language.Name, *language.Score)
+        }
+        fmt.Println()
+    }
+
+    // Printing document errors
+    fmt.Println("Document Errors")
+    for _, error := range *result.Errors {
+        fmt.Printf("Document ID: %s Message : %s\n", *error.ID, *error.Message)
+    }
 }
 
 // extracts key-phrases from a text documen
-func ExtractKeyPhrases(apiEndPoint string, apiKey string) {
-	textAnalyticsclient := GetTextAnalyticsClient(apiEndPoint, apiKey)
-	ctx := context.Background()
-	inputDocuments := []textanalytics.MultiLanguageInput{
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("ja"),
-			ID:       to.StringPtr("0"),
-			Text:     to.StringPtr("猫は幸せ"),
-		},
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("de"),
-			ID:       to.StringPtr("1"),
-			Text:     to.StringPtr("Fahrt nach Stuttgart und dann zum Hotel zu Fu."),
-		},
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("en"),
-			ID:       to.StringPtr("2"),
-			Text:     to.StringPtr("My cat might need to see a veterinarian."),
-		},
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("es"),
-			ID:       to.StringPtr("3"),
-			Text:     to.StringPtr("A mi me encanta el fútbol!"),
-		},
-	}
-
-	batchInput := textanalytics.MultiLanguageBatchInput{Documents: &inputDocuments}
-	result, _ := textAnalyticsclient.KeyPhrases(ctx, to.BoolPtr(false), &batchInput)
-
-	// Printing extracted key phrases results
-	for _, document := range *result.Documents {
-		fmt.Printf("Document ID: %s\n", *document.ID)
-		fmt.Printf("\tExtracted Key Phrases:\n")
-		for _, keyPhrase := range *document.KeyPhrases {
-			fmt.Printf("\t\t%s\n", keyPhrase)
-		}
-		fmt.Println()
-	}
-
-	// Printing document errors
-	fmt.Println("Document Errors")
-	for _, error := range *result.Errors {
-		fmt.Printf("Document ID: %s Message : %s\n", *error.ID, *error.Message)
-	}
+func ExtractKeyPhrases() {
+    textAnalyticsclient := GetTextAnalyticsClient()
+    ctx := context.Background()
+    inputDocuments := []textanalytics.MultiLanguageInput{
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("ja"),
+            ID:       to.StringPtr("0"),
+            Text:     to.StringPtr("猫は幸せ"),
+        },
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("de"),
+            ID:       to.StringPtr("1"),
+            Text:     to.StringPtr("Fahrt nach Stuttgart und dann zum Hotel zu Fu."),
+        },
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("en"),
+            ID:       to.StringPtr("2"),
+            Text:     to.StringPtr("My cat might need to see a veterinarian."),
+        },
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("es"),
+            ID:       to.StringPtr("3"),
+            Text:     to.StringPtr("A mi me encanta el fútbol!"),
+        },
+    }
+
+    batchInput := textanalytics.MultiLanguageBatchInput{Documents: &inputDocuments}
+    result, _ := textAnalyticsclient.KeyPhrases(ctx, to.BoolPtr(false), &batchInput)
+
+    // Printing extracted key phrases results
+    for _, document := range *result.Documents {
+        fmt.Printf("Document ID: %s\n", *document.ID)
+        fmt.Printf("\tExtracted Key Phrases:\n")
+        for _, keyPhrase := range *document.KeyPhrases {
+            fmt.Printf("\t\t%s\n", keyPhrase)
+        }
+        fmt.Println()
+    }
+
+    // Printing document errors
+    fmt.Println("Document Errors")
+    for _, error := range *result.Errors {
+        fmt.Printf("Document ID: %s Message : %s\n", *error.ID, *error.Message)
+    }
 }
 
 //  identifies well-known entities in a text document
-func ExtractEntities(apiEndPoint string, apiKey string) {
-	textAnalyticsclient := GetTextAnalyticsClient(apiEndPoint, apiKey)
-	ctx := context.Background()
-	inputDocuments := []textanalytics.MultiLanguageInput{
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("en"),
-			ID:       to.StringPtr("0"),
-			Text:     to.StringPtr("Microsoft was founded by Bill Gates and Paul Allen on April 4, 1975, to develop and sell BASIC interpreters for the Altair 8800."),
-		},
-		textanalytics.MultiLanguageInput{
-			Language: to.StringPtr("es"),
-			ID:       to.StringPtr("1"),
-			Text:     to.StringPtr("La sede principal de Microsoft se encuentra en la ciudad de Redmond, a 21 kilómetros de Seattle."),
-		},
-	}
-
-	batchInput := textanalytics.MultiLanguageBatchInput{Documents: &inputDocuments}
-	result, _ := textAnalyticsclient.Entities(ctx, to.BoolPtr(false), &batchInput)
-
-	// Printing extracted entities results
-	for _, document := range *result.Documents {
-		fmt.Printf("Document ID: %s\n", *document.ID)
-		fmt.Printf("\tExtracted Entities:\n")
-		for _, entity := range *document.Entities {
-			fmt.Printf("\t\tName: %s\tType: %s", *entity.Name, *entity.Type)
-			if entity.SubType != nil {
-				fmt.Printf("\tSub-Type: %s\n", *entity.SubType)
-			}
-			fmt.Println()
-			for _, match := range *entity.Matches {
-				fmt.Printf("\t\t\tOffset: %v\tLength: %v\tScore: %f\n", *match.Offset, *match.Length, *match.EntityTypeScore)
-			}
-		}
-		fmt.Println()
-	}
-
-	// Printing document errors
-	fmt.Println("Document Errors")
-	for _, error := range *result.Errors {
-		fmt.Printf("Document ID: %s Message : %s\n", *error.ID, *error.Message)
-	}
+func ExtractEntities() {
+    textAnalyticsclient := GetTextAnalyticsClient()
+    ctx := context.Background()
+    inputDocuments := []textanalytics.MultiLanguageInput{
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("en"),
+            ID:       to.StringPtr("0"),
+            Text:     to.StringPtr("Microsoft was founded by Bill Gates and Paul Allen on April 4, 1975, to develop and sell BASIC interpreters for the Altair 8800."),
+        },
+        textanalytics.MultiLanguageInput{
+            Language: to.StringPtr("es"),
+            ID:       to.StringPtr("1"),
+            Text:     to.StringPtr("La sede principal de Microsoft se encuentra en la ciudad de Redmond, a 21 kilómetros de Seattle."),
+        },
+    }
+
+    batchInput := textanalytics.MultiLanguageBatchInput{Documents: &inputDocuments}
+    result, _ := textAnalyticsclient.Entities(ctx, to.BoolPtr(false), &batchInput)
+
+    // Printing extracted entities results
+    for _, document := range *result.Documents {
+        fmt.Printf("Document ID: %s\n", *document.ID)
+        fmt.Printf("\tExtracted Entities:\n")
+        for _, entity := range *document.Entities {
+            fmt.Printf("\t\tName: %s\tType: %s", *entity.Name, *entity.Type)
+            if entity.SubType != nil {
+                fmt.Printf("\tSub-Type: %s\n", *entity.SubType)
+            }
+            fmt.Println()
+            for _, match := range *entity.Matches {
+                fmt.Printf("\t\t\tOffset: %v\tLength: %v\tScore: %f\n", *match.Offset, *match.Length, *match.EntityTypeScore)
+            }
+        }
+        fmt.Println()
+    }
+
+    // Printing document errors
+    fmt.Println("Document Errors")
+    for _, error := range *result.Errors {
+        fmt.Printf("Document ID: %s Message : %s\n", *error.ID, *error.Message)
+    }
 }