如何在Go语言编写的Azure Function中获取订阅ID？

嗨，我之前也碰到过这个困扰，给你几个实用的方案来获取Go编写的Azure Function所在的订阅ID：

方法一：通过Azure实例元数据服务（IMDS）获取（推荐） #

这是Azure官方提供的元数据服务，专门用来获取运行中实例的各种信息，不需要额外配置权限，只要你的Function在Azure内部运行就能访问。

你可以在Go代码里发送HTTP请求到IMDS的端点来直接获取订阅ID，示例代码如下：

package main

import (
    "fmt"
    "io/ioutil"
    "net/http"
)

func getSubscriptionID() (string, error) {
    url := "http://169.254.169.254/metadata/instance/compute/subscriptionId?api-version=2021-02-01&format=text"
    req, err := http.NewRequest("GET", url, nil)
    if err != nil {
        return "", err
    }
    // IMDS要求必须设置这个请求头才能正常返回数据
    req.Header.Set("Metadata", "true")
    
    client := &http.Client{}
    resp, err := client.Do(req)
    if err != nil {
        return "", err
    }
    defer resp.Body.Close()
    
    body, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        return "", err
    }
    return string(body), nil
}

// 在你的Function handler里调用这个方法
func HandleRequest(w http.ResponseWriter, r *http.Request) {
    subID, err := getSubscriptionID()
    if err != nil {
        http.Error(w, fmt.Sprintf("获取订阅ID失败: %v", err), http.StatusInternalServerError)
        return
    }
    fmt.Fprintf(w, "当前订阅ID: %s", subID)
}

⚠️ 注意：这个端点只能在Azure内部运行的实例上访问，本地调试时会失效，你可以在本地测试时加个判断逻辑或者用其他方案替代。

方法二：通过Azure管理SDK获取 #

如果你的Function需要结合其他Azure资源操作，比如查询资源组、虚拟机等，可以用Azure的Go管理SDK来动态获取订阅ID。不过这个方案需要提前给你的Function配置系统托管标识，并且给该标识分配对应的RBAC权限（比如Reader角色）到当前订阅或资源组。

示例代码如下：

package main

import (
    "context"
    "fmt"
    "os"
    "strings"

    "github.com/Azure/azure-sdk-for-go/sdk/azidentity"
    "github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/resources/armresources"
)

func getSubscriptionIDViaSDK() (string, error) {
    // 从环境变量获取当前Function所在的资源组名称（Azure会自动注入这个变量）
    resourceGroup := os.Getenv("WEBSITE_RESOURCE_GROUP")
    if resourceGroup == "" {
        return "", fmt.Errorf("环境变量WEBSITE_RESOURCE_GROUP未设置")
    }

    // 使用系统托管标识创建身份凭证
    cred, err := azidentity.NewDefaultAzureCredential(nil)
    if err != nil {
        return "", err
    }

    // 创建资源组管理客户端
    client, err := armresources.NewResourceGroupsClient(cred, nil)
    if err != nil {
        return "", err
    }

    // 获取目标资源组的详细信息
    resp, err := client.Get(context.TODO(), resourceGroup, nil)
    if err != nil {
        return "", err
    }

    // 从资源组ID中解析订阅ID，资源组ID格式为：/subscriptions/{订阅ID}/resourceGroups/{资源组名称}
    idParts := *resp.ResourceGroup.ID
    parts := strings.Split(idParts, "/")
    if len(parts) < 3 {
        return "", fmt.Errorf("资源组ID格式无效")
    }
    return parts[2], nil
}

这个方案的优势是可以和其他Azure资源操作联动，但需要额外的权限配置，适合有复杂资源管理需求的场景。

方法三：手动配置应用程序设置（最简单） #

如果你的Function所在的订阅是固定不变的，那最省事的办法就是直接在Azure Portal里给Function App添加一个应用程序设置：

1. 打开你的Function App，进入「配置」-「应用程序设置」
2. 添加一个新的设置，比如键为AZURE_SUBSCRIPTION_ID，值填你的实际订阅ID
3. 保存后，Azure会自动将这个变量注入到Function的运行环境中

然后在Go代码里直接读取这个环境变量即可：

package main

import (
    "fmt"
    "os"
    "net/http"
)

func HandleRequest(w http.ResponseWriter, r *http.Request) {
    subID := os.Getenv("AZURE_SUBSCRIPTION_ID")
    if subID == "" {
        http.Error(w, "环境变量AZURE_SUBSCRIPTION_ID未配置", http.StatusInternalServerError)
        return
    }
    fmt.Fprintf(w, "当前订阅ID: %s", subID)
}

这个方案零代码复杂度，本地调试时也可以直接在本地设置环境变量进行测试，适合订阅固定的场景。

备注：内容来源于stack exchange，提问作者fatehks