python 基于dns 轮询的业务检测

最新推荐文章于 2024-02-09 18:54:03 发布

运维记事本

最新推荐文章于 2024-02-09 18:54:03 发布

阅读量1.2k

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/nieming92/article/details/48160417

版权

python 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

本脚本源自刘天斯的《python 自动化运维技术与最佳实践》

该脚本功能为判断业务是否正常。过程为：首先脚本会向dns服务器提出解析请求，拿到请求返回结果后（一个域名返回多个A记录）

通过模仿浏览器访问业务ip，比对返回的html头是否与预期头相同，进而判断业务是否正常。

在文中脚本使用python2 缩写，在python3 下几乎不可运行。稍加改动脚本如下：

import dns.resolver
import os
import httplib2

iplist=[]
appdomain = 'www.baidu.com'

def get_iplist(domain=""):                                           ##解析域名为多个ip
    try:
        A = dns.resolver.query(domain,'A')
    except Exception as e:
        print('dns resolver error:' + str(e))
        return
    for i in A:
            iplist.append(i)
    return True

def checkip(ip):
    checkurl = str(ip) + ":80"
    getcontent=""
    httplib2.socket.setdefaulttimeout(5)
    conn = httplib2.HTTPConnectionWithTimeout(checkurl)

    try:
        conn.request("GET","/",headers = {"HOST": appdomain})                    ##通过构造html头访问目标业务主机
        response = conn.getresponse()
        getcontent = response.read(15)
    finally:
        if getcontent == b"<!DOCTYPE html>" :                                ##判断返回字符串是否与预期相同
            print(str(ip)+'[ok]')
        else:
            print(str(ip)+'[error]')
if __name__ =="__main__":
    if get_iplist(appdomain) and len(iplist) > 0:
        for ip in iplist:
            checkip(ip)
    else:
        print('dns resolve error')

文中使用的python LIB 包，dnspython 与httplib 使用pip install 直接安装均报错。

在dnspython 官网下载dnspython 包: